test/test_helper.rb in katello-foreman-engine-0.0.1 vs test/test_helper.rb in katello-foreman-engine-0.0.2

@@ -9,12 +9,69 @@
 require 'rails/test_help'
 require 'mocha/setup'
 
 Rails.backtrace_cleaner.remove_silencers!
 
-# returns the execution plan for given action, given
+# returns the subactions being planned from the plan method of
+# +action_class+. It prevents calling plan methods on the subactions,
+# letting to test only the action_class in isolation.
+def planned_actions(action_class, input, *args)
+  planned_actions = []
+  action = action_class.new(input)
+  action.stubs(:plan_action).with do |*args|
+    planned_actions << args
+  end
+  action.plan(*args)
+  return planned_actions
+end
+
+# returns the run steps planned for given action, given
 # input from parent action and *args for plan method
-def prepare_plan(action_class, input, *args)
+def run_steps(action_class, input, *args)
   action = action_class.new(input)
   action.plan(*args)
-  return action.execution_plan
+  return action.execution_plan.run_steps
+end
+
+def expect_foreman_search(*args)
+  args = args + [true]
+  stub_foreman_search(*args)
+end
+
+def stub_foreman_search(resource, search, response, expect = false)
+  response = case response
+             when nil
+               []
+             when Array
+               response
+             else
+               [response]
+             end
+  stub_foreman_call(resource, :index, {'search' => search}, response, expect)
+end
+
+def expect_foreman_call(*args)
+  args = args + [true]
+  stub_foreman_call(*args)
+end
+
+def stub_foreman_call(resource, action, request = nil, response = nil, expect = false)
+  setup_user
+  resource_key = resource.to_s
+  # difference between class name and resource key in the API
+  resource_key.gsub!('_','') if resource_key == 'operating_system'
+  if request && [:create, :update].include?(action) && !request.has_key?(resource_key)
+    request = {resource_key => request}
+  end
+  resource_class ="ForemanApi::Resources::#{resource.to_s.camelize}".constantize
+  stub = resource_class.any_instance.stubs(action)
+  if request
+    stub.with(request)
+  end
+  if response
+    stub.returns([response, response.to_json])
+  end
+end
+
+def setup_user
+  User.current ||= User.new(:username => 'admin')
 end