From 65afd2cfe722e57241d27cf8a1069ca67aafc3e0 Mon Sep 17 00:00:00 2001
From: Alban Peignier
Date: Thu, 10 May 2018 19:39:10 +0200
Subject: Fix specs. Refs #6960

---
 app/models/import/gtfs.rb | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

(limited to 'app/models')

diff --git a/app/models/import/gtfs.rb b/app/models/import/gtfs.rb
index 520cc88e9..9dab11f0e 100644
--- a/app/models/import/gtfs.rb
+++ b/app/models/import/gtfs.rb
@@ -11,15 +11,15 @@ class Import::Gtfs < Import::Base
   end
 
   def main_resource
-    @resource ||= parent.resources.find_or_create_by(name: self.name, resource_type: "referential", reference: self.name)
+    @resource ||= parent.resources.find_or_create_by(name: self.name, resource_type: "referential", reference: self.name) if parent
   end
 
   def next_step
-    main_resource.next_step
+    main_resource&.next_step
   end
 
   def create_message args
-    main_resource.messages.create args
+    (main_resource || self).messages.build args
   end
 
   def import
@@ -53,7 +53,7 @@ class Import::Gtfs < Import::Base
       workbench_id: workbench.id,
       metadatas: [referential_metadata]
     )
-    main_resource.update referential: referential
+    main_resource.update referential: referential if main_resource
   end
 
   def referential_metadata
-- 
cgit v1.2.3