Add deploy_key and personal_key factories

2 files changed, 16 insertions, 0 deletions

diff --git a/spec/factories.rb b/spec/factories.rb
index eb3882a6bee..82fd0db2ea4 100644
--- a/spec/factories.rb
+++ b/spec/factories.rb
@@ -76,6 +76,14 @@ FactoryGirl.define do
   factory :key do
     title
     key { File.read(File.join(Rails.root, "db", "pkey.example")) }
+
+    factory :deploy_key do
+      project
+    end
+
+    factory :personal_key do
+      user
+    end
   end
 
   factory :milestone do
diff --git a/spec/factories_spec.rb b/spec/factories_spec.rb
index 126a5b2309f..5760aad423b 100644
--- a/spec/factories_spec.rb
+++ b/spec/factories_spec.rb
@@ -49,6 +49,14 @@ describe "Factories" do
     it "builds a valid instance" do
       build(:key).should be_valid
     end
+
+    it "builds a valid deploy key instance" do
+      build(:deploy_key).should be_valid
+    end
+
+    it "builds a valid personal key instance" do
+      build(:personal_key).should be_valid
+    end
   end
 
   describe 'Milestone' do