def test_check_status_working(self):
"""Test check_status returns True when storage is working properly."""
default_storage_health = DefaultFileStorageHealthCheck()
default_storage = default_storage_health.get_storage()
default_storage_open = '{}.{}.open'.format(
default_storage.__module__,
default_storage.__class__.__name__
)
with mock.patch(default_storage_open, mock.mock_open(read_data=default_storage_health.get_file_content())):
self.assertTrue(default_storage_health.check_status())
def testHealthCheck(self):
request = self.factory.get('/?type=db')
request = Request(request)
# print "debug>>", request.query_params['type'], "<<debug"
plugin = DjangoDatabaseBackend()
# print plugin.status, "debug"
response = healthcheck(request)
response.render()
response_json = json.loads(response.content)
self.assertEqual(response_json['status'], plugin.pretty_status())
request = self.factory.get('/?type=cache')
request = Request(request)
# print "debug>>", request.query_params['type'], "<<debug"
plugin = CacheBackend()
# print plugin.status, "debug"
response = healthcheck(request)
response.render()
response_json = json.loads(response.content)
self.assertEqual(response_json['status'], plugin.pretty_status())
request = self.factory.get('/?type=storage')
request = Request(request)
# print "debug>>", request.query_params['type'], "<<debug"
plugin = DefaultFileStorageHealthCheck()
# print plugin.status, "debug"
response = healthcheck(request)
response.render()
response_json = json.loads(response.content)
self.assertEqual(response_json['status'], plugin.pretty_status())
def healthcheck(request):
plugin = None
if request.query_params['type'] == 'db':
plugin = DjangoDatabaseBackend()
elif request.query_params['type'] == 'cache':
plugin = CacheBackend()
elif request.query_params['type'] == 'storage':
plugin = DefaultFileStorageHealthCheck()
if not plugin.status: # Will return True or None
return Response({"status": plugin.pretty_status()}, status=500)
return Response({"status": plugin.pretty_status()}, status=200)
def test_file_not_deleted(self):
"""Test check_status raises ServiceUnavailable when file is not deleted."""
default_storage_health = DefaultFileStorageHealthCheck()
with self.assertRaises(ServiceUnavailable):
default_storage_health.check_status()
def test_get_storage(self):
"""Test get_storage method returns None on the base class, but a Storage instance on default."""
base_storage = StorageHealthCheck()
self.assertIsNone(base_storage.get_storage())
default_storage = DefaultFileStorageHealthCheck()
self.assertIsInstance(default_storage.get_storage(), Storage)
def test_file_not_deleted(self):
default_storage_health = DefaultFileStorageHealthCheck()
with self.assertRaises(ServiceUnavailable):
default_storage_health.check_status()
def test_check_status_working(self):
default_storage = DefaultFileStorageHealthCheck()
self.assertTrue(default_storage.check_status())
def test_get_storage(self):
base_storage = StorageHealthCheck()
self.assertIsNone(base_storage.get_storage())
default_storage = DefaultFileStorageHealthCheck()
self.assertIsInstance(default_storage.get_storage(), Storage)
