diff --git a/pkg/handler/handler_mock_test.go b/pkg/handler/handler_mock_test.go index a2ca27b..11c687a 100644 --- a/pkg/handler/handler_mock_test.go +++ b/pkg/handler/handler_mock_test.go @@ -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) } diff --git a/pkg/handler/handler_test.go b/pkg/handler/handler_test.go index eefdaf5..be042d8 100644 --- a/pkg/handler/handler_test.go +++ b/pkg/handler/handler_test.go @@ -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() diff --git a/pkg/support/patreon_test.go b/pkg/support/patreon_test.go index 72c1845..3443acf 100644 --- a/pkg/support/patreon_test.go +++ b/pkg/support/patreon_test.go @@ -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 {