diff --git a/lib/active_admin_import/import_result.rb b/lib/active_admin_import/import_result.rb index 8ff75c4..8d7c57d 100644 --- a/lib/active_admin_import/import_result.rb +++ b/lib/active_admin_import/import_result.rb @@ -1,16 +1,18 @@ # frozen_string_literal: true module ActiveAdminImport class ImportResult - attr_reader :failed, :total + attr_reader :failed, :ids, :total def initialize @failed = [] + @ids = [] @total = 0 end def add(result, qty) @failed += result.failed_instances - @total += qty + @ids += result.ids + @total += qty end def imported_qty diff --git a/spec/import_result_spec.rb b/spec/import_result_spec.rb index 266ceed..32c248a 100644 --- a/spec/import_result_spec.rb +++ b/spec/import_result_spec.rb @@ -14,15 +14,23 @@ ] end + let(:ids){ [1,2] } + before do @result = double \ - failed_instances: failed_instances + failed_instances: failed_instances, + ids: ids end it 'should work without any failed instances' do expect(import_result.failed_message).to eq('') end + it 'should store the supplied ids' do + import_result.add(@result, 4) + expect(import_result.ids).to eq(ids) + end + it 'should work' do import_result.add(@result, 4) expect(import_result.failed_message)