From cef457949309ee9484300a3d4710fc7ac0d026bc Mon Sep 17 00:00:00 2001 From: Elephant Liu Date: Tue, 23 Aug 2016 20:00:56 +0800 Subject: [PATCH] add test for django force_login about cache of user attr --- tests/test_testing.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/tests/test_testing.py b/tests/test_testing.py index f80859f0c..b71dd5b99 100644 --- a/tests/test_testing.py +++ b/tests/test_testing.py @@ -237,6 +237,24 @@ class TestAPITestClient(TestCase): response = self.client.get('/my_hat/') self.assertEqual(response.status_code, 404) + def test_get_my_hat_with_django_force_login(self): + user = User.objects.create_user('example', 'example@example.com', 'password') + self.client.force_login(user) + + response = self.client.get('/my_hat/') + self.assertEqual(response.status_code, 404) + + hat = Hat.objects.create(user=user, some_key='some_value') + response = self.client.get('/my_hat/') + self.assertEqual(response.status_code, 200) + self.assertEqual(response.data, {'id': hat.id, 'some_key': 'some_value'}) + + another_user = User.objects.create_user('another_example', 'another_example@example.com', 'password') + hat.user = another_user + hat.save() + response = self.client.get('/my_hat/') + self.assertEqual(response.status_code, 404) + def test_get_my_hat_with_login(self): user = User.objects.create_user('example', 'example@example.com', 'password') self.client.login(username='example', password='password')