Move Prometheus service to project model

1 files changed, 4 insertions, 3 deletions

diff --git a/spec/models/deployment_spec.rb b/spec/models/deployment_spec.rb
index eba0175c54c..bb84d3fc13d 100644
--- a/spec/models/deployment_spec.rb
+++ b/spec/models/deployment_spec.rb
@@ -30,7 +30,7 @@ describe Deployment, models: true do
   end
 
   describe '#includes_commit?' do
-    let(:project)     { create(:project, :repository) }
+    let(:project) { create(:project, :repository) }
     let(:environment) { create(:environment, project: project) }
     let(:deployment) do
       create(:deployment, environment: environment, sha: project.commit.id)
@@ -91,7 +91,8 @@ describe Deployment, models: true do
   end
 
   describe '#additional_metrics' do
-    let(:deployment) { create(:deployment) }
+    let(:project) { create(:project) }
+    let(:deployment) { create(:deployment, project: project) }
 
     subject { deployment.additional_metrics }
 
@@ -111,7 +112,7 @@ describe Deployment, models: true do
       let(:prometheus_service) { double('prometheus_service') }
 
       before do
-        allow(deployment).to receive(:prometheus_service).and_return(prometheus_service)
+        allow(project).to receive(:prometheus_service).and_return(prometheus_service)
         allow(prometheus_service).to receive(:additional_deployment_metrics).and_return(simple_metrics)
       end