debian-mirror-gitlab/spec/workers/irker_worker_spec.rb
2022-02-05 19:09:49 +05:30

112 lines
3.1 KiB
Ruby

# frozen_string_literal: true
require 'spec_helper'
RSpec.describe IrkerWorker, '#perform' do
let_it_be(:project) { create(:project, :repository) }
let_it_be(:user) { create(:user) }
let_it_be(:push_data) { HashWithIndifferentAccess.new(Gitlab::DataBuilder::Push.build_sample(project, user)) }
let_it_be(:channels) { ['irc://test.net/#test'] }
let_it_be(:server_settings) do
{
server_host: 'localhost',
server_port: 6659
}
end
let_it_be(:arguments) do
[
project.id,
channels,
false,
push_data,
HashWithIndifferentAccess.new(server_settings)
]
end
let(:tcp_socket) { double('socket') }
subject(:worker) { described_class.new }
before do
allow(TCPSocket).to receive(:new).and_return(tcp_socket)
allow(tcp_socket).to receive(:puts).and_return(true)
allow(tcp_socket).to receive(:close).and_return(true)
end
context 'local requests are not allowed' do
before do
allow(Gitlab::CurrentSettings).to receive(:allow_local_requests_from_web_hooks_and_services?).and_return(false)
end
it { expect(worker.perform(*arguments)).to be_falsey }
end
context 'connection fails' do
before do
allow(TCPSocket).to receive(:new).and_raise(Errno::ECONNREFUSED.new('test'))
end
it { expect(subject.perform(*arguments)).to be_falsey }
end
context 'connection successful' do
before do
allow(Gitlab::CurrentSettings)
.to receive(:allow_local_requests_from_web_hooks_and_services?).and_return(true)
end
it { expect(subject.perform(*arguments)).to be_truthy }
context 'new branch' do
it 'sends a correct message with branches url' do
branches_url = Gitlab::Routing.url_helpers
.project_branches_url(project)
push_data['before'] = '0000000000000000000000000000000000000000'
message = "has created a new branch master: #{branches_url}"
expect(tcp_socket).to receive(:puts).with(wrap_message(message))
subject.perform(*arguments)
end
end
context 'deleted branch' do
it 'sends a correct message' do
push_data['after'] = '0000000000000000000000000000000000000000'
message = "has deleted the branch master"
expect(tcp_socket).to receive(:puts).with(wrap_message(message))
subject.perform(*arguments)
end
end
context 'new commits to existing branch' do
it 'sends a correct message with a compare url' do
compare_url = Gitlab::Routing.url_helpers
.project_compare_url(project,
from: Commit.truncate_sha(push_data[:before]),
to: Commit.truncate_sha(push_data[:after]))
message = "pushed #{push_data['total_commits_count']} " \
"new commits to master: #{compare_url}"
expect(tcp_socket).to receive(:puts).with(wrap_message(message))
subject.perform(*arguments)
end
end
end
def wrap_message(text)
message = "[#{project.path}] #{push_data['user_name']} #{text}"
to_send = { to: channels, privmsg: message }
Gitlab::Json.dump(to_send)
end
end