Add tests

NethermindEth · Jan 24, 2025 · c99a65c · c99a65c
1 parent f7494e1
commit c99a65c
Show file tree

Hide file tree

Showing 2 changed files with 102 additions and 0 deletions.
diff --git a/rpc/transaction_test.go b/rpc/transaction_test.go
@@ -1365,3 +1365,54 @@ func TestTransactionStatus(t *testing.T) {
 		})
 	}
 }
+
+func TestResourceMarshalText(t *testing.T) {
+	tests := []struct {
+		resource rpc.Resource
+		text     string
+		hasError bool
+	}{
+		{rpc.ResourceL1Gas, "l1_gas", false},
+		{rpc.ResourceL2Gas, "l2_gas", false},
+		{rpc.ResourceL1DataGas, "l1_data_gas", false},
+		{rpc.Resource(999), "error", true},
+	}
+
+	for _, test := range tests {
+		t.Run(test.text, func(t *testing.T) {
+			b, err := test.resource.MarshalText()
+			if test.hasError {
+				require.Error(t, err)
+				return
+			}
+			require.NoError(t, err)
+			require.Equal(t, test.text, string(b))
+		})
+	}
+}
+
+func TestResourceUnmarshalText(t *testing.T) {
+	tests := []struct {
+		resource rpc.Resource
+		text     string
+		hasError bool
+	}{
+		{rpc.ResourceL1Gas, `"l1_gas"`, false},
+		{rpc.ResourceL2Gas, `"l2_gas"`, false},
+		{rpc.ResourceL1DataGas, `"l1_data_gas"`, false},
+		{rpc.Resource(999), "error", true},
+	}
+
+	for _, test := range tests {
+		t.Run(test.text, func(t *testing.T) {
+			var r rpc.Resource
+			err := r.UnmarshalText([]byte(test.text))
+			if test.hasError {
+				require.Error(t, err)
+				return
+			}
+			require.NoError(t, err)
+			require.Equal(t, test.resource, r)
+		})
+	}
+}
diff --git a/starknet/transaction_test.go b/starknet/transaction_test.go
@@ -43,3 +43,54 @@ func TestUnmarshalFinalityStatus(t *testing.T) {
 
 	require.ErrorContains(t, json.Unmarshal([]byte(`"ABC"`), fs), "unknown FinalityStatus")
 }
+
+func TestResourceMarshalText(t *testing.T) {
+	tests := []struct {
+		resource starknet.Resource
+		text     string
+		hasError bool
+	}{
+		{starknet.ResourceL1Gas, "L1_GAS", false},
+		{starknet.ResourceL2Gas, "L2_GAS", false},
+		{starknet.ResourceL1DataGas, "L1_DATA_GAS", false},
+		{starknet.Resource(999), "error", true},
+	}
+
+	for _, test := range tests {
+		t.Run(test.text, func(t *testing.T) {
+			b, err := test.resource.MarshalText()
+			if test.hasError {
+				require.Error(t, err)
+				return
+			}
+			require.NoError(t, err)
+			require.Equal(t, test.text, string(b))
+		})
+	}
+}
+
+func TestResourceUnmarshalText(t *testing.T) {
+	tests := []struct {
+		resource starknet.Resource
+		text     string
+		hasError bool
+	}{
+		{starknet.ResourceL1Gas, "L1_GAS", false},
+		{starknet.ResourceL2Gas, "L2_GAS", false},
+		{starknet.ResourceL1DataGas, "L1_DATA_GAS", false},
+		{starknet.Resource(999), "error", true},
+	}
+
+	for _, test := range tests {
+		t.Run(test.text, func(t *testing.T) {
+			var r starknet.Resource
+			err := r.UnmarshalText([]byte(test.text))
+			if test.hasError {
+				require.Error(t, err)
+				return
+			}
+			require.NoError(t, err)
+			require.Equal(t, test.resource, r)
+		})
+	}
+}