refactor(tests): add missing build constraints and update mock for ha…

…ndler-cache
telekom · Feb 20, 2025 · ecff49f · ecff49f
1 parent b9125a2
commit ecff49f
Show file tree

Hide file tree

Showing 6 changed files with 43 additions and 26 deletions.
diff --git a/internal/test/handler_cache_mock.go b/internal/test/handler_cache_mock.go
@@ -2,6 +2,8 @@
 //
 // SPDX-License-Identifier: Apache-2.0
 
+//go:build testing
+
 package test
 
 import (
@@ -15,67 +17,72 @@ type MockHandlerCache struct {
 	mock.Mock
 }
 
-func (f *MockHandlerCache) Get(ctx context.Context, key interface{}) (interface{}, error) {
-	args := f.Called(ctx, key)
+func (c *MockHandlerCache) Get(ctx context.Context, key interface{}) (interface{}, error) {
+	args := c.Called(ctx, key)
 	return args.Get(0), args.Error(1)
 }
 
-func (f *MockHandlerCache) Set(ctx context.Context, key interface{}, value interface{}) error {
-	args := f.Called(ctx, key, value)
+func (c *MockHandlerCache) Set(ctx context.Context, key interface{}, value interface{}) error {
+	args := c.Called(ctx, key, value)
 	return args.Error(0)
 }
 
-func (f *MockHandlerCache) TryLockWithTimeout(ctx context.Context, key interface{}, timeout time.Duration) (bool, error) {
-	args := f.Called(ctx, key, timeout)
+func (c *MockHandlerCache) TryLockWithTimeout(ctx context.Context, key interface{}, timeout time.Duration) (bool, error) {
+	args := c.Called(ctx, key, timeout)
+	return args.Bool(0), args.Error(1)
+}
+
+func (c *MockHandlerCache) TryLockWithLease(ctx context.Context, key interface{}, duration time.Duration) (bool, error) {
+	args := c.Called(ctx, key, duration)
 	return args.Bool(0), args.Error(1)
 }
 
-func (d *MockHandlerCache) TryLockWithLeaseAndTimeout(ctx context.Context, key interface{}, lease time.Duration, timeout time.Duration) (bool, error) {
-	args := d.Called(ctx, key, timeout)
+func (c *MockHandlerCache) TryLockWithLeaseAndTimeout(ctx context.Context, key interface{}, lease time.Duration, timeout time.Duration) (bool, error) {
+	args := c.Called(ctx, key, timeout)
 	return args.Bool(0), args.Error(1)
 }
 
-func (f *MockHandlerCache) GetEntrySet(ctx context.Context) ([]types.Entry, error) {
-	args := f.Called(ctx)
+func (c *MockHandlerCache) GetEntrySet(ctx context.Context) ([]types.Entry, error) {
+	args := c.Called(ctx)
 	return args.Get(0).([]types.Entry), args.Error(1)
 }
 
-func (f *MockHandlerCache) NewLockContext(ctx context.Context) context.Context {
-	args := f.Called(ctx)
+func (c *MockHandlerCache) NewLockContext(ctx context.Context) context.Context {
+	args := c.Called(ctx)
 	return args.Get(0).(context.Context)
 }
 
-func (f *MockHandlerCache) Delete(ctx context.Context, key interface{}) error {
-	args := f.Called(ctx, key)
+func (c *MockHandlerCache) Delete(ctx context.Context, key interface{}) error {
+	args := c.Called(ctx, key)
 	return args.Error(0)
 }
 
-func (f *MockHandlerCache) Unlock(ctx context.Context, key interface{}) error {
-	args := f.Called(ctx, key)
+func (c *MockHandlerCache) Unlock(ctx context.Context, key interface{}) error {
+	args := c.Called(ctx, key)
 	return args.Error(0)
 }
 
-func (f *MockHandlerCache) IsLocked(ctx context.Context, key interface{}) (bool, error) {
-	args := f.Called(ctx, key)
+func (c *MockHandlerCache) IsLocked(ctx context.Context, key interface{}) (bool, error) {
+	args := c.Called(ctx, key)
 	return args.Bool(0), args.Error(1)
 }
 
-func (f *MockHandlerCache) ForceUnlock(ctx context.Context, key interface{}) error {
-	args := f.Called(ctx, key)
+func (c *MockHandlerCache) ForceUnlock(ctx context.Context, key interface{}) error {
+	args := c.Called(ctx, key)
 	return args.Error(0)
 }
 
-func (f *MockHandlerCache) ContainsKey(ctx context.Context, key interface{}) (bool, error) {
-	args := f.Called(ctx, key)
+func (c *MockHandlerCache) ContainsKey(ctx context.Context, key interface{}) (bool, error) {
+	args := c.Called(ctx, key)
 	return args.Bool(0), args.Error(1)
 }
 
-func (f *MockHandlerCache) Clear(ctx context.Context) error {
-	args := f.Called(ctx)
+func (c *MockHandlerCache) Clear(ctx context.Context) error {
+	args := c.Called(ctx)
 	return args.Error(0)
 }
 
-func (f *MockHandlerCache) Lock(ctx context.Context, key interface{}) error {
-	args := f.Called(ctx, key)
+func (c *MockHandlerCache) Lock(ctx context.Context, key interface{}) error {
+	args := c.Called(ctx, key)
 	return args.Error(0)
 }
diff --git a/internal/test/health_check_mock.go b/internal/test/health_check_mock.go
@@ -2,6 +2,8 @@
 //
 // SPDX-License-Identifier: Apache-2.0
 
+//go:build testing
+
 package test
 
 import (

diff --git a/internal/test/kafkahandler_mock.go b/internal/test/kafkahandler_mock.go
@@ -2,6 +2,8 @@
 //
 // SPDX-License-Identifier: Apache-2.0
 
+//go:build testing
+
 package test
 
 import (

diff --git a/internal/test/mongohandler_mock.go b/internal/test/mongohandler_mock.go
@@ -2,6 +2,8 @@
 //
 // SPDX-License-Identifier: Apache-2.0
 
+//go:build testing
+
 package test
 
 import (

diff --git a/internal/test/republishing_mock.go b/internal/test/republishing_mock.go
@@ -2,6 +2,8 @@
 //
 // SPDX-License-Identifier: Apache-2.0
 
+//go:build testing
+
 package test
 
 import (

diff --git a/internal/test/waitinghandler_mock.go b/internal/test/waitinghandler_mock.go
@@ -2,6 +2,8 @@
 //
 // SPDX-License-Identifier: Apache-2.0
 
+//go:build testing
+
 package test
 
 import "github.com/stretchr/testify/mock"