Added rescue block for the test method for the prometheus service

1 files changed, 3 insertions, 2 deletions

diff --git a/spec/models/project_services/prometheus_service_spec.rb b/spec/models/project_services/prometheus_service_spec.rb
index d15079b686b..5ef1b53be15 100644
--- a/spec/models/project_services/prometheus_service_spec.rb
+++ b/spec/models/project_services/prometheus_service_spec.rb
@@ -93,11 +93,12 @@ describe PrometheusService, models: true, caching: true do
 
     [404, 500].each do |status|
       context "when Prometheus responds with #{status}" do
+        body_response = 'QUERY_FAILED'
         before do
-          stub_all_prometheus_requests(environment.slug, status: status, body: 'QUERY FAILED!')
+          stub_all_prometheus_requests(environment.slug, status: status, body: body_response)
         end
 
-        it { is_expected.to eq(success: false, result: %(#{status} - "QUERY FAILED!")) }
+        it { is_expected.to eq(success: false, result: %(#{status} - \"#{body_response}\")) }
       end
     end
   end