Refs #2670 : Update Import model

2 files changed, 6 insertions, 2 deletions

diff --git a/spec/factories/imports.rb b/spec/factories/imports.rb
index bb9a97c9c..fc8668606 100644
--- a/spec/factories/imports.rb
+++ b/spec/factories/imports.rb
@@ -1,11 +1,13 @@
 FactoryGirl.define do
   factory :import do
-    name "MyString"
+    sequence(:name) { |n| "Import #{n}" }
     current_step_id "MyString"
     current_step_progress 1.5
     association :workbench
     association :referential
     file {File.open(File.join(Rails.root, 'spec', 'fixtures', 'terminated_job.json'))}
-    status "MyString"
+    status :new
+    started_at nil
+    ended_at nil
   end
 end
diff --git a/spec/models/import_spec.rb b/spec/models/import_spec.rb
index 4c44eb99c..e34f368ae 100644
--- a/spec/models/import_spec.rb
+++ b/spec/models/import_spec.rb
@@ -4,5 +4,7 @@ RSpec.describe Import, :type => :model do
   it { should belong_to(:referential) }
   it { should belong_to(:workbench) }
 
+  it { should enumerize(:status).in(:new, :pending, :successful, :failed, :canceled) }
+
   it { should validate_presence_of(:file) }
 end