Skip to content
Snippets Groups Projects
Commit 80939647 authored by Yordan Kinkov's avatar Yordan Kinkov
Browse files

Change vault mock struct

parent 5c902688
Branches
No related tags found
No related merge requests found
......@@ -9,6 +9,16 @@ import (
)
type FakeVault struct {
AlgStub func() string
algMutex sync.RWMutex
algArgsForCall []struct {
}
algReturns struct {
result1 string
}
algReturnsOnCall map[int]struct {
result1 string
}
KeyStub func(context.Context, string, string) (*signer.VaultKey, error)
keyMutex sync.RWMutex
keyArgsForCall []struct {
......@@ -94,6 +104,59 @@ type FakeVault struct {
invocationsMutex sync.RWMutex
}
func (fake *FakeVault) Alg() string {
fake.algMutex.Lock()
ret, specificReturn := fake.algReturnsOnCall[len(fake.algArgsForCall)]
fake.algArgsForCall = append(fake.algArgsForCall, struct {
}{})
stub := fake.AlgStub
fakeReturns := fake.algReturns
fake.recordInvocation("Alg", []interface{}{})
fake.algMutex.Unlock()
if stub != nil {
return stub()
}
if specificReturn {
return ret.result1
}
return fakeReturns.result1
}
func (fake *FakeVault) AlgCallCount() int {
fake.algMutex.RLock()
defer fake.algMutex.RUnlock()
return len(fake.algArgsForCall)
}
func (fake *FakeVault) AlgCalls(stub func() string) {
fake.algMutex.Lock()
defer fake.algMutex.Unlock()
fake.AlgStub = stub
}
func (fake *FakeVault) AlgReturns(result1 string) {
fake.algMutex.Lock()
defer fake.algMutex.Unlock()
fake.AlgStub = nil
fake.algReturns = struct {
result1 string
}{result1}
}
func (fake *FakeVault) AlgReturnsOnCall(i int, result1 string) {
fake.algMutex.Lock()
defer fake.algMutex.Unlock()
fake.AlgStub = nil
if fake.algReturnsOnCall == nil {
fake.algReturnsOnCall = make(map[int]struct {
result1 string
})
}
fake.algReturnsOnCall[i] = struct {
result1 string
}{result1}
}
func (fake *FakeVault) Key(arg1 context.Context, arg2 string, arg3 string) (*signer.VaultKey, error) {
fake.keyMutex.Lock()
ret, specificReturn := fake.keyReturnsOnCall[len(fake.keyArgsForCall)]
......@@ -488,6 +551,8 @@ func (fake *FakeVault) WithKeyReturnsOnCall(i int, result1 signer.Vault) {
func (fake *FakeVault) Invocations() map[string][][]interface{} {
fake.invocationsMutex.RLock()
defer fake.invocationsMutex.RUnlock()
fake.algMutex.RLock()
defer fake.algMutex.RUnlock()
fake.keyMutex.RLock()
defer fake.keyMutex.RUnlock()
fake.keysMutex.RLock()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment