Add SubGuide association to GuideCard

app/models/sub_guide_card.rb

@@ -1,4 +1,7 @@
 # frozen_string_literal: true
 
+# SubGuideCard class which each GuideCard belongs to
+# Line 6 is a validation. The validation states that in order for a sub_guide to save it needs an associated guide_card
 class SubGuideCard < ApplicationRecord
+  belongs_to :guide_card
 end

app/services/sub_guide_loading_service.rb

@@ -10,22 +10,23 @@ def initialize(csv_location: nil)
     @csv_location = csv_location || Rails.root.join('data', 'dbo-subguides', 'dbo.Subguides.17917.csv')
   end
 
-  def import
+  def import(parent_guide:)
     sub_guide_card_data = CSV.parse(File.read(csv_location), headers: true)
-    sub_guide_card_data.each do |entry|
-      import_sub_guide_card(entry)
+    sub_guide_card_data.each do |card|
+      import_sub_guide_card(card:, parent_guide:)
     end
   end
 
   private
 
-  def import_sub_guide_card(card)
+  def import_sub_guide_card(card:, parent_guide:)
     sgc = SubGuideCard.new
     sgc.id = card[0]
     sgc.heading = card[1]
     sgc.sortid = card[2]
     sgc.parentid = card[3]
     sgc.path = card[4]
+    sgc.guide_card = parent_guide
     sgc.save
   end
 end

db/migrate/20230517135006_add_guide_card_id_to_sub_guide_card.rb

@@ -0,0 +1,10 @@
+# frozen_string_literal: true
+
+# db migration for GuideCard and SubGuide association
+class AddGuideCardIdToSubGuideCard < ActiveRecord::Migration[7.0]
+  def change
+    change_table :sub_guide_cards do |t|
+      t.belongs_to :guide_card
+    end
+  end
+end

spec/models/sub_guide_card_spec.rb

@@ -10,11 +10,13 @@
     sub_guide_card.sortid = '50350.5'
     sub_guide_card.parentid = '50345.5'
     sub_guide_card.path = '9/0091/A3067'
+    sub_guide_card.guide_card = GuideCard.create
     sub_guide_card.save
     expect(sub_guide_card.id).to eq 2
     expect(sub_guide_card.heading).to eq 'Afdeling natuurkunde'
     expect(sub_guide_card.sortid).to eq '50350.5'
     expect(sub_guide_card.parentid).to eq '50345.5'
     expect(sub_guide_card.path).to eq '9/0091/A3067'
+    expect(sub_guide_card.guide_card.class).to eq GuideCard
   end
 end

spec/services/sub_guide_loading_service_spec.rb

@@ -14,7 +14,8 @@
 
   it 'imports all data from the CSV file' do
     expect(SubGuideCard.count).to eq 0
-    sgls.import
+    new_guide_card = GuideCard.new
+    sgls.import(parent_guide: new_guide_card)
     expect(SubGuideCard.count).to eq 5
   end
 end