feat: replaced exit with ValueError (#182)

Author: anushkrishnav

prometheus_api_client/prometheus_connect.py

@@ -2,7 +2,6 @@
 from urllib.parse import urlparse
 import bz2
 import os
-import sys
 import json
 import logging
 import numpy
@@ -190,7 +189,7 @@ def get_metric_range_data(
         end = round(end_time.timestamp())
 
         if (end_time - start_time).total_seconds() < chunk_size.total_seconds():
-            sys.exit("specified chunk_size is too big")
+            raise ValueError("specified chunk_size is too big")
         chunk_seconds = round(chunk_size.total_seconds())
 
         if label_config:

tests/test_prometheus_connect.py

@@ -88,6 +88,13 @@ def test_get_metric_range_data_with_incorrect_input_types(self):  # noqa D102
         chunk_size = timedelta(minutes=7)
         end_time = datetime.now() - timedelta(minutes=10)
 
+        with self.assertRaises(ValueError, msg="specified chunk_size is too big"):
+            _ = self.pc.get_metric_range_data(
+                metric_name="up",
+                start_time=start_time,
+                end_time=end_time,
+                chunk_size=timedelta(minutes=30),
+            )
         with self.assertRaises(TypeError, msg="start_time accepted invalid value type"):
             _ = self.pc.get_metric_range_data(
                 metric_name="up", start_time="20m", end_time=end_time, chunk_size=chunk_size