Add tests

rpc/transaction_test.go

@@ -1358,3 +1358,88 @@ func TestTransactionStatus(t *testing.T) {
 		})
 	}
 }
+
+func TestResourceMarshalText(t *testing.T) {
+	tests := []struct {
+		name     string
+		resource rpc.Resource
+		want     []byte
+		err      bool
+	}{
+		{
+			name:     "l1 gas",
+			resource: rpc.ResourceL1Gas,
+			want:     []byte("l1_gas"),
+		},
+		{
+			name:     "l2 gas",
+			resource: rpc.ResourceL2Gas,
+			want:     []byte("l2_gas"),
+		},
+		{
+			name:     "l1 data gas",
+			resource: rpc.ResourceL1DataGas,
+			want:     []byte("l1_data_gas"),
+		},
+		{
+			name:     "error",
+			resource: rpc.Resource(0),
+			err:      true,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got, err := tt.resource.MarshalText()
+			if tt.err {
+				require.Error(t, err)
+				require.Nil(t, got)
+				return
+			}
+			require.NoError(t, err)
+			require.Equal(t, tt.want, got)
+		})
+	}
+}
+
+func TestResourceUnmarshalJSON(t *testing.T) {
+	tests := []struct {
+		name string
+		data []byte
+		want rpc.Resource
+		err  bool
+	}{
+		{
+			name: "l1 gas",
+			data: []byte("l1_gas"),
+			want: rpc.ResourceL1Gas,
+		},
+		{
+			name: "l2 gas",
+			data: []byte("l2_gas"),
+			want: rpc.ResourceL2Gas,
+		},
+		{
+			name: "l1 data gas",
+			data: []byte("l1_data_gas"),
+			want: rpc.ResourceL1DataGas,
+		},
+		{
+			name: "error",
+			data: []byte("unknown"),
+			err:  true,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			var got rpc.Resource
+			err := got.UnmarshalJSON(tt.data)
+			if tt.err {
+				require.Error(t, err)
+				require.Zero(t, got)
+				return
+			}
+			require.NoError(t, err)
+			require.Equal(t, tt.want, got)
+		})
+	}
+}