resolver_mock.go

package upstream

// Code generated by http://github.com/gojuno/minimock (dev). DO NOT EDIT.

import (
	"sync"
	mm_atomic "sync/atomic"
	mm_time "time"

	"github.com/gojuno/minimock/v3"
	"oss.indeed.com/go/modprox/pkg/coordinates"
)

// ResolverMock implements Resolver
type ResolverMock struct {
	t minimock.Tester

	funcResolve          func(m1 coordinates.Module) (rp1 *Request, err error)
	inspectFuncResolve   func(m1 coordinates.Module)
	afterResolveCounter  uint64
	beforeResolveCounter uint64
	ResolveMock          mResolverMockResolve

	funcUseProxy          func(m1 coordinates.Module) (b1 bool, err error)
	inspectFuncUseProxy   func(m1 coordinates.Module)
	afterUseProxyCounter  uint64
	beforeUseProxyCounter uint64
	UseProxyMock          mResolverMockUseProxy
}

// NewResolverMock returns a mock for Resolver
func NewResolverMock(t minimock.Tester) *ResolverMock {
	m := &ResolverMock{t: t}
	if controller, ok := t.(minimock.MockController); ok {
		controller.RegisterMocker(m)
	}

	m.ResolveMock = mResolverMockResolve{mock: m}
	m.ResolveMock.callArgs = []*ResolverMockResolveParams{}

	m.UseProxyMock = mResolverMockUseProxy{mock: m}
	m.UseProxyMock.callArgs = []*ResolverMockUseProxyParams{}

	return m
}

type mResolverMockResolve struct {
	mock               *ResolverMock
	defaultExpectation *ResolverMockResolveExpectation
	expectations       []*ResolverMockResolveExpectation

	callArgs []*ResolverMockResolveParams
	mutex    sync.RWMutex
}

// ResolverMockResolveExpectation specifies expectation struct of the Resolver.Resolve
type ResolverMockResolveExpectation struct {
	mock    *ResolverMock
	params  *ResolverMockResolveParams
	results *ResolverMockResolveResults
	Counter uint64
}

// ResolverMockResolveParams contains parameters of the Resolver.Resolve
type ResolverMockResolveParams struct {
	m1 coordinates.Module
}

// ResolverMockResolveResults contains results of the Resolver.Resolve
type ResolverMockResolveResults struct {
	rp1 *Request
	err error
}

// Expect sets up expected params for Resolver.Resolve
func (mmResolve *mResolverMockResolve) Expect(m1 coordinates.Module) *mResolverMockResolve {
	if mmResolve.mock.funcResolve != nil {
		mmResolve.mock.t.Fatalf("ResolverMock.Resolve mock is already set by Set")
	}

	if mmResolve.defaultExpectation == nil {
		mmResolve.defaultExpectation = &ResolverMockResolveExpectation{}
	}

	mmResolve.defaultExpectation.params = &ResolverMockResolveParams{m1}
	for _, e := range mmResolve.expectations {
		if minimock.Equal(e.params, mmResolve.defaultExpectation.params) {
			mmResolve.mock.t.Fatalf("Expectation set by When has same params: %#v", *mmResolve.defaultExpectation.params)
		}
	}

	return mmResolve
}

// Inspect accepts an inspector function that has same arguments as the Resolver.Resolve
func (mmResolve *mResolverMockResolve) Inspect(f func(m1 coordinates.Module)) *mResolverMockResolve {
	if mmResolve.mock.inspectFuncResolve != nil {
		mmResolve.mock.t.Fatalf("Inspect function is already set for ResolverMock.Resolve")
	}

	mmResolve.mock.inspectFuncResolve = f

	return mmResolve
}

// Return sets up results that will be returned by Resolver.Resolve
func (mmResolve *mResolverMockResolve) Return(rp1 *Request, err error) *ResolverMock {
	if mmResolve.mock.funcResolve != nil {
		mmResolve.mock.t.Fatalf("ResolverMock.Resolve mock is already set by Set")
	}

	if mmResolve.defaultExpectation == nil {
		mmResolve.defaultExpectation = &ResolverMockResolveExpectation{mock: mmResolve.mock}
	}
	mmResolve.defaultExpectation.results = &ResolverMockResolveResults{rp1, err}
	return mmResolve.mock
}

//Set uses given function f to mock the Resolver.Resolve method
func (mmResolve *mResolverMockResolve) Set(f func(m1 coordinates.Module) (rp1 *Request, err error)) *ResolverMock {
	if mmResolve.defaultExpectation != nil {
		mmResolve.mock.t.Fatalf("Default expectation is already set for the Resolver.Resolve method")
	}

	if len(mmResolve.expectations) > 0 {
		mmResolve.mock.t.Fatalf("Some expectations are already set for the Resolver.Resolve method")
	}

	mmResolve.mock.funcResolve = f
	return mmResolve.mock
}

// When sets expectation for the Resolver.Resolve which will trigger the result defined by the following
// Then helper
func (mmResolve *mResolverMockResolve) When(m1 coordinates.Module) *ResolverMockResolveExpectation {
	if mmResolve.mock.funcResolve != nil {
		mmResolve.mock.t.Fatalf("ResolverMock.Resolve mock is already set by Set")
	}

	expectation := &ResolverMockResolveExpectation{
		mock:   mmResolve.mock,
		params: &ResolverMockResolveParams{m1},
	}
	mmResolve.expectations = append(mmResolve.expectations, expectation)
	return expectation
}

// Then sets up Resolver.Resolve return parameters for the expectation previously defined by the When method
func (e *ResolverMockResolveExpectation) Then(rp1 *Request, err error) *ResolverMock {
	e.results = &ResolverMockResolveResults{rp1, err}
	return e.mock
}

// Resolve implements Resolver
func (mmResolve *ResolverMock) Resolve(m1 coordinates.Module) (rp1 *Request, err error) {
	mm_atomic.AddUint64(&mmResolve.beforeResolveCounter, 1)
	defer mm_atomic.AddUint64(&mmResolve.afterResolveCounter, 1)

	if mmResolve.inspectFuncResolve != nil {
		mmResolve.inspectFuncResolve(m1)
	}

	mm_params := &ResolverMockResolveParams{m1}

	// Record call args
	mmResolve.ResolveMock.mutex.Lock()
	mmResolve.ResolveMock.callArgs = append(mmResolve.ResolveMock.callArgs, mm_params)
	mmResolve.ResolveMock.mutex.Unlock()

	for _, e := range mmResolve.ResolveMock.expectations {
		if minimock.Equal(e.params, mm_params) {
			mm_atomic.AddUint64(&e.Counter, 1)
			return e.results.rp1, e.results.err
		}
	}

	if mmResolve.ResolveMock.defaultExpectation != nil {
		mm_atomic.AddUint64(&mmResolve.ResolveMock.defaultExpectation.Counter, 1)
		mm_want := mmResolve.ResolveMock.defaultExpectation.params
		mm_got := ResolverMockResolveParams{m1}
		if mm_want != nil && !minimock.Equal(*mm_want, mm_got) {
			mmResolve.t.Errorf("ResolverMock.Resolve got unexpected parameters, want: %#v, got: %#v%s\n", *mm_want, mm_got, minimock.Diff(*mm_want, mm_got))
		}

		mm_results := mmResolve.ResolveMock.defaultExpectation.results
		if mm_results == nil {
			mmResolve.t.Fatal("No results are set for the ResolverMock.Resolve")
		}
		return (*mm_results).rp1, (*mm_results).err
	}
	if mmResolve.funcResolve != nil {
		return mmResolve.funcResolve(m1)
	}
	mmResolve.t.Fatalf("Unexpected call to ResolverMock.Resolve. %v", m1)
	return
}

// ResolveAfterCounter returns a count of finished ResolverMock.Resolve invocations
func (mmResolve *ResolverMock) ResolveAfterCounter() uint64 {
	return mm_atomic.LoadUint64(&mmResolve.afterResolveCounter)
}

// ResolveBeforeCounter returns a count of ResolverMock.Resolve invocations
func (mmResolve *ResolverMock) ResolveBeforeCounter() uint64 {
	return mm_atomic.LoadUint64(&mmResolve.beforeResolveCounter)
}

// Calls returns a list of arguments used in each call to ResolverMock.Resolve.
// The list is in the same order as the calls were made (i.e. recent calls have a higher index)
func (mmResolve *mResolverMockResolve) Calls() []*ResolverMockResolveParams {
	mmResolve.mutex.RLock()

	argCopy := make([]*ResolverMockResolveParams, len(mmResolve.callArgs))
	copy(argCopy, mmResolve.callArgs)

	mmResolve.mutex.RUnlock()

	return argCopy
}

// MinimockResolveDone returns true if the count of the Resolve invocations corresponds
// the number of defined expectations
func (m *ResolverMock) MinimockResolveDone() bool {
	for _, e := range m.ResolveMock.expectations {
		if mm_atomic.LoadUint64(&e.Counter) < 1 {
			return false
		}
	}

	// if default expectation was set then invocations count should be greater than zero
	if m.ResolveMock.defaultExpectation != nil && mm_atomic.LoadUint64(&m.afterResolveCounter) < 1 {
		return false
	}
	// if func was set then invocations count should be greater than zero
	if m.funcResolve != nil && mm_atomic.LoadUint64(&m.afterResolveCounter) < 1 {
		return false
	}
	return true
}

// MinimockResolveInspect logs each unmet expectation
func (m *ResolverMock) MinimockResolveInspect() {
	for _, e := range m.ResolveMock.expectations {
		if mm_atomic.LoadUint64(&e.Counter) < 1 {
			m.t.Errorf("Expected call to ResolverMock.Resolve with params: %#v", *e.params)
		}
	}

	// if default expectation was set then invocations count should be greater than zero
	if m.ResolveMock.defaultExpectation != nil && mm_atomic.LoadUint64(&m.afterResolveCounter) < 1 {
		if m.ResolveMock.defaultExpectation.params == nil {
			m.t.Error("Expected call to ResolverMock.Resolve")
		} else {
			m.t.Errorf("Expected call to ResolverMock.Resolve with params: %#v", *m.ResolveMock.defaultExpectation.params)
		}
	}
	// if func was set then invocations count should be greater than zero
	if m.funcResolve != nil && mm_atomic.LoadUint64(&m.afterResolveCounter) < 1 {
		m.t.Error("Expected call to ResolverMock.Resolve")
	}
}

type mResolverMockUseProxy struct {
	mock               *ResolverMock
	defaultExpectation *ResolverMockUseProxyExpectation
	expectations       []*ResolverMockUseProxyExpectation

	callArgs []*ResolverMockUseProxyParams
	mutex    sync.RWMutex
}

// ResolverMockUseProxyExpectation specifies expectation struct of the Resolver.UseProxy
type ResolverMockUseProxyExpectation struct {
	mock    *ResolverMock
	params  *ResolverMockUseProxyParams
	results *ResolverMockUseProxyResults
	Counter uint64
}

// ResolverMockUseProxyParams contains parameters of the Resolver.UseProxy
type ResolverMockUseProxyParams struct {
	m1 coordinates.Module
}

// ResolverMockUseProxyResults contains results of the Resolver.UseProxy
type ResolverMockUseProxyResults struct {
	b1  bool
	err error
}

// Expect sets up expected params for Resolver.UseProxy
func (mmUseProxy *mResolverMockUseProxy) Expect(m1 coordinates.Module) *mResolverMockUseProxy {
	if mmUseProxy.mock.funcUseProxy != nil {
		mmUseProxy.mock.t.Fatalf("ResolverMock.UseProxy mock is already set by Set")
	}

	if mmUseProxy.defaultExpectation == nil {
		mmUseProxy.defaultExpectation = &ResolverMockUseProxyExpectation{}
	}

	mmUseProxy.defaultExpectation.params = &ResolverMockUseProxyParams{m1}
	for _, e := range mmUseProxy.expectations {
		if minimock.Equal(e.params, mmUseProxy.defaultExpectation.params) {
			mmUseProxy.mock.t.Fatalf("Expectation set by When has same params: %#v", *mmUseProxy.defaultExpectation.params)
		}
	}

	return mmUseProxy
}

// Inspect accepts an inspector function that has same arguments as the Resolver.UseProxy
func (mmUseProxy *mResolverMockUseProxy) Inspect(f func(m1 coordinates.Module)) *mResolverMockUseProxy {
	if mmUseProxy.mock.inspectFuncUseProxy != nil {
		mmUseProxy.mock.t.Fatalf("Inspect function is already set for ResolverMock.UseProxy")
	}

	mmUseProxy.mock.inspectFuncUseProxy = f

	return mmUseProxy
}

// Return sets up results that will be returned by Resolver.UseProxy
func (mmUseProxy *mResolverMockUseProxy) Return(b1 bool, err error) *ResolverMock {
	if mmUseProxy.mock.funcUseProxy != nil {
		mmUseProxy.mock.t.Fatalf("ResolverMock.UseProxy mock is already set by Set")
	}

	if mmUseProxy.defaultExpectation == nil {
		mmUseProxy.defaultExpectation = &ResolverMockUseProxyExpectation{mock: mmUseProxy.mock}
	}
	mmUseProxy.defaultExpectation.results = &ResolverMockUseProxyResults{b1, err}
	return mmUseProxy.mock
}

//Set uses given function f to mock the Resolver.UseProxy method
func (mmUseProxy *mResolverMockUseProxy) Set(f func(m1 coordinates.Module) (b1 bool, err error)) *ResolverMock {
	if mmUseProxy.defaultExpectation != nil {
		mmUseProxy.mock.t.Fatalf("Default expectation is already set for the Resolver.UseProxy method")
	}

	if len(mmUseProxy.expectations) > 0 {
		mmUseProxy.mock.t.Fatalf("Some expectations are already set for the Resolver.UseProxy method")
	}

	mmUseProxy.mock.funcUseProxy = f
	return mmUseProxy.mock
}

// When sets expectation for the Resolver.UseProxy which will trigger the result defined by the following
// Then helper
func (mmUseProxy *mResolverMockUseProxy) When(m1 coordinates.Module) *ResolverMockUseProxyExpectation {
	if mmUseProxy.mock.funcUseProxy != nil {
		mmUseProxy.mock.t.Fatalf("ResolverMock.UseProxy mock is already set by Set")
	}

	expectation := &ResolverMockUseProxyExpectation{
		mock:   mmUseProxy.mock,
		params: &ResolverMockUseProxyParams{m1},
	}
	mmUseProxy.expectations = append(mmUseProxy.expectations, expectation)
	return expectation
}

// Then sets up Resolver.UseProxy return parameters for the expectation previously defined by the When method
func (e *ResolverMockUseProxyExpectation) Then(b1 bool, err error) *ResolverMock {
	e.results = &ResolverMockUseProxyResults{b1, err}
	return e.mock
}

// UseProxy implements Resolver
func (mmUseProxy *ResolverMock) UseProxy(m1 coordinates.Module) (b1 bool, err error) {
	mm_atomic.AddUint64(&mmUseProxy.beforeUseProxyCounter, 1)
	defer mm_atomic.AddUint64(&mmUseProxy.afterUseProxyCounter, 1)

	if mmUseProxy.inspectFuncUseProxy != nil {
		mmUseProxy.inspectFuncUseProxy(m1)
	}

	mm_params := &ResolverMockUseProxyParams{m1}

	// Record call args
	mmUseProxy.UseProxyMock.mutex.Lock()
	mmUseProxy.UseProxyMock.callArgs = append(mmUseProxy.UseProxyMock.callArgs, mm_params)
	mmUseProxy.UseProxyMock.mutex.Unlock()

	for _, e := range mmUseProxy.UseProxyMock.expectations {
		if minimock.Equal(e.params, mm_params) {
			mm_atomic.AddUint64(&e.Counter, 1)
			return e.results.b1, e.results.err
		}
	}

	if mmUseProxy.UseProxyMock.defaultExpectation != nil {
		mm_atomic.AddUint64(&mmUseProxy.UseProxyMock.defaultExpectation.Counter, 1)
		mm_want := mmUseProxy.UseProxyMock.defaultExpectation.params
		mm_got := ResolverMockUseProxyParams{m1}
		if mm_want != nil && !minimock.Equal(*mm_want, mm_got) {
			mmUseProxy.t.Errorf("ResolverMock.UseProxy got unexpected parameters, want: %#v, got: %#v%s\n", *mm_want, mm_got, minimock.Diff(*mm_want, mm_got))
		}

		mm_results := mmUseProxy.UseProxyMock.defaultExpectation.results
		if mm_results == nil {
			mmUseProxy.t.Fatal("No results are set for the ResolverMock.UseProxy")
		}
		return (*mm_results).b1, (*mm_results).err
	}
	if mmUseProxy.funcUseProxy != nil {
		return mmUseProxy.funcUseProxy(m1)
	}
	mmUseProxy.t.Fatalf("Unexpected call to ResolverMock.UseProxy. %v", m1)
	return
}

// UseProxyAfterCounter returns a count of finished ResolverMock.UseProxy invocations
func (mmUseProxy *ResolverMock) UseProxyAfterCounter() uint64 {
	return mm_atomic.LoadUint64(&mmUseProxy.afterUseProxyCounter)
}

// UseProxyBeforeCounter returns a count of ResolverMock.UseProxy invocations
func (mmUseProxy *ResolverMock) UseProxyBeforeCounter() uint64 {
	return mm_atomic.LoadUint64(&mmUseProxy.beforeUseProxyCounter)
}

// Calls returns a list of arguments used in each call to ResolverMock.UseProxy.
// The list is in the same order as the calls were made (i.e. recent calls have a higher index)
func (mmUseProxy *mResolverMockUseProxy) Calls() []*ResolverMockUseProxyParams {
	mmUseProxy.mutex.RLock()

	argCopy := make([]*ResolverMockUseProxyParams, len(mmUseProxy.callArgs))
	copy(argCopy, mmUseProxy.callArgs)

	mmUseProxy.mutex.RUnlock()

	return argCopy
}

// MinimockUseProxyDone returns true if the count of the UseProxy invocations corresponds
// the number of defined expectations
func (m *ResolverMock) MinimockUseProxyDone() bool {
	for _, e := range m.UseProxyMock.expectations {
		if mm_atomic.LoadUint64(&e.Counter) < 1 {
			return false
		}
	}

	// if default expectation was set then invocations count should be greater than zero
	if m.UseProxyMock.defaultExpectation != nil && mm_atomic.LoadUint64(&m.afterUseProxyCounter) < 1 {
		return false
	}
	// if func was set then invocations count should be greater than zero
	if m.funcUseProxy != nil && mm_atomic.LoadUint64(&m.afterUseProxyCounter) < 1 {
		return false
	}
	return true
}

// MinimockUseProxyInspect logs each unmet expectation
func (m *ResolverMock) MinimockUseProxyInspect() {
	for _, e := range m.UseProxyMock.expectations {
		if mm_atomic.LoadUint64(&e.Counter) < 1 {
			m.t.Errorf("Expected call to ResolverMock.UseProxy with params: %#v", *e.params)
		}
	}

	// if default expectation was set then invocations count should be greater than zero
	if m.UseProxyMock.defaultExpectation != nil && mm_atomic.LoadUint64(&m.afterUseProxyCounter) < 1 {
		if m.UseProxyMock.defaultExpectation.params == nil {
			m.t.Error("Expected call to ResolverMock.UseProxy")
		} else {
			m.t.Errorf("Expected call to ResolverMock.UseProxy with params: %#v", *m.UseProxyMock.defaultExpectation.params)
		}
	}
	// if func was set then invocations count should be greater than zero
	if m.funcUseProxy != nil && mm_atomic.LoadUint64(&m.afterUseProxyCounter) < 1 {
		m.t.Error("Expected call to ResolverMock.UseProxy")
	}
}

// MinimockFinish checks that all mocked methods have been called the expected number of times
func (m *ResolverMock) MinimockFinish() {
	if !m.minimockDone() {
		m.MinimockResolveInspect()

		m.MinimockUseProxyInspect()
		m.t.FailNow()
	}
}

// MinimockWait waits for all mocked methods to be called the expected number of times
func (m *ResolverMock) MinimockWait(timeout mm_time.Duration) {
	timeoutCh := mm_time.After(timeout)
	for {
		if m.minimockDone() {
			return
		}
		select {
		case <-timeoutCh:
			m.MinimockFinish()
			return
		case <-mm_time.After(10 * mm_time.Millisecond):
		}
	}
}

func (m *ResolverMock) minimockDone() bool {
	done := true
	return done &&
		m.MinimockResolveDone() &&
		m.MinimockUseProxyDone()
}