diff options
| -rw-r--r-- | app/models/import.rb | 4 | ||||
| -rw-r--r-- | spec/models/import_spec.rb | 22 |
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 |