add export method

next-l · Dec 29, 2019 · a42faa1 · a42faa1
1 parent e290c26
commit a42faa1
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/app/models/inventory_file.rb b/app/models/inventory_file.rb
@@ -39,6 +39,19 @@ def import
     true
   end
 
+  def export(col_sep: "\t")
+    file = Tempfile.create do |f|
+      inventories.each do |inventory|
+        f.write inventory.to_hash.values.to_csv(col_sep)
+      end
+
+      f.rewind
+      f.read
+    end
+
+    file
+  end
+
   def missing_items
     Item.where(Inventory.where('items.id = inventories.item_id AND inventories.inventory_file_id = ?', id).exists.not)
   end

diff --git a/spec/models/inventory_file_spec.rb b/spec/models/inventory_file_spec.rb
@@ -12,6 +12,10 @@
   it "should be imported" do
     expect(@file.import).to be_truthy
   end
+
+  it "should export results" do
+    expect(@file.export).to be_truthy
+  end
 end
 
 # == Schema Information