fix spec for ConsumptionWithRollups

ManageIQ · Jun 14, 2018 · 541dfe7 · 541dfe7
1 parent 4875cb3
commit 541dfe7
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/spec/models/chargeback/consumption_with_rollups_spec.rb b/spec/models/chargeback/consumption_with_rollups_spec.rb
@@ -1,6 +1,6 @@
 describe Chargeback::ConsumptionWithRollups do
-  let(:vm)          { FactoryGirl.build(:vm_microsoft) }
-  let(:consumption) { described_class.new([metric_rollup], starting_date, starting_date + 1.day) }
+  let(:vm)          { FactoryGirl.create(:vm_microsoft) }
+  let(:consumption) { described_class.new(pluck_rollup([metric_rollup]), starting_date, starting_date + 1.day) }
 
   describe '#sub_metric_rollups' do
     let(:starting_date)        { Time.parse('2012-09-01 23:59:59Z').utc }
@@ -9,6 +9,10 @@
     let!(:state)         { FactoryGirl.create(:vim_performance_state, :resource => vm, :state_data => nil, :timestamp => starting_date, :capture_interval => 3_600) }
     let!(:metric_rollup) { FactoryGirl.create(:metric_rollup_vm_hr, :timestamp => starting_date + 1.hour, :resource => vm) }
 
+    def pluck_rollup(metric_rollup_records)
+      metric_rollup_records.pluck(*ChargeableField.cols_on_metric_rollup)
+    end
+
     before do
       Timecop.travel(starting_date + 10.hours)
     end