rebase

Layr-Labs · Apr 9, 2024 · 59f5af6 · 59f5af6
1 parent 5c748d0
commit 59f5af6
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 5 deletions.
diff --git a/core/mock/state.go b/core/mock/state.go
@@ -194,7 +194,7 @@ func (d *ChainDataMock) GetTotalOperatorStateWithQuorums(ctx context.Context, bl
 		for opID := range operatorsByID {
 			if aggPubKeys[quorumID] == nil {
 				key := privateOperators[opID].KeyPair.GetPubKeyG1()
-				aggPubKeys[quorumID] = key.Deserialize(key.Serialize())
+				aggPubKeys[quorumID] = key.Clone()
 			} else {
 				aggPubKeys[quorumID].Add(privateOperators[opID].KeyPair.GetPubKeyG1())
 			}

diff --git a/core/thegraph/state_test.go b/core/thegraph/state_test.go
@@ -28,7 +28,11 @@ func (m mockGraphQLQuerier) Query(ctx context.Context, q any, variables map[stri
 func TestIndexedChainState_GetIndexedOperatorState(t *testing.T) {
 	logger := logging.NewNoopLogger()
 
-	chainState, _ := mock.MakeChainDataMock(1)
+	chainState, _ := mock.MakeChainDataMock(map[uint8]int{
+		0: 1,
+		1: 1,
+		2: 1,
+	})
 	chainState.On("GetCurrentBlockNumber").Return(uint(1), nil)
 
 	state, err := chainState.GetOperatorState(context.Background(), 1, quorums)
@@ -91,7 +95,11 @@ func TestIndexedChainState_GetIndexedOperatorState(t *testing.T) {
 func TestIndexedChainState_GetIndexedOperatorStateMissingOperator(t *testing.T) {
 	logger := logging.NewNoopLogger()
 
-	chainState, _ := mock.MakeChainDataMock(2)
+	chainState, _ := mock.MakeChainDataMock(map[uint8]int{
+		0: 2,
+		1: 2,
+		2: 2,
+	})
 	chainState.On("GetCurrentBlockNumber").Return(uint(1), nil)
 
 	state, err := chainState.GetOperatorState(context.Background(), 1, quorums)
@@ -154,7 +162,11 @@ func TestIndexedChainState_GetIndexedOperatorStateMissingOperator(t *testing.T)
 func TestIndexedChainState_GetIndexedOperatorStateExtraOperator(t *testing.T) {
 	logger := logging.NewNoopLogger()
 
-	chainState, _ := mock.MakeChainDataMock(1)
+	chainState, _ := mock.MakeChainDataMock(map[uint8]int{
+		0: 1,
+		1: 1,
+		2: 1,
+	})
 	chainState.On("GetCurrentBlockNumber").Return(uint(1), nil)
 
 	state, err := chainState.GetOperatorState(context.Background(), 1, quorums)
@@ -233,7 +245,11 @@ func TestIndexedChainState_GetIndexedOperatorStateExtraOperator(t *testing.T) {
 func TestIndexedChainState_GetIndexedOperatorInfoByOperatorId(t *testing.T) {
 	logger := logging.NewNoopLogger()
 
-	chainState, _ := mock.MakeChainDataMock(1)
+	chainState, _ := mock.MakeChainDataMock(map[uint8]int{
+		0: 1,
+		1: 1,
+		2: 1,
+	})
 	chainState.On("GetCurrentBlockNumber").Return(uint(1), nil)
 
 	state, err := chainState.GetOperatorState(context.Background(), 1, quorums)