🎨 split each document type into its own spec file (#21)

Author: ianardee

spec/data.rb

@@ -1,15 +1,12 @@
 # frozen_string_literal: true
 
 DATA_DIR = File.join(__dir__, 'data').freeze
-DIR_CUSTOM_V1 = File.join(DATA_DIR, 'custom', 'response_v1').freeze
-DIR_INVOICE_V4 = File.join(DATA_DIR, 'invoice', 'response_v4').freeze
-DIR_RECEIPT_V4 = File.join(DATA_DIR, 'receipt', 'response_v4').freeze
-DIR_FINANCIAL_DOCUMENT_V1 = File.join(DATA_DIR, 'financial_document', 'response_v1').freeze
-DIR_PROOF_OF_ADDRESS_V1 = File.join(DATA_DIR, 'proof_of_address', 'response_v1').freeze
-DIR_PASSPORT_V1 = File.join(DATA_DIR, 'passport', 'response_v1').freeze
-DIR_EU_LICENSE_PLATE_V1 = File.join(DATA_DIR, 'eu', 'license_plate', 'response_v1').freeze
-DIR_SHIPPING_CONTAINER_V1 = File.join(DATA_DIR, 'shipping_container', 'response_v1').freeze
-DIR_US_BANK_CHECK_V1 = File.join(DATA_DIR, 'us', 'bank_check', 'response_v1').freeze
-DIR_FR_BANK_ACCOUNT_DETAILS_V1 = File.join(DATA_DIR, 'fr', 'bank_account_details', 'response_v1').freeze
-DIR_FR_CARTE_VITALE_V1 = File.join(DATA_DIR, 'fr', 'carte_vitale', 'response_v1').freeze
-DIR_FR_ID_CARD_V1 = File.join(DATA_DIR, 'fr', 'id_card', 'response_v1').freeze
+
+def load_json(dir_path, file_name)
+  file_data = File.read(File.join(dir_path, file_name))
+  JSON.parse(file_data)
+end
+
+def read_file(dir_path, file_name)
+  File.read(File.join(dir_path, file_name)).strip
+end

spec/document/custom_v1_spec.rb

@@ -0,0 +1,57 @@
+# frozen_string_literal: true
+
+require 'json'
+require 'mindee/parsing'
+
+require_relative '../data'
+
+DIR_CUSTOM_V1 = File.join(DATA_DIR, 'custom', 'response_v1').freeze
+
+describe Mindee::Prediction::CustomV1 do
+  context 'A custom document V1' do
+    it 'should load an empty document prediction' do
+      response = load_json(DIR_CUSTOM_V1, 'empty.json')
+      inference = Mindee::Document.new(Mindee::Prediction::CustomV1, response['document']).inference
+      expect(inference.product.type).to eq('constructed')
+      expect(inference.prediction.fields.length).to eq(10)
+      expect(inference.prediction.classifications.length).to eq(1)
+    end
+
+    it 'should load a complete document prediction' do
+      to_string = read_file(DIR_CUSTOM_V1, 'summary_full.rst')
+      response = load_json(DIR_CUSTOM_V1, 'complete.json')
+      document = Mindee::Document.new(Mindee::Prediction::CustomV1, response['document'])
+      expect(document.to_s).to eq(to_string)
+      prediction = document.inference.prediction
+
+      prediction.fields.each do |field_name, field_data|
+        expect(field_name).to be_kind_of(Symbol)
+        expect(field_data.values).to be_kind_of(Array)
+        expect(field_data.contents_str).to be_kind_of(String)
+      end
+
+      expect(prediction.fields[:string_all].values.size).to eq(3)
+      expect(prediction.fields['string_all']).to be_nil
+      expect(prediction.fields[:string_all].contents_str).to eq('Mindee is awesome')
+      expect(prediction.fields[:string_all].contents_list).to eq(['Mindee', 'is', 'awesome'])
+
+      expect(prediction.classifications[:doc_type].value).to eq('type_b')
+    end
+
+    it 'should load a complete page 0 prediction' do
+      to_string = read_file(DIR_CUSTOM_V1, 'summary_page0.rst')
+      response = load_json(DIR_CUSTOM_V1, 'complete.json')
+      inference = Mindee::Document.new(Mindee::Prediction::CustomV1, response['document']).inference
+      expect(inference.pages[0].prediction.fields[:string_all].contents_str(separator: '_')).to eq('Jenny_is_great')
+      expect(inference.pages[0].prediction.fields[:string_all].contents_list).to eq(['Jenny', 'is', 'great'])
+      expect(inference.pages[0].to_s).to eq(to_string)
+    end
+
+    it 'should load a complete page 1 prediction' do
+      to_string = read_file(DIR_CUSTOM_V1, 'summary_page1.rst')
+      response = load_json(DIR_CUSTOM_V1, 'complete.json')
+      inference = Mindee::Document.new(Mindee::Prediction::CustomV1, response['document']).inference
+      expect(inference.pages[1].to_s).to eq(to_string)
+    end
+  end
+end