Import: Add #ready?

This boolean method answers whether or not the current import's
`current_step` and `total_steps` fields match. If so, all of its
subtasks are finished, in which case it can be thought of as "ready".

Refs #4174, #3511

2 files changed, 26 insertions, 0 deletions

diff --git a/app/models/import.rb b/app/models/import.rb
index 3cbdb964f..46de0d727 100644
--- a/app/models/import.rb
+++ b/app/models/import.rb
@@ -20,6 +20,10 @@ class Import < ActiveRecord::Base
   def child_change(child)
   end
 
+  def ready?
+    current_step == total_steps
+  end
+
   private
 
   def initialize_fields
diff --git a/spec/models/import_spec.rb b/spec/models/import_spec.rb
index 53513eb0c..1b8fe72c2 100644
--- a/spec/models/import_spec.rb
+++ b/spec/models/import_spec.rb
@@ -23,4 +23,26 @@ RSpec.describe Import, :type => :model do
       netex_import.notify_parent
     end
   end
+
+  describe "#ready?" do
+    it "returns true if #current_step == #total_steps" do
+      import = build_stubbed(
+        :import,
+        total_steps: 4,
+        current_step: 4
+      )
+
+      expect(import.ready?).to be true
+    end
+
+    it "returns false if #current_step != #total_steps" do
+      import = build_stubbed(
+        :import,
+        total_steps: 6,
+        current_step: 3
+      )
+
+      expect(import.ready?).to be false
+    end
+  end
 end