Create an old_relation_member factory, and start using it for tests.

[rails.git] / test / factories / node.rb

diff --git a/test/factories/node.rb b/test/factories/node.rb
index 8daadd35ad074258da98c6c2050cabf7ce2ed0bb..cbebbb00986096245b972bafa98223472199198c 100644 (file)
--- a/test/factories/node.rb
+++ b/test/factories/node.rb
@@ -9,11 +9,22 @@ FactoryGirl.define do
     timestamp Time.now
     version 1
 
+    trait :deleted do
+      visible false
+    end
+
     trait :with_history do
       after(:create) do |node, _evaluator|
         (1..node.version).each do |n|
           create(:old_node, :node_id => node.id, :version => n)
         end
+
+        # For deleted nodes, make sure the most recent old_node is also deleted.
+        if node.visible == false
+          latest = node.old_nodes.find_by(:version => node.version)
+          latest.visible = false
+          latest.save
+        end
       end
     end
   end