How to use Write method of got_test Package

...50	}51	for _, tt := range cases {52		t.Run("", func(t *testing.T) {53			ioStream := newTestIOStream()54			ioStream.In.(*bytes.Buffer).WriteString(tt.stdin)55			cfg := &got.PrompterConfig{56				IsDebug: true,57			}58			p := got.NewPrompter(ioStream, cfg)59			p.SelectExecutableToDisable(tt.input)60			if diff := cmp.Diff(tt.want, tt.input); diff != "" {61				t.Fatalf("mismatch (-want, +got): %s\n", diff)62			}63			t.Logf("\n%s", ioStream.Err.(*bytes.Buffer).String())64		})65	}66}67func TestPrompter_ChooseToForceOverwrite(t *testing.T) {68	cases := []struct {69		input string70		stdin string71		want  bool72	}{73		{74			input: "/path/to/overwrite",75			stdin: "y",76			want:  true,77		},78		{79			input: "/path/to/overwrite",80			stdin: "n",81			want:  false,82		},83	}84	for _, tt := range cases {85		t.Run("", func(t *testing.T) {86			ioStream := newTestIOStream()87			ioStream.In.(*bytes.Buffer).WriteString(tt.stdin)88			cfg := &got.PrompterConfig{89				IsDebug: true,90			}91			p := got.NewPrompter(ioStream, cfg)92			got := p.ChooseToForceOverwrite(tt.input)93			if diff := cmp.Diff(tt.want, got); diff != "" {94				t.Fatalf("mismatch (-want, +got): %s\n", diff)95			}96			t.Logf("\n%s", ioStream.Err.(*bytes.Buffer).String())97		})98	}99}100func TestPrompter_AskYN(t *testing.T) {101	cases := []struct {102		input   string103		msg     string104		wantMsg string105		want    bool106	}{107		{108			input:   "y",109			msg:     "test",110			wantMsg: "test [Y/n]: ",111			want:    true,112		},113		{114			input:   "Y",115			msg:     "test",116			wantMsg: "test [Y/n]: ",117			want:    true,118		},119		{120			input:   "n",121			msg:     "test",122			wantMsg: "test [Y/n]: ",123			want:    false,124		},125		{126			input:   "N",127			msg:     "test",128			wantMsg: "test [Y/n]: ",129			want:    false,130		},131		{132			input:   "",133			msg:     "test",134			wantMsg: "test [Y/n]: ",135			want:    true,136		},137		{138			input:   "hoge\ny",139			msg:     "test",140			wantMsg: "test [Y/n]: Please enter one of the following: Y/y/N/n.\ntest [Y/n]: ",141			want:    true,142		},143	}144	for _, tt := range cases {145		t.Run("", func(t *testing.T) {146			ioStream := newTestIOStream()147			ioStream.In.(*bytes.Buffer).WriteString(tt.input)148			cfg := &got.PrompterConfig{149				IsDebug: true,150			}151			p := got.NewPrompter(ioStream, cfg)152			got := p.AskYN(tt.msg)153			if diff := cmp.Diff(tt.want, got); diff != "" {154				t.Fatalf("mismatch (-want, +got): %s\n", diff)155			}156			stdout := string(ioStream.Out.(*bytes.Buffer).Bytes())157			if diff := cmp.Diff(tt.wantMsg, stdout); diff != "" {158				t.Fatalf("mismatch (-want, +got): %s\n", diff)159			}160			t.Logf("\n%s", ioStream.Err.(*bytes.Buffer).String())161		})162	}163}164func TestPrompter_Select(t *testing.T) {165	cases := []struct {166		input      string167		msg        string168		candidates []string169		wantMsg    string170		want       int171	}{172		{173			input:      "0",174			msg:        "test",175			candidates: []string{"a", "b", "c"},176			wantMsg:    "\ta\n\tb\n\tc\ntest: ",177			want:       0,178		},179		{180			input:      "1",181			msg:        "test",182			candidates: []string{"a", "b", "c"},183			wantMsg:    "\ta\n\tb\n\tc\ntest: ",184			want:       1,185		},186		{187			input:      "2",188			msg:        "test",189			candidates: []string{"a", "b", "c"},190			wantMsg:    "\ta\n\tb\n\tc\ntest: ",191			want:       2,192		},193		{194			input:      "3\n2",195			msg:        "test",196			candidates: []string{"a", "b", "c"},197			wantMsg:    "\ta\n\tb\n\tc\ntest: Invalid input: 3.\n\ta\n\tb\n\tc\ntest: ",198			want:       2,199		},200	}201	for _, tt := range cases {202		t.Run("", func(t *testing.T) {203			ioStream := newTestIOStream()204			ioStream.In.(*bytes.Buffer).WriteString(tt.input)205			cfg := &got.PrompterConfig{206				IsDebug: true,207			}208			p := got.NewPrompter(ioStream, cfg)209			got := p.Select(tt.msg, tt.candidates, 1)210			if diff := cmp.Diff(tt.want, got); diff != "" {211				t.Fatalf("mismatch (-want, +got): %s\n", diff)212			}213			stdout := string(ioStream.Out.(*bytes.Buffer).Bytes())214			if diff := cmp.Diff(tt.wantMsg, stdout); diff != "" {215				t.Fatalf("mismatch (-want, +got): %s\n", diff)216			}217			t.Logf("\n%s", ioStream.Err.(*bytes.Buffer).String())218		})...

...9	"time"10	"github.com/melbahja/got"11)12func NewHttptestServer() *httptest.Server {13	return httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {14		switch r.URL.String() {15		case "/ok_file":16			http.ServeFile(w, r, "go.mod")17			return18		case "/found_and_head_not_allowed":19			if r.Method == http.MethodHead {20				w.WriteHeader(http.StatusMethodNotAllowed)21				return22			}23			fmt.Fprint(w, "helloworld")24			return25		case "/not_found_and_method_not_allowed":26			w.WriteHeader(http.StatusMethodNotAllowed)27			return28		case "/ok_file_with_range_delay":29			if r.Method == http.MethodGet && strings.Contains(r.Header.Get("range"), "3-") {30				time.Sleep(3 * time.Second)31			}32			http.ServeFile(w, r, "go.mod")33			return34		case "/file_name":35			w.Header().Set("Content-Disposition", `attachment; filename="go.mod"`)36			http.ServeFile(w, r, "go.mod")37			return38		case "/header_values":39			if r.Header.Get("x-test-header") == "foobar" {40				http.ServeFile(w, r, "go.mod")41				return42			}43			w.WriteHeader(403)44			return45		case "/not_found":46		}47		w.WriteHeader(http.StatusNotFound)48	}))49}50var testUrl = httpt.URL + "/ok_file"51func ExampleGot() {52	// Just for testing53	destPath := createTemp()54	defer clean(destPath)55	g := got.New()56	err := g.Download(testUrl, destPath)57	if err != nil {58		log.Fatal(err)59		return60	}61	fmt.Println("done")...

...6)7type mockBuffer struct {8	got.Mock9}10func (t *mockBuffer) Write(p []byte) (n int, err error) {11	return t.Proxy("Write").(func([]byte) (int, error))(p)12}13func (t *mockBuffer) Len() int {14	return t.Proxy("Len").(func() int)()15}16func (t *mockBuffer) Nonexists() int {17	return t.Proxy("Nonexists").(func() int)()18}19func TestMock(t *testing.T) {20	g := setup(t)21	b := bytes.NewBuffer(nil)22	m := mockBuffer{}23	m.Fallback(b)24	m.Stub("Write", func(p []byte) (int, error) {25		return b.Write(append(p, []byte("  ")...))26	})27	n, err := m.Write([]byte("test"))28	g.Nil(err)29	g.Eq(n, 6)30	g.Eq(m.Len(), 6)31	val := g.Panic(func() {32		m := mockBuffer{}33		m.Len()34	})35	g.Eq(val, "you should specify the got.Mock.Origin")36	val = g.Panic(func() {37		m := mockBuffer{}38		m.Fallback(b)39		m.Nonexists()40	})41	g.Eq(val, `*bytes.Buffer doesn't have method: Nonexists`)42}43func TestMockUtils(t *testing.T) {44	g := setup(t)45	b := bytes.NewBuffer(nil)46	m := &mockBuffer{}47	m.Fallback(b)48	{49		m.On(m, "Write").When([]byte{}).Return(2, nil).Times(2)50		n, err := m.Write(nil)51		g.Nil(err)52		g.Eq(n, 2)53		n, err = m.Write(nil)54		g.Nil(err)55		g.Eq(n, 2)56		n, err = m.Write(nil)57		g.Nil(err)58		g.Eq(n, 0)59	}60	{61		m.On(m, "Write").When(got.Any).Return(2, nil)62		n, err := m.Write(nil)63		g.Nil(err)64		g.Eq(n, 2)65	}66	{67		m.On(m, "Write").When([]byte{}).Return(2, nil).Once()68		n, err := m.Write(nil)69		g.Nil(err)70		g.Eq(n, 2)71		n, err = m.Write(nil)72		g.Nil(err)73		g.Eq(n, 0)74	}75	{76		m.On(m, "Write").When(true).Return(2, nil)77		v := g.Panic(func() {78			_, _ = m.Write(nil)79		})80		g.Eq(v, "No got.StubOn.When matches: []interface {}{[]uint8(nil)}")81	}82}...

1import (2func main() {3	fmt.Println("Hello World")4	a := got.Got_test{1, 2, 3}5	a.Write()6}

1import (2func main() {3	reader = bufio.NewReader(os.Stdin)4	writer = bufio.NewWriter(os.Stdout)5	fmt.Fprintln(writer, "Enter the first number")6	writer.Flush()7	input, err = reader.ReadString('\n')8	if err != nil {9		fmt.Println("Error while reading input")10	}11	input = strings.TrimSpace(input)12	a, err = strconv.Atoi(input)13	if err != nil {14		fmt.Println("Error converting string to integer")15	}16	fmt.Fprintln(writer, "Enter the second number")17	writer.Flush()18	input, err = reader.ReadString('\n')19	if err != nil {20		fmt.Println("Error while reading input")21	}22	input = strings.TrimSpace(input)23	b, err = strconv.Atoi(input)24	if err != nil {25		fmt.Println("Error converting string to integer")26	}27	fmt.Fprintln(writer, "Enter the operator")28	writer.Flush()29	input, err = reader.ReadString('\n')30	if err != nil {31		fmt.Println("Error while reading input")32	}33	input = strings.TrimSpace(input)34	switch op {35		fmt.Println("Invalid operator")36	}37	fmt.Fprintln(writer, "Result is", result)38	writer.Flush()39}

1import (2func main() {3	fmt.Println("Hello World")4	got_test := gotest.Got_test{1, 2}5	got_test.Write()6}