func TestSplitter(t *testing.T) { ctrl := gomock.NewController(t) defer ctrl.Finish() lib.MOCK().SetController(ctrl) splitter := lib.MOCK().NewSplitter() splitter.EXPECT().Split(5).Return(0, 0) var s lib.Splitter = splitter x, y := s.Split(5) if x+y != 0 { t.Errorf("expected mock splitter") } }
func TestAdd(t *testing.T) { ctrl := gomock.NewController(t) defer ctrl.Finish() lib.MOCK().SetController(ctrl) foo := lib.Foo{} foo.EXPECT().Add(1, 2).Return(0) bar := &Bar{&foo} res := bar.Add(1, 2) if res != 0 { t.Errorf("wrong result: %d\n", res) } }