From 920be373011374e74918855e9e3a25f7e000ebff Mon Sep 17 00:00:00 2001
From: Ari Zellner <azellner@redhat.com>
Date: Sun, 26 Feb 2017 14:57:01 +0200
Subject: [PATCH] add enterprise as rate parent of projects

---
 app/models/metric/chargeback_helper.rb        |  2 +-
 .../chargeback_container_project_spec.rb      | 27 +++++++++++++++++++
 2 files changed, 28 insertions(+), 1 deletion(-)

diff --git a/app/models/metric/chargeback_helper.rb b/app/models/metric/chargeback_helper.rb
index 50e39e71580..9c9761e380b 100644
--- a/app/models/metric/chargeback_helper.rb
+++ b/app/models/metric/chargeback_helper.rb
@@ -42,7 +42,7 @@ def parents_determining_rate
     when VmOrTemplate.name
       (resource_parents + [MiqEnterprise.my_enterprise]).compact
     when ContainerProject.name
-      [parent_ems].compact
+      [parent_ems, MiqEnterprise.my_enterprise].compact
     when Container.name
       [parent_ems]
     end
diff --git a/spec/models/chargeback_container_project_spec.rb b/spec/models/chargeback_container_project_spec.rb
index fd554906151..975d4082b90 100644
--- a/spec/models/chargeback_container_project_spec.rb
+++ b/spec/models/chargeback_container_project_spec.rb
@@ -181,4 +181,31 @@
       expect(subject.fixed_compute_metric).to eq(@metric_size / 2)
     end
   end
+
+  context "gets rate from enterprise" do
+    let(:options) { base_options.merge(:interval => 'monthly', :entity_id => @project.id, :tag => nil) }
+    let(:miq_enterprise) { FactoryGirl.create(:miq_enterprise) }
+
+    before do
+      add_metric_rollups_for(@project, month_beginning...month_end, 24.hours, metric_rollup_params)
+
+      metric_rollup_params[:cpu_usage_rate_average] = 0.0
+      metric_rollup_params[:derived_memory_used] = 0.0
+
+      add_metric_rollups_for(@project, (month_beginning + 12.hours)...month_end, 24.hours, metric_rollup_params, nil)
+
+      @metric_size = @project.metric_rollups.size
+
+      temp = {:cb_rate => chargeback_rate, :object => miq_enterprise}
+      ChargebackRate.set_assignments(:compute, [temp])
+    end
+
+    subject { ChargebackContainerProject.build_results_for_report_ChargebackContainerProject(options).first.first }
+
+    it "fixed_compute" do
+      # .to be_within(0.01) is used since theres a float error here
+      expect(subject.fixed_compute_1_cost).to be_within(0.01).of(hourly_rate * hours_in_month)
+      expect(subject.fixed_compute_metric).to eq(@metric_size / 2)
+    end
+  end
 end