diff options
Diffstat (limited to 'spec/lib/gitlab/email/handler/create_issue_handler_spec.rb')
-rw-r--r-- | spec/lib/gitlab/email/handler/create_issue_handler_spec.rb | 30 |
1 files changed, 30 insertions, 0 deletions
diff --git a/spec/lib/gitlab/email/handler/create_issue_handler_spec.rb b/spec/lib/gitlab/email/handler/create_issue_handler_spec.rb index dd230140b30..bd4f1d164a8 100644 --- a/spec/lib/gitlab/email/handler/create_issue_handler_spec.rb +++ b/spec/lib/gitlab/email/handler/create_issue_handler_spec.rb @@ -136,6 +136,36 @@ RSpec.describe Gitlab::Email::Handler::CreateIssueHandler do expect { handler.execute }.to raise_error(Gitlab::Email::ProjectNotFound) end end + + context 'rate limiting' do + let(:rate_limited_service_feature_enabled) { nil } + + before do + stub_feature_flags(rate_limited_service_issues_create: rate_limited_service_feature_enabled) + end + + context 'when :rate_limited_service Feature is disabled' do + let(:rate_limited_service_feature_enabled) { false } + + it 'does not attempt to throttle' do + expect(::Gitlab::ApplicationRateLimiter).not_to receive(:throttled?) + + setup_attachment + receiver.execute + end + end + + context 'when :rate_limited_service Feature is enabled' do + let(:rate_limited_service_feature_enabled) { true } + + it 'raises a RateLimitedService::RateLimitedError' do + allow(::Gitlab::ApplicationRateLimiter).to receive(:throttled?).and_return(true) + + setup_attachment + expect { receiver.execute }.to raise_error(RateLimitedService::RateLimitedError, _('This endpoint has been requested too many times. Try again later.')) + end + end + end end def email_fixture(path) |