Try different way to mock redis ping

This commit is contained in:
shamoon 2024-02-13 00:41:09 -08:00
parent 318ef61dfa
commit eae506889e

View File

@ -52,16 +52,9 @@ class TestSystemStatusView(APITestCase):
response = self.client.get(self.ENDPOINT) response = self.client.get(self.ENDPOINT)
self.assertEqual(response.data["install_type"], "kubernetes") self.assertEqual(response.data["install_type"], "kubernetes")
class MockRedis: @mock.patch("redis.Redis.execute_command")
def from_url(self, url):
pass
def ping(self):
return True
@mock.patch("redis.Redis")
def test_system_status_redis_ping(self, mock_ping): 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) self.client.force_login(self.user)
response = self.client.get(self.ENDPOINT) response = self.client.get(self.ENDPOINT)
self.assertEqual(response.status_code, status.HTTP_200_OK) self.assertEqual(response.status_code, status.HTTP_200_OK)