From c6ac38edb3da749442796959b9aca9837fc0782a Mon Sep 17 00:00:00 2001
From: Sergey Cherepanov <cheggaaa@gmail.com>
Date: Wed, 3 Apr 2024 12:36:18 +0200
Subject: [PATCH] anymock helper

---
 testutil/anymock/anymock.go | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)
 create mode 100644 testutil/anymock/anymock.go

diff --git a/testutil/anymock/anymock.go b/testutil/anymock/anymock.go
new file mode 100644
index 00000000..0a6f7922
--- /dev/null
+++ b/testutil/anymock/anymock.go
@@ -0,0 +1,22 @@
+package anymock
+
+import "go.uber.org/mock/gomock"
+
+type MockComp interface {
+	Name() *gomock.Call
+	Init(x any) *gomock.Call
+}
+
+type MockCompRunnable interface {
+	Run(x any) *gomock.Call
+	Close(x any) *gomock.Call
+}
+
+func ExpectComp(c MockComp, name string) {
+	c.Name().Return(name).AnyTimes()
+	c.Init(gomock.Any()).AnyTimes()
+	if cr, ok := c.(MockCompRunnable); ok {
+		cr.Run(gomock.Any()).AnyTimes()
+		cr.Close(gomock.Any()).AnyTimes()
+	}
+}