From eae506889e2e823a62830978a6262996bf5fd988 Mon Sep 17 00:00:00 2001
From: shamoon <4887959+shamoon@users.noreply.github.com>
Date: Tue, 13 Feb 2024 00:41:09 -0800
Subject: [PATCH] Try different way to mock redis ping

---
 src/documents/tests/test_api_status.py | 11 ++---------
 1 file changed, 2 insertions(+), 9 deletions(-)

diff --git a/src/documents/tests/test_api_status.py b/src/documents/tests/test_api_status.py
index c2dad2914..9a1859585 100644
--- a/src/documents/tests/test_api_status.py
+++ b/src/documents/tests/test_api_status.py
@@ -52,16 +52,9 @@ class TestSystemStatusView(APITestCase):
         response = self.client.get(self.ENDPOINT)
         self.assertEqual(response.data["install_type"], "kubernetes")
 
-    class MockRedis:
-        def from_url(self, url):
-            pass
-
-        def ping(self):
-            return True
-
-    @mock.patch("redis.Redis")
+    @mock.patch("redis.Redis.execute_command")
     def test_system_status_redis_ping(self, mock_ping):
-        mock_ping.return_value = self.MockRedis()
+        mock_ping.return_value = True
         self.client.force_login(self.user)
         response = self.client.get(self.ENDPOINT)
         self.assertEqual(response.status_code, status.HTTP_200_OK)