test/factories/task_factory.rb in foreman-tasks-1.0.1 vs test/factories/task_factory.rb in foreman-tasks-1.1.0

@@ -20,11 +20,11 @@
 
       after(:build) do |task, evaluator|
         execution_plan = ForemanTasks.dynflow.world.plan(Support::DummyDynflowAction)
         # remove the task created automatically by the persistence
         ForemanTasks::Task.where(:external_id => execution_plan.id).delete_all
-        task.update_attributes!(:external_id => execution_plan.id)
+        task.update!(:external_id => execution_plan.id)
         if evaluator.sync_with_dynflow
           task.update_from_dynflow(execution_plan.to_hash)
         end
       end
 
@@ -36,11 +36,43 @@
         label { 'Actions::Katello::Product::Create' }
       end
 
       trait :inconsistent_dynflow_task do
         after(:build) do |task|
-          task.update_attributes!(:state => 'running')
+          task.update!(:state => 'running')
         end
       end
+
+      factory :task_with_locks do
+        # posts_count is declared as a transient attribute and available in
+        # attributes on the factory, as well as the callback via the evaluator
+        transient do
+          locks_count { 3 }
+          resource_id { 1 }
+          resource_type { 'type1' }
+        end
+
+        # the after(:create) yields two values; the user instance itself and the
+        # evaluator, which stores all values from the factory, including transient
+        # attributes; `create_list`'s second argument is the number of records
+        # to create and we make sure the user is associated properly to the post
+        after(:create) do |task, evaluator|
+          create_list(
+            :lock,
+            evaluator.locks_count,
+            task: task,
+            resource_type: evaluator.resource_type,
+            resource_id: evaluator.resource_id
+          )
+        end
+      end
     end
+  end
+
+  factory :lock, :class => ForemanTasks::Lock do
+    name { 'read' }
+    resource_type { 'Katello::Repository' }
+    resource_id { 1 }
+    exclusive { true }
+    association :task, factory: :task_with_locks
   end
 end