def test_get_all_cached_users_from_db(self):
"""Method to test get_all_cached_users when data is not present in the cache"""
get_user_model().objects.create_user(username='******',
password='******')
cache.clear()
self.assertEqual(cache.get(CACHE_KEYS['all_users']), None)
test_cache = TheCacher.get_all_cached_users()
self.assertEqual(test_cache.__class__.__name__, 'QuerySet')
def view_users(request):
""" View User """
users = TheCacher.get_all_cached_users()
if users is not None:
serializer = UserSerializer(users, many=True)
response = Response({'users': serializer.data}, status=200)
else:
response = Response({'error': 'Unknown error'}, status=500)
return response
def test_get_all_cached_users_when_cache_is_present(self):
"""Method to test get_all_cached_users when data is present in the cache"""
cache.set(CACHE_KEYS['all_users'], True)
test_cache = TheCacher.get_all_cached_users()
self.assertEqual(test_cache, True)
cache.clear()