如何覆盖在GoMock中进行的调用？

import (
    "testing"
    "github.com/golang/mock/gomock"
    "your-package"
)

func TestYourFunction(t *testing.T) {
    ctrl := gomock.NewController(t)
    defer ctrl.Finish()

    mockObj := yourpackage.NewMockYourInterface(ctrl)
}

mockObj.EXPECT().YourMethod(arg1, arg2).Return(result)

yourFunction(mockObj)

if err := mockObj.YourMethod(arg1, arg2); err != nil {
    t.Errorf("Unexpected error: %v", err)
}