Fix: test added

baking-bad · Feb 16, 2021 · 188b5c6 · 188b5c6
1 parent 170979e
commit 188b5c6
Showing 1 changed file with 13 additions and 5 deletions.
diff --git a/internal/views/michelson_storage_view_test.go b/internal/views/michelson_storage_view_test.go
@@ -24,12 +24,20 @@ func TestMichelsonStorageView_GetCode(t *testing.T) {
 		{
 			name: "test 1",
 			fields: fields{
-				Parameter:  []byte(`{"prim": "unit"}`),
-				Code:       []byte(`{"prim": "unit"}`),
-				ReturnType: []byte(`{"prim": "unit"}`),
+				Parameter:  []byte(`{"prim":"unit"}`),
+				Code:       []byte(`{"prim":"unit"}`),
+				ReturnType: []byte(`{"prim":"unit"}`),
 			},
-			storageType: gjson.Parse(`{"prim": "unit"}`),
-			want:        `[{ "prim": "parameter", "args": [{"prim": "pair", "args": [{"prim": "unit"}, {"prim": "unit"}]}]}, {"prim": "storage", "args": [{"prim": "option", "args": [{"prim": "unit"}]}]},  {"prim": "code", "args": [[{"prim": "CAR"}, {"prim": "unit"}, {"prim": "SOME"}, { "prim": "NIL", "args": [{ "prim": "operation" }]}, { "prim": "PAIR" } ]]}]`,
+			storageType: gjson.Parse(`{"prim":"unit"}`),
+			want:        `[{"prim":"parameter","args":[{"prim":"pair","args":[{"prim":"unit"},{"prim":"unit"}]}]},{"prim":"storage","args":[{"prim":"option","args":[{"prim":"unit"}]}]},{"prim":"code","args":[[{"prim":"CAR"},{"prim":"unit"},{"prim":"SOME"},{"prim":"NIL","args":[{"prim":"operation"}]},{"prim":"PAIR"}]]}]`,
+		}, {
+			name: "test 2: parameter is nil",
+			fields: fields{
+				Code:       []byte(`{"prim":"unit"}`),
+				ReturnType: []byte(`{"prim":"unit"}`),
+			},
+			storageType: gjson.Parse(`{"prim":"unit"}`),
+			want:        `[{"prim":"parameter","args":[{"prim":"unit"}]},{"prim":"storage","args":[{"prim":"option","args":[{"prim":"unit"}]}]},{"prim":"code","args":[[{"prim":"CAR"},{"prim":"unit"},{"prim":"SOME"},{"prim":"NIL","args":[{"prim":"operation"}]},{"prim":"PAIR"}]]}]`,
 		},
 	}
 	for _, tt := range tests {