From bd7b3374f87f083ffcb2c9d5d52ac2f3d2049e2b Mon Sep 17 00:00:00 2001
From: Teddy Wing
Date: Wed, 30 Aug 2017 18:26:17 +0200
Subject: Import spec: Test that `ended_at` gets updated correctly

---
 spec/models/import_spec.rb | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/spec/models/import_spec.rb b/spec/models/import_spec.rb
index 2a0c7539e..cc0fdd4ce 100644
--- a/spec/models/import_spec.rb
+++ b/spec/models/import_spec.rb
@@ -128,7 +128,18 @@ RSpec.describe Import, type: :model do
     end
 
     it "updates :ended_at to now when status is finished" do
-      skip "Redo the `#update_status` code to make it easier to write this."
+      workbench_import = create(:workbench_import)
+      create(
+        :netex_import,
+        parent: workbench_import,
+        status: 'failed'
+      )
+
+      Timecop.freeze(Time.now) do
+        workbench_import.update_status
+
+        expect(workbench_import.ended_at).to eq(Time.now)
+      end
     end
   end
 
-- 
cgit v1.2.3