class TokenAuthTestCase(BaseRestTestCase): def _assert_status_check(self, status_code, partner_id=None, token=None): c = Client() # good token resp = c.post( '/sms/status/9000/', { 'id': partner_id or self.partner.id, 'token': token or self.partner.token, }) self.assertEqual(resp.status_code, status_code) def setUp(self): super(TokenAuthTestCase, self).setUp() self.backend = PartnerTokenBackend() def test_backend_authenticate(self): u = self.backend.authenticate(id=self.partner_user.id, token=self.partner.token) self.assertEqual(u, self.partner_user) def test_full_cycle_authnticate(self): # good token self._assert_status_check(404) # bad token self._assert_status_check(403, token='bad_token')
class TokenAuthTestCase(BaseRestTestCase): def _assert_status_check(self, status_code, partner_id=None, token=None): c = Client() # good token resp = c.post('/sms/status/9000/', { 'id': partner_id or self.partner.id, 'token': token or self.partner.token, }) self.assertEqual(resp.status_code, status_code) def setUp(self): super(TokenAuthTestCase, self).setUp() self.backend = PartnerTokenBackend() def test_backend_authenticate(self): u = self.backend.authenticate(id=self.partner_user.id, token=self.partner.token) self.assertEqual(u, self.partner_user) def test_full_cycle_authnticate(self): # good token self._assert_status_check(404) # bad token self._assert_status_check(403, token='bad_token')
def setUp(self): super(TokenAuthTestCase, self).setUp() self.backend = PartnerTokenBackend()