class AuthMiddlewareTestCase(TestCase): def setUp(self): self.user = User.objects.create_user(username='******', password='******') self.token = ModelBackend().create_token(self.user) self.bad_token = self.token.lower() self.log = io.StringIO() self.handler = logging.StreamHandler(self.log) self.logger = logging.getLogger('sesame') self.logger.addHandler(self.handler) def tearDown(self): self.logger.removeHandler(self.handler) def test_token(self): response = self.client.get('/', {'url_auth_token': self.token}) self.assertEqual(response.content, b'john') def test_bad_token(self): response = self.client.get('/', {'url_auth_token': self.bad_token}) self.assertEqual(response.content, b'anonymous') def test_no_token(self): response = self.client.get('/') self.assertEqual(response.content, b'anonymous')