Go1.16으로 추가된 iotest.ErrReader io를 사용합니다.Reader의 예외 시스템 테스트

io.ReadAll과 같은 예외 시스템을 테스트하고 싶습니다.

import "io"

func read(r io.Reader) (string, error) {
	b, err := io.ReadAll(r)
	if err != nil {
		return "", err
	}

	return string(b), nil
}

iotest.ErrReader를 사용하면 됩니다.

import (
	"errors"
	"io"
	"strings"
	"testing"
	"testing/iotest"
)

func Test_read(t *testing.T) {
	t.Parallel()

	cases := map[string]struct {
		r     io.Reader
		str   string
		noErr bool
	}{
		"success": {
			r:     strings.NewReader("test"),
			str:   "test",
			noErr: true,
		},
		"failure": {
			r:     iotest.ErrReader(errors.New("test")),
			str:   "",
			noErr: false,
		},
	}

	for name, c := range cases {
		c := c
		t.Run(name, func(t *testing.T) {
			t.Parallel()

			str, err := read(c.r)

			if str != c.str {
				t.Errorf("want '%v', got '%v'", c.str, str)
			}
			if c.noErr {
				if err != nil {
					t.Error("want 'nil', got 'not nil'")
				}
			} else {
				if err == nil {
					t.Error("want 'not nil', got 'nil'")
				}
			}
		})
	}
}

덮어쓰기도 좋아요.