Function TestJsonArrayWriter_Copy

pkg/cmd/api/pagination_test.go:228–264

(t *testing.T)

}

func TestJsonArrayWriter_Copy(t *testing.T) {
	tests := []struct {
		name  string
		limit int
	}{
		{
			name: "unlimited",
		},
		{
			name:  "limited",
			limit: 2,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			buf := &bytes.Buffer{}
			w := &jsonArrayWriter{
				Writer: buf,
			}

			r := &noWriteToReader{
				Reader: bytes.NewBufferString(`[1,2]`),
				limit:  tt.limit,
			}

			require.NoError(t, startPage(w))

			n, err := io.Copy(w, r)
			require.NoError(t, err)
			assert.Equal(t, int64(5), n)

			require.NoError(t, w.Close())
			assert.Equal(t, `[[1,2]]`, buf.String())
		})
	}
}

type noWriteToReader struct {
	io.Reader