Fix broken tests

This commit is contained in:
Maksym Pavlenko
2017-11-10 17:22:12 -08:00
parent 05a1b0f8d5
commit 87af1d1f90
3 changed files with 22 additions and 10 deletions
+18 -6
View File
@@ -120,14 +120,26 @@ func (_mr *MockpatreonServiceMockRecorder) Hook(arg0, arg1 interface{}) *gomock.
return _mr.mock.ctrl.RecordCallWithMethodType(_mr.mock, "Hook", reflect.TypeOf((*MockpatreonService)(nil).Hook), arg0, arg1)
}
// GetFeatureLevel mocks base method
func (_m *MockpatreonService) GetFeatureLevel(patronID string) int {
ret := _m.ctrl.Call(_m, "GetFeatureLevel", patronID)
// GetFeatureLevelByID mocks base method
func (_m *MockpatreonService) GetFeatureLevelByID(patronID string) int {
ret := _m.ctrl.Call(_m, "GetFeatureLevelByID", patronID)
ret0, _ := ret[0].(int)
return ret0
}
// GetFeatureLevel indicates an expected call of GetFeatureLevel
func (_mr *MockpatreonServiceMockRecorder) GetFeatureLevel(arg0 interface{}) *gomock.Call {
return _mr.mock.ctrl.RecordCallWithMethodType(_mr.mock, "GetFeatureLevel", reflect.TypeOf((*MockpatreonService)(nil).GetFeatureLevel), arg0)
// GetFeatureLevelByID indicates an expected call of GetFeatureLevelByID
func (_mr *MockpatreonServiceMockRecorder) GetFeatureLevelByID(arg0 interface{}) *gomock.Call {
return _mr.mock.ctrl.RecordCallWithMethodType(_mr.mock, "GetFeatureLevelByID", reflect.TypeOf((*MockpatreonService)(nil).GetFeatureLevelByID), arg0)
}
// GetFeatureLevelFromAmount mocks base method
func (_m *MockpatreonService) GetFeatureLevelFromAmount(amount int) int {
ret := _m.ctrl.Call(_m, "GetFeatureLevelFromAmount", amount)
ret0, _ := ret[0].(int)
return ret0
}
// GetFeatureLevelFromAmount indicates an expected call of GetFeatureLevelFromAmount
func (_mr *MockpatreonServiceMockRecorder) GetFeatureLevelFromAmount(arg0 interface{}) *gomock.Call {
return _mr.mock.ctrl.RecordCallWithMethodType(_mr.mock, "GetFeatureLevelFromAmount", reflect.TypeOf((*MockpatreonService)(nil).GetFeatureLevelFromAmount), arg0)
}
+1 -1
View File
@@ -33,7 +33,7 @@ func TestCreateFeed(t *testing.T) {
feed.EXPECT().CreateFeed(gomock.Eq(req), gomock.Any()).Times(1).Return("456", nil)
patreon := NewMockpatreonService(ctrl)
patreon.EXPECT().GetFeatureLevel(gomock.Any()).Return(api.DefaultFeatures)
patreon.EXPECT().GetFeatureLevelByID(gomock.Any()).Return(api.DefaultFeatures)
srv := httptest.NewServer(New(feed, patreon, cfg))
defer srv.Close()
+3 -3
View File
@@ -72,9 +72,9 @@ func TestGetFeatureLevel(t *testing.T) {
err := hook.Hook(pledge, patreon.EventCreatePledge)
require.NoError(t, err)
require.Equal(t, api.PodcasterFeature, hook.GetFeatureLevel(creatorID))
require.Equal(t, api.DefaultFeatures, hook.GetFeatureLevel("xyz"))
require.Equal(t, api.ExtendedPagination, hook.GetFeatureLevel(pledge.Relationships.Patron.Data.ID))
require.Equal(t, api.PodcasterFeature, hook.GetFeatureLevelByID(creatorID))
require.Equal(t, api.DefaultFeatures, hook.GetFeatureLevelByID("xyz"))
require.Equal(t, api.ExtendedPagination, hook.GetFeatureLevelByID(pledge.Relationships.Patron.Data.ID))
}
func createHandler(t *testing.T) *Patreon {