allow change headers in callabcks, specs added

activeadmin-plugins · Mar 18, 2015 · 771d851 · 771d851
1 parent cb44dc2
commit 771d851
Show file tree

Hide file tree

Showing 4 changed files with 107 additions and 77 deletions.
diff --git a/lib/active_admin_import/importer.rb b/lib/active_admin_import/importer.rb
@@ -2,7 +2,8 @@
 module ActiveAdminImport
   class Importer
 
-    attr_reader :resource, :options, :result, :headers, :csv_lines, :model
+    attr_reader :resource, :options, :result, :model
+    attr_accessor :csv_lines, :headers
 
     OPTIONS = [
         :validate,
@@ -85,7 +86,7 @@ def process_file
     end
 
     def prepare_headers
-      headers = self.headers.present? ? self.headers : yield
+      headers = self.headers.present? ? self.headers.map(&:to_s) : yield
       @headers = Hash[headers.zip(headers.map { |el| el.underscore.gsub(/\s+/, '_') })].with_indifferent_access
       @headers.merge!(options[:headers_rewrites].symbolize_keys.slice(*@headers.symbolize_keys.keys))
       @headers

diff --git a/spec/fixtures/files/posts_for_author.csv b/spec/fixtures/files/posts_for_author.csv
@@ -0,0 +1,3 @@
+"Title","Body"
+"title1","some body"
+"title2","some body"
diff --git a/spec/fixtures/files/posts_for_author_no_headers.csv b/spec/fixtures/files/posts_for_author_no_headers.csv
@@ -0,0 +1,2 @@
+"title1","some body"
+"title2","some body"