コード例 #1
0
ファイル: code_test.go プロジェクト: Grigs-b/withmock
func TestTryMe(t *testing.T) {
	ctrl := gomock.NewController(t)
	defer ctrl.Finish()

	dep.MOCK().SetController(ctrl)
	dep.MOCK().MockAll(true)

	dep.EXPECT().Wibble(1, 2).Return(5)

	ret := TryMe(1, 2)

	if ret != 5 {
		t.Errorf("TryMe returned %d, not 5", ret)
	}
}
コード例 #2
0
ファイル: code_test.go プロジェクト: Grigs-b/withmock
func TestDisableMock(t *testing.T) {
	ctrl := gomock.NewController(t)
	defer ctrl.Finish()

	dep.MOCK().SetController(ctrl)
	dep.MOCK().MockAll(true)
	dep.MOCK().DisableMock("NewFoo")
	dep.MOCK().DisableMock("Foo.Wibble")

	dep.EXPECT().Wibble(1, 2).Return(5)

	ret := TryMe(1, 2)

	if ret != 5 {
		t.Errorf("TryMe returned %d, not 5", ret)
	}

	ret = TryMe3(2, 3)

	if ret != 6 {
		t.Errorf("TryMe returned %d, not 6", ret)
	}
}