From 80939647c54f7ed3af83ee217ab3efadbdcaee75 Mon Sep 17 00:00:00 2001
From: Yordan Kinkov <yordan.kinkov@vereign.com>
Date: Mon, 14 Aug 2023 11:35:15 +0300
Subject: [PATCH] Change vault mock struct
---
.../service/signer/signerfakes/fake_vault.go | 65 +++++++++++++++++++
1 file changed, 65 insertions(+)
diff --git a/internal/service/signer/signerfakes/fake_vault.go b/internal/service/signer/signerfakes/fake_vault.go
index 071c76e..ea191a9 100644
--- a/internal/service/signer/signerfakes/fake_vault.go
+++ b/internal/service/signer/signerfakes/fake_vault.go
@@ -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()
--
GitLab