Generalize from PDF import to just files

test/controllers/imports_controller_test.rb

@@ -35,6 +35,70 @@ class ImportsControllerTest < ActionDispatch::IntegrationTest
     assert_redirected_to import_upload_url(Import.all.ordered.first)
   end
 
+  test "uploads supported non-pdf document for vector store without creating import" do
+    adapter = mock("vector_store_adapter")
+    adapter.stubs(:supported_extensions).returns(%w[.csv .pdf])
+    VectorStore::Registry.stubs(:adapter).returns(adapter)
+
+    @user.family.expects(:upload_document).with do |file_content:, filename:, **|
+      assert_not_empty file_content
+      assert_equal "valid.csv", filename
+      true
+    end.returns(family_documents(:tax_return))
+
+    assert_no_difference "Import.count" do
+      post imports_url, params: {
+        import: {
+          type: "DocumentImport",
+          import_file: file_fixture_upload("imports/valid.csv", "text/csv")
+        }
+      }
+    end
+
+    assert_redirected_to new_import_url
+    assert_equal I18n.t("imports.create.document_uploaded"), flash[:notice]
+  end
+
+  test "uploads pdf document as PdfImport when using DocumentImport option" do
+    adapter = mock("vector_store_adapter")
+    adapter.stubs(:supported_extensions).returns(%w[.pdf .txt])
+    VectorStore::Registry.stubs(:adapter).returns(adapter)
+
+    @user.family.expects(:upload_document).never
+
+    assert_difference "Import.count", 1 do
+      post imports_url, params: {
+        import: {
+          type: "DocumentImport",
+          import_file: file_fixture_upload("imports/sample_bank_statement.pdf", "application/pdf")
+        }
+      }
+    end
+
+    created_import = Import.order(:created_at).last
+    assert_equal "PdfImport", created_import.type
+    assert_redirected_to import_url(created_import)
+    assert_equal I18n.t("imports.create.pdf_processing"), flash[:notice]
+  end
+
+  test "rejects unsupported document type for DocumentImport option" do
+    adapter = mock("vector_store_adapter")
+    adapter.stubs(:supported_extensions).returns(%w[.pdf .txt])
+    VectorStore::Registry.stubs(:adapter).returns(adapter)
+
+    assert_no_difference "Import.count" do
+      post imports_url, params: {
+        import: {
+          type: "DocumentImport",
+          import_file: file_fixture_upload("profile_image.png", "image/png")
+        }
+      }
+    end
+
+    assert_redirected_to new_import_url
+    assert_equal I18n.t("imports.create.invalid_document_file_type"), flash[:alert]
+  end
+
   test "publishes import" do
     import = imports(:transaction)