Exemplo n.º 1
0
func TestMe(t *testing.T) {
	ctrl := gomock.NewController(t)
	defer ctrl.Finish()

	http.MOCK().SetController(ctrl)
	resp := &http.Response{Status: "567 Test"}
	http.EXPECT().Get("http://www.google.com").Return(resp, nil)

	if RunMe() != "567 Test" {
		t.Errorf("expected mock net/http")
	}
}
Exemplo n.º 2
0
func TestMe(t *testing.T) {
	ctrl := gomock.NewController(t)
	defer ctrl.Finish()

	l := net.MOCK().NewListener()
	c := net.MOCK().NewConn()

	addr := ":8080"

	net.MOCK().SetController(ctrl)
	http.MOCK().SetController(ctrl)

	gomock.InOrder(
		net.EXPECT().Listen("tcp", addr).Return(l, nil),
		time.EXPECT().Sleep(2*time.Second),
	)

	if RunMe(addr) != nil {
		t.Errorf("didn't expect error")
	}
}