Merge pull request #175 from linqueta/add-active-admin-import-exception

Add ActiveAdminImport::Exception

Author: Fivell

lib/active_admin_import.rb

@@ -3,6 +3,7 @@
 require 'active_admin'
 require 'active_admin_import/version'
 require 'active_admin_import/engine'
+require 'active_admin_import/exception'
 require 'active_admin_import/import_result'
 require 'active_admin_import/options'
 require 'active_admin_import/dsl'

lib/active_admin_import/dsl.rb

@@ -92,7 +92,11 @@ def active_admin_import(options = {}, &block)
           else
             instance_exec result, options, &DEFAULT_RESULT_PROC
           end
-        rescue ActiveRecord::Import::MissingColumnError, NoMethodError, ActiveRecord::StatementInvalid, CSV::MalformedCSVError => e
+        rescue ActiveRecord::Import::MissingColumnError,
+               NoMethodError,
+               ActiveRecord::StatementInvalid,
+               CSV::MalformedCSVError,
+               ActiveAdminImport::Exception => e
           Rails.logger.error(I18n.t('active_admin_import.file_error', message: e.message))
           Rails.logger.error(e.backtrace.join("\n"))
           flash[:error] = I18n.t('active_admin_import.file_error', message: e.message[0..200])

lib/active_admin_import/exception.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+module ActiveAdminImport
+  class Exception < StandardError
+  end
+end

spec/import_spec.rb

@@ -430,6 +430,28 @@ def upload_file!(name, ext = 'csv')
         upload_file!(:authors)
         expect(Author.count).to eq(2)
       end
+
+      context 'when the option before_import raises a ActiveAdminImport::Exception' do
+        let(:options) { { before_import: ->(_) { raise ActiveAdminImport::Exception, 'error message' } } }
+
+        before { upload_file!(:authors) }
+
+        it 'should show error' do
+          expect(page).to have_content I18n.t('active_admin_import.file_error', message: 'error message')
+          expect(Author.count).to eq(0)
+        end
+      end
+
+      context 'when the option before_batch_import raises a ActiveAdminImport::Exception' do
+        let(:options) { { before_batch_import: ->(_) { raise ActiveAdminImport::Exception, 'error message' } } }
+
+        before { upload_file!(:authors) }
+
+        it 'should show error' do
+          expect(page).to have_content I18n.t('active_admin_import.file_error', message: 'error message')
+          expect(Author.count).to eq(0)
+        end
+      end
     end
   end