X-Git-Url: https://git.openstreetmap.org/rails.git/blobdiff_plain/9ae3df1a785fb2e71c7431fc69a1c2ac4204ba8c..de1bc30b83791b1e68c2382c27aee6380a19367e:/test/factories/node.rb

diff --git a/test/factories/node.rb b/test/factories/node.rb
index 85eded405..cbebbb009 100644
--- a/test/factories/node.rb
+++ b/test/factories/node.rb
@@ -8,5 +8,24 @@ FactoryGirl.define do
     visible true
     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
 end