1 files changed, 88 insertions, 0 deletions
diff --git a/spec/models/project_services/chat_message/push_message_spec.rb b/spec/models/project_services/chat_message/push_message_spec.rb
new file mode 100644
index 00000000000..b781c4505db
--- /dev/null
+++ b/spec/models/project_services/chat_message/push_message_spec.rb
@@ -0,0 +1,88 @@
+require 'spec_helper'
+
+describe ChatMessage::PushMessage, models: true do
+  subject { described_class.new(args) }
+
+  let(:args) do
+    {
+      after: 'after',
+      before: 'before',
+      project_name: 'project_name',
+      ref: 'refs/heads/master',
+      user_name: 'test.user',
+      project_url: 'url'
+    }
+  end
+
+  let(:color) { '#345' }
+
+  context 'push' do
+    before do
+      args[:commits] = [
+        { message: 'message1', url: 'url1', id: 'abcdefghijkl', author: { name: 'author1' } },
+        { message: 'message2', url: 'url2', id: '123456789012', author: { name: 'author2' } },
+      ]
+    end
+
+    it 'returns a message regarding pushes' do
+      expect(subject.pretext).to eq(
+        'test.user pushed to branch <url/commits/master|master> of '\
+        '<url|project_name> (<url/compare/before...after|Compare changes>)'
+      )
+      expect(subject.attachments).to eq([
+        {
+          text: "<url1|abcdefgh>: message1 - author1\n"\
+                "<url2|12345678>: message2 - author2",
+          color: color,
+        }
+      ])
+    end
+  end
+
+  context 'tag push' do
+    let(:args) do
+      {
+        after: 'after',
+        before: Gitlab::Git::BLANK_SHA,
+        project_name: 'project_name',
+        ref: 'refs/tags/new_tag',
+        user_name: 'test.user',
+        project_url: 'url'
+      }
+    end
+
+    it 'returns a message regarding pushes' do
+      expect(subject.pretext).to eq('test.user pushed new tag ' \
+       '<url/commits/new_tag|new_tag> to ' \
+       '<url|project_name>')
+      expect(subject.attachments).to be_empty
+    end
+  end
+
+  context 'new branch' do
+    before do
+      args[:before] = Gitlab::Git::BLANK_SHA
+    end
+
+    it 'returns a message regarding a new branch' do
+      expect(subject.pretext).to eq(
+        'test.user pushed new branch <url/commits/master|master> to '\
+        '<url|project_name>'
+      )
+      expect(subject.attachments).to be_empty
+    end
+  end
+
+  context 'removed branch' do
+    before do
+      args[:after] = Gitlab::Git::BLANK_SHA
+    end
+
+    it 'returns a message regarding a removed branch' do
+      expect(subject.pretext).to eq(
+        'test.user removed branch master from <url|project_name>'
+      )
+      expect(subject.attachments).to be_empty
+    end
+  end
+end