adds run modes (#29)

Co-authored-by: fsainz <[email protected]>

lib/worker_tools/basics.rb

@@ -35,7 +35,7 @@ def perform(model_id = nil)
 
       default_reset
       with_wrappers(wrapper_methods) do
-        run
+        send(run_method)
       end
     end
 
@@ -92,6 +92,23 @@ def silent_error?(error)
 
     private
 
+    def run_mode
+      model.try(:options).try(:[], 'run_mode')
+    end
+
+    def run_mode_option
+      model.try(:options).try(:[], 'run_mode_option').try(:to_sym)
+    end
+
+    def run_method
+      return :run unless run_mode.present?
+
+      method_name = "run_in_#{run_mode}_mode"
+      return method_name.to_sym if respond_to?(method_name, true)
+
+      raise "Missing method #{method_name}"
+    end
+
     def default_reset
       model.attributes = { notes: [], meta: {}, state: 'running' }
       model.save!(validate: false)

test/schema.rb

@@ -6,6 +6,7 @@
     t.string 'state', default: 'waiting', null: false
     t.json 'notes', default: []
     t.json 'meta', default: {}
+    t.json 'options', default: {}
     t.timestamps
   end
 end

test/worker_tools/basics_test.rb

@@ -153,6 +153,29 @@ def run
       importer.expects(:custom_reset)
       importer.perform
     end
+
+    describe 'run modes' do
+      it 'calls run with the correct mode if defined in options' do
+        import = create_import
+        import.update!(options: { run_mode: 'foo', run_mode_option: 'bar' })
+        importer = Importer.new
+        importer.model = import
+        importer.expects(:run_in_foo_mode)
+        importer.perform
+
+        assert_equal(importer.send(:run_mode_option), :bar)
+      end
+
+      it 'raises if the run mode is not defined' do
+        import = create_import
+        import.update!(options: { run_mode: 'foo' })
+        importer = Importer.new
+        importer.model = import
+
+        err = assert_raises(StandardError) { importer.perform }
+        assert_includes err.message, 'run_in_foo_mode'
+      end
+    end
   end
 
   describe '#finalize' do