0
@@ -3,12 +3,22 @@ class Report < ActiveRecord::Base
0
   validates_presence_of :code, :name, :kind
0
   validates_uniqueness_of :name
0
   
0
-  
0
   def run
0
     if kind == 'sql'
0
-      ActiveRecord::Base.connection.select_all(code).map(&:values)
0
+      rows = ActiveRecord::Base.connection.select_all(code)
0
+      header(rows) + rows.map(&:values)
0
     else
0
       eval(code)
0
     end
0
   end
0
+  
0
+  private
0
+  
0
+  def header(rows)
0
+    if rows.first
0
+      [rows.first.map{|k,v| k}]
0
+    else
0
+      []
0
+    end
0
+  end
0
 end

0
@@ -18,7 +18,13 @@ describe Report do
0
   
0
   it "should run the sql and return an array of arrays" do
0
     @report.kind = 'sql'
0
-    @report.code = 'SELECT 1, 2 UNION SELECT 2, 3'
0
-    @report.run.should == [['1', '2'], ['2', '3']]
0
+    @report.code = 'SELECT 1 AS col_1, 2 AS col_2 UNION SELECT 2 AS col_1, 3 AS col_3'
0
+    @report.run.should == [['col_1', 'col_2'], ['1', '2'], ['2', '3']]
0
+  end
0
+  
0
+  it "should return an empty array if no results" do
0
+    @report.kind = 'sql'
0
+    @report.code = 'SELECT 1 WHERE 0'
0
+    @report.run.should == []
0
   end
0
 end