Merge pull request #447 from elisasre/feat/must-encode

Add must json encoding

v2/must/must.go

@@ -49,6 +49,18 @@ func Unmarshal(t T, data []byte, v interface{}) {
 	require.NoError(t, json.Unmarshal(data, v))
 }
 
+func EncodeJSON(t T, w io.Writer, v interface{}) {
+	t.Helper()
+	err := json.NewEncoder(w).Encode(v)
+	require.NoError(t, err)
+}
+
+func DecodeJSON(t T, r io.Reader, v interface{}) {
+	t.Helper()
+	err := json.NewDecoder(r).Decode(v)
+	require.NoError(t, err)
+}
+
 func ReadAll(t T, r io.Reader) []byte {
 	t.Helper()
 	b, err := io.ReadAll(r)

v2/must/must_test.go

@@ -43,6 +43,19 @@ func TestHappyPath(t *testing.T) {
 	require.Equal(t, body, data)
 }
 
+func TestMustEncodeJSON(t *testing.T) {
+	buf := &bytes.Buffer{}
+	noFail(t, func(mt *mockT) { must.EncodeJSON(mt, buf, "hello") })
+	assert.JSONEq(t, `"hello"`, buf.String())
+}
+
+func TestMustDecodeJSON(t *testing.T) {
+	buf := bytes.NewReader([]byte(`"world"`))
+	var s string
+	noFail(t, func(mt *mockT) { must.DecodeJSON(mt, buf, &s) })
+	assert.Equal(t, "world", s)
+}
+
 type mockT struct {
 	failed bool
 	msg    string