test: add test for new method

Layr-Labs · Feb 12, 2025 · 98afcd2 · 98afcd2
1 parent 99dcc89
commit 98afcd2
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/chainio/clients/avsregistry/reader_test.go b/chainio/clients/avsregistry/reader_test.go
@@ -417,6 +417,12 @@ func TestReaderMethods(t *testing.T) {
 		require.Equal(t, []common.Address{strategy}, retrievedStrat)
 	})
 
+	t.Run("Get operator restaked strategies", func(t *testing.T) {
+		retrievedStrat, err := chainReader.GetOperatorRestakedStrategies(&bind.CallOpts{}, operatorAddress)
+		require.NoError(t, err)
+		require.Equal(t, []common.Address{strategy}, retrievedStrat)
+	})
+
 	t.Run("Get restakeable strategies", func(t *testing.T) {
 		retrievedStrat, err := chainReader.GetRestakeableStrategies(&bind.CallOpts{})
 		require.NoError(t, err)