From d9f7110c3998b1b61ef4707914807ea1060e2b92 Mon Sep 17 00:00:00 2001
From: Chintalagiri Shashank <shashank.chintalagiri@gmail.com>
Date: Fri, 12 Apr 2024 18:39:13 +0530
Subject: [PATCH] Make test_grafana_client_non_json_response more permissive.
 Update test_grafana_client_no_verify to not break once params are added in.

---
 test/test_grafana_client.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/test/test_grafana_client.py b/test/test_grafana_client.py
index e3c6f9c..38013a7 100644
--- a/test/test_grafana_client.py
+++ b/test/test_grafana_client.py
@@ -6,6 +6,7 @@
 from grafana_client.api import GrafanaApi
 from grafana_client.client import (
     GrafanaClientError,
+    GrafanaServerError,
     GrafanaTimeoutError,
     HeaderAuth,
     TokenAuth,
@@ -95,6 +96,7 @@ def test_grafana_client_no_verify(self):
             auth=basic_auth,
             headers=None,
             json=None,
+            params=None,
             data=None,
             verify=False,
             timeout=5.0,
@@ -160,7 +162,7 @@ def test_grafana_client_version(self, mock_get):
 
     def test_grafana_client_non_json_response(self):
         grafana = GrafanaApi.from_url("https://example.org/")
-        self.assertRaises(GrafanaClientError, lambda: grafana.connect())
+        self.assertRaises((GrafanaClientError, GrafanaServerError), lambda: grafana.connect())
 
     def test_grafana_client_204_no_content_response(self):
         grafana = GrafanaApi.from_url()