Golang EXPECTの例

コード例 #1

ファイル: 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

ファイル: 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)
	}
}