diff --git a/antireplay/stable_bloom_filter.go b/antireplay/stable_bloom_filter.go index 5ab15f3..6eef3f5 100644 --- a/antireplay/stable_bloom_filter.go +++ b/antireplay/stable_bloom_filter.go @@ -16,7 +16,7 @@ func (s *stableBloomFilter) SeenBefore(digest []byte) bool { func NewStableBloomFilter(byteSize uint, errorRate float64) mtglib.AntiReplayCache { sf := boom.NewDefaultStableBloomFilter(byteSize*8, errorRate) // nolint: gomnd - sf.SetHash(xxhash.New64()) + sf.SetHash(xxhash.New64()) return &stableBloomFilter{ filter: *sf, diff --git a/testlib/mtglib_antireplay_cache_mock.go b/testlib/mtglib_antireplay_cache_mock.go new file mode 100644 index 0000000..20ac993 --- /dev/null +++ b/testlib/mtglib_antireplay_cache_mock.go @@ -0,0 +1,11 @@ +package testlib + +import "github.com/stretchr/testify/mock" + +type MtglibAntiReplayCacheMock struct { + mock.Mock +} + +func (m *MtglibAntiReplayCacheMock) SeenBefore(data []byte) bool { + return m.Called(data).Bool(0) +} diff --git a/testlib/mtglib_network_mock.go b/testlib/mtglib_network_mock.go index 5a8192f..23fa8ba 100644 --- a/testlib/mtglib_network_mock.go +++ b/testlib/mtglib_network_mock.go @@ -13,14 +13,14 @@ type MtglibNetworkMock struct { mock.Mock } -func (n *MtglibNetworkMock) Dial(network, address string) (net.Conn, error) { - args := n.Called(network, address) +func (m *MtglibNetworkMock) Dial(network, address string) (net.Conn, error) { + args := m.Called(network, address) return args.Get(0).(net.Conn), args.Error(1) } -func (n *MtglibNetworkMock) DialContext(ctx context.Context, network, address string) (net.Conn, error) { - args := n.Called(ctx, network, address) +func (m *MtglibNetworkMock) DialContext(ctx context.Context, network, address string) (net.Conn, error) { + args := m.Called(ctx, network, address) return args.Get(0).(net.Conn), args.Error(1) }