fix: lint for mocks

Layr-Labs · Jan 2, 2025 · 8b4969c · 8b4969c
1 parent 1af099c
commit 8b4969c
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/core/eth/state.go b/core/eth/state.go
@@ -22,8 +22,6 @@ type ChainState struct {
 	socketPrevBlockNumber uint32
 }
 
-var defaultQuorumIDs = []core.QuorumID{core.QuorumID(0), core.QuorumID(1)}
-
 func NewChainState(tx core.Reader, client common.EthClient) *ChainState {
 	return &ChainState{
 		Client: client,

diff --git a/core/mock/writer.go b/core/mock/writer.go
@@ -112,6 +112,18 @@ func (t *MockWriter) StakeRegistry(ctx context.Context) (gethcommon.Address, err
 	return result.(gethcommon.Address), args.Error(1)
 }
 
+func (t *MockWriter) SocketRegistry(ctx context.Context) (gethcommon.Address, error) {
+	args := t.Called()
+	result := args.Get(0)
+	return result.(gethcommon.Address), args.Error(1)
+}
+
+func (t *MockWriter) RegistryCoordinator(ctx context.Context) (gethcommon.Address, error) {
+	args := t.Called()
+	result := args.Get(0)
+	return result.(gethcommon.Address), args.Error(1)
+}
+
 func (t *MockWriter) OperatorIDToAddress(ctx context.Context, operatorId core.OperatorID) (gethcommon.Address, error) {
 	args := t.Called()
 	result := args.Get(0)