Commit e3a91089 authored by Alex Kalderimis's avatar Alex Kalderimis Committed by Thong Kuah

Allow be_url to specify the type

This allows the be_url matcher to be more specific. By default, it only
matches HTTP and HTTPS URIs.
parent 1a513aaf
...@@ -24,16 +24,16 @@ module ChatNames ...@@ -24,16 +24,16 @@ module ChatNames
end end
def chat_name_token def chat_name_token
Gitlab::ChatNameToken.new @chat_name_token ||= Gitlab::ChatNameToken.new
end end
def chat_name_params def chat_name_params
{ {
service_id: @service.id, service_id: @service.id,
team_id: @params[:team_id], team_id: @params[:team_id],
team_domain: @params[:team_domain], team_domain: @params[:team_domain],
chat_id: @params[:user_id], chat_id: @params[:user_id],
chat_name: @params[:user_name] chat_name: @params[:user_name]
} }
end end
end end
......
...@@ -4,23 +4,36 @@ require 'spec_helper' ...@@ -4,23 +4,36 @@ require 'spec_helper'
describe ChatNames::AuthorizeUserService do describe ChatNames::AuthorizeUserService do
describe '#execute' do describe '#execute' do
let(:service) { create(:service) } subject { described_class.new(service, params) }
subject { described_class.new(service, params).execute } let(:result) { subject.execute }
let(:service) { create(:service) }
context 'when all parameters are valid' do context 'when all parameters are valid' do
let(:params) { { team_id: 'T0001', team_domain: 'myteam', user_id: 'U0001', user_name: 'user' } } let(:params) { { team_id: 'T0001', team_domain: 'myteam', user_id: 'U0001', user_name: 'user' } }
it 'produces a valid HTTP URL' do
expect(result).to be_http_url
end
it 'requests a new token' do it 'requests a new token' do
is_expected.to be_url expect(subject).to receive(:request_token).once.and_call_original
subject.execute
end end
end end
context 'when there are missing parameters' do context 'when there are missing parameters' do
let(:params) { {} } let(:params) { {} }
it 'does not produce a URL' do
expect(result).to be_nil
end
it 'does not request a new token' do it 'does not request a new token' do
is_expected.to be_nil expect(subject).not_to receive(:request_token)
subject.execute
end end
end end
end end
......
# frozen_string_literal: true # frozen_string_literal: true
RSpec::Matchers.define :be_url do |_| # Assert that this value is a valid URL of at least one type.
#
# By default, this checks that the URL is either a HTTP or HTTPS URI,
# but you can check other URI schemes by passing the type, eg:
#
# ```
# expect(value).to be_url(URI::FTP)
# ```
#
# Pass an empty array of types if you want to match any URI scheme (be
# aware that this might not do what you think it does! `foo` is a valid
# URI, for instance).
RSpec::Matchers.define :be_url do |types = [URI::HTTP, URI::HTTPS]|
match do |actual| match do |actual|
URI.parse(actual) rescue false next false unless actual.present?
uri = URI.parse(actual)
Array.wrap(types).any? { |t| uri.is_a?(t) }
rescue URI::InvalidURIError
false
end end
end end
# looks better when used like: # looks better when used like:
# expect(thing).to receive(:method).with(a_valid_url) # expect(thing).to receive(:method).with(a_valid_url)
RSpec::Matchers.alias_matcher :a_valid_url, :be_url RSpec::Matchers.alias_matcher :a_valid_url, :be_url
RSpec::Matchers.alias_matcher :be_http_url, :be_url
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment