1 files changed, 70 insertions, 0 deletions
diff --git a/spec/ci/models/runner_spec.rb b/spec/ci/models/runner_spec.rb
new file mode 100644
index 00000000000..6902c0a94e6
--- /dev/null
+++ b/spec/ci/models/runner_spec.rb
@@ -0,0 +1,70 @@
+# == Schema Information
+#
+# Table name: runners
+#
+#  id           :integer          not null, primary key
+#  token        :string(255)
+#  created_at   :datetime
+#  updated_at   :datetime
+#  description  :string(255)
+#  contacted_at :datetime
+#  active       :boolean          default(TRUE), not null
+#  is_shared    :boolean          default(FALSE)
+#  name         :string(255)
+#  version      :string(255)
+#  revision     :string(255)
+#  platform     :string(255)
+#  architecture :string(255)
+#
+
+require 'spec_helper'
+
+describe Runner do
+  describe '#display_name' do
+    it 'should return the description if it has a value' do
+      runner = FactoryGirl.build(:runner, description: 'Linux/Ruby-1.9.3-p448')
+      expect(runner.display_name).to eq 'Linux/Ruby-1.9.3-p448'
+    end
+
+    it 'should return the token if it does not have a description' do
+      runner = FactoryGirl.create(:runner)
+      expect(runner.display_name).to eq runner.description
+    end
+
+    it 'should return the token if the description is an empty string' do
+      runner = FactoryGirl.build(:runner, description: '')
+      expect(runner.display_name).to eq runner.token
+    end
+  end
+
+  describe :assign_to do
+    let!(:project) { FactoryGirl.create :project }
+    let!(:shared_runner) { FactoryGirl.create(:shared_runner) }
+
+    before { shared_runner.assign_to(project) }
+
+    it { shared_runner.should be_specific }
+    it { shared_runner.projects.should == [project] }
+    it { shared_runner.only_for?(project).should be_true }
+  end
+
+  describe "belongs_to_one_project?" do
+    it "returns false if there are two projects runner assigned to" do
+      runner = FactoryGirl.create(:specific_runner)
+      project = FactoryGirl.create(:project)
+      project1 = FactoryGirl.create(:project)
+      project.runners << runner
+      project1.runners << runner
+
+      runner.belongs_to_one_project?.should be_false
+    end
+
+    it "returns true" do
+      runner = FactoryGirl.create(:specific_runner)
+      project = FactoryGirl.create(:project)
+      project.runners << runner
+
+      runner.belongs_to_one_project?.should be_true
+    end
+  end
+end