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")
}
}
func TestMe(t *testing.T) {
ctrl := gomock.NewController(t)
defer ctrl.Finish()
l := net.MOCK().NewListener()
c := net.MOCK().NewConn()
e := fmt.Errorf("goodbye")
net.MOCK().SetController(ctrl)
gomock.InOrder(
net.EXPECT().Listen("tcp", ":8080").Return(l, nil),
l.EXPECT().Accept().Return(c, nil),
c.EXPECT().Close(),
l.EXPECT().Accept().Return(nil, e),
l.EXPECT().Close(),
)
if RunMe() != e {
t.Errorf("expected mock net")
}
}