Add test

Author: vearutop

vars_test.go

@@ -2,6 +2,7 @@ package shared_test
 
 import (
 	"context"
+	"encoding/json"
 	"sync"
 	"testing"
 
@@ -96,3 +97,23 @@ func TestDecodeJSONNumber(t *testing.T) {
 	assert.Equal(t, 1.23, shared.DecodeJSONNumber("1.23"))
 	assert.Equal(t, uint64(17294094973108486143), shared.DecodeJSONNumber("17294094973108486143"))
 }
+
+func TestVars_Set_jsonNumber(t *testing.T) {
+	v := shared.Vars{}
+
+	val := map[string]interface{}{
+		"foo": json.Number("123"),
+		"bar": json.Number("123.456"),
+		"baz": []interface{}{json.Number("123"), json.Number("123.456"), 789},
+	}
+
+	v.Set("test", val)
+
+	val2, ok := v.Get("test")
+	assert.True(t, ok)
+	assert.Equal(t, map[string]interface{}{
+		"foo": int64(123),
+		"bar": 123.456,
+		"baz": []interface{}{int64(123), 123.456, 789},
+	}, val2)
+}