Query#to_csv

tessro · May 6, 2012 · 967af6b · 967af6b
1 parent 2f45fba
commit 967af6b
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/lib/oculus/query.rb b/lib/oculus/query.rb
@@ -56,6 +56,14 @@ def succeeded?
       complete? && !error
     end
 
+    def to_csv
+      CSV.generate do |csv|
+        results.each do |row|
+          csv << row
+        end
+      end
+    end
+
     class << self
       def create(attributes)
         query = new(attributes)

diff --git a/spec/query_spec.rb b/spec/query_spec.rb
@@ -137,4 +137,10 @@
     query.execute(connection)
     query.succeeded?.should be false
   end
+
+  it "exports to CSV" do
+    query = Oculus::Query.new
+    query.results = [['id', 'name'], [1, 'Paul']]
+    query.to_csv.should == "id,name\n1,Paul\n"
+  end
 end