1 files changed, 59 insertions, 1 deletions
diff --git a/spec/factories.rb b/spec/factories.rb
index 704b91e04..18ad3ab38 100644
--- a/spec/factories.rb
+++ b/spec/factories.rb
@@ -1,23 +1,81 @@
 FactoryGirl.define do
 
+    sequence(:email) { |n| "person#{n}@example.com" }
+
+    factory :foi_attachment do
+        content_type 'text/pdf'
+        filename 'interesting.pdf'
+        body 'thisisthebody'
+    end
+
     factory :incoming_message do
         info_request
         raw_email
+        last_parsed { 1.week.ago }
+
+        factory :incoming_message_with_attachments do
+            # foi_attachments_count is declared as an ignored attribute and available in
+            # attributes on the factory, as well as the callback via the evaluator
+            ignore do
+                foi_attachments_count 5
+            end
+
+            # the after(:create) yields two values; the incoming_message instance itself and the
+            # evaluator, which stores all values from the factory, including ignored
+            # attributes;
+            after(:create) do |incoming_message, evaluator|
+                evaluator.foi_attachments_count.times do |count|
+                    FactoryGirl.create(:foi_attachment,
+                                       incoming_message: incoming_message,
+                                       url_part_number: count+1)
+                end
+            end
+        end
     end
 
     factory :raw_email
 
+    factory :outgoing_message do
+        factory :initial_request do
+            status 'ready'
+            message_type 'initial_request'
+            body 'Some information please'
+            what_doing 'normal_sort'
+            initialize_with { new(attributes) }
+            after(:create) do |outgoing_message|
+                outgoing_message.send_message
+            end
+        end
+    end
+
     factory :info_request do
         title "Example Title"
         public_body
         user
+
+        after(:create) do |info_request, evaluator|
+            FactoryGirl.create(:initial_request, info_request: info_request)
+        end
+
+        factory :info_request_with_attachments do
+            after(:create) do |info_request, evaluator|
+                FactoryGirl.create(:incoming_message_with_attachments, info_request: info_request)
+            end
+
+        end
+
     end
 
     factory :user do
         name 'Example User'
-        email 'user@example.com'
+        email
         salt "-6116981980.392287733335677"
         hashed_password '6b7cd45a5f35fd83febc0452a799530398bfb6e8' # jonespassword
+
+        factory :admin_user do
+            name 'Admin User'
+            admin_level 'super'
+        end
     end
 
     factory :public_body do