fix: handle 0x prefix

primev · Jul 8, 2024 · 8bde6c8 · 8bde6c8
1 parent f8e0ac6
commit 8bde6c8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/p2p/pkg/rpc/provider/service.go b/p2p/pkg/rpc/provider/service.go
@@ -205,6 +205,7 @@ func (s *Service) RegisterStake(
 	}
 	opts.Value = amount
 
+	stake.BlsPublicKey = strings.TrimPrefix(stake.BlsPublicKey, "0x")
 	blsPubkeyBytes, err := hex.DecodeString(stake.BlsPublicKey)
 	if err != nil {
 		return nil, status.Errorf(codes.InvalidArgument, "decoding bls public key: %v", err)

diff --git a/p2p/pkg/rpc/provider/service_test.go b/p2p/pkg/rpc/provider/service_test.go
@@ -184,6 +184,7 @@ func TestStakeHandling(t *testing.T) {
 				if stake.Amount != tc.amount {
 					t.Fatalf("expected amount to be %v, got %v", tc.amount, stake.Amount)
 				}
+				tc.blsPublicKey = strings.TrimPrefix(tc.blsPublicKey, "0x")
 				if stake.BlsPublicKey != tc.blsPublicKey {
 					t.Fatalf("expected bls_public_key to be %v, got %v", tc.blsPublicKey, stake.BlsPublicKey)
 				}