Merge pull request #138 from pulibrary/create-import-method

Create import method for GuideCard and SubGuideCard images
pulibrary · Sep 1, 2023 · 234cb76 · 234cb76
2 parents 95ad1a2 + 4237443
commit 234cb76
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/app/services/card_image_loading_service.rb b/app/services/card_image_loading_service.rb
@@ -10,6 +10,13 @@ def initialize(progressbar: nil)
     @progressbar = progressbar || ProgressBar.create(format: "\e[1;35m%t: |%B|\e[0m")
   end
 
+  def import
+    @progressbar = ProgressBar.create(format: "\e[1;35m%t: |%B|\e[0m")
+    import_guide_card_images
+    @progressbar = ProgressBar.create(format: "\e[1;35m%t: |%B|\e[0m")
+    import_sub_guide_images
+  end
+
   # For each SubGuideCard, take its path and query s3 to get all of the image names
   # for that path. For each image file, create a CardImage object with the path and
   # image name.

diff --git a/spec/services/card_image_loading_service_spec.rb b/spec/services/card_image_loading_service_spec.rb
@@ -55,4 +55,16 @@
     cils.import_sub_guide_images
     expect(cils.progressbar.to_h['percentage']).to eq 100
   end
+
+  it 'imports both GuideCard and SubGuideCard images' do
+    expect(GuideCard.count).to eq 0
+    expect(SubGuideCard.count).to eq 0
+    gcls.import
+    expect(GuideCard.count).to eq 12
+    sgls.import
+    expect(SubGuideCard.count).to eq 7
+    expect(CardImage.count).to eq 0
+    cils.import
+    expect(CardImage.count).to eq 38
+  end
 end