Function TestInjectJSONPropertiesFromBytes

base/util_test.go:1079–1137

(t *testing.T)

}

func TestInjectJSONPropertiesFromBytes(t *testing.T) {
	newKVBytes := KVPairBytes{
		Key: "newval",
		Val: []byte(`{"abc":123,"nums":["one","two","three"],"test":true}`),
	}

	tests := []struct {
		input          string
		expectedOutput string
		expectedErr    string
	}{
		{
			input:       ``,
			expectedErr: `not a JSON object`,
		},
		{
			input:       `null`,
			expectedErr: `not a JSON object`,
		},
		{
			input:       "123",
			expectedErr: `not a JSON object`,
		},
		{
			input:          "{}",
			expectedOutput: `{"newval":{"abc":123,"nums":["one","two","three"],"test":true}}`,
		},
		{
			input:          `{"key":"val"}`,
			expectedOutput: `{"key":"val","newval":{"abc":123,"nums":["one","two","three"],"test":true}}`,
		},
		{
			input:          `{"newval":"old"}`,
			expectedOutput: `{"newval":"old","newval":{"abc":123,"nums":["one","two","three"],"test":true}}`,
		},
		{
			input:          `    {"key":"val"}  `,
			expectedOutput: `{"key":"val","newval":{"abc":123,"nums":["one","two","three"],"test":true}}`,
		},
	}

	for _, test := range tests {
		t.Run(test.input, func(tt *testing.T) {
			output, err := InjectJSONPropertiesFromBytes([]byte(test.input), newKVBytes)
			if test.expectedErr != "" {
				require.Errorf(tt, err, test.expectedErr, "expected error did not match")
				return
			} else {
				require.NoError(tt, err, "unexpected error")
			}

			assert.Equal(tt, test.expectedOutput, string(output))

			var m map[string]interface{}
			err = JSONUnmarshal(output, &m)
			assert.NoError(tt, err, "produced invalid JSON")
		})
	}