class SnapchatTestCase(unittest.TestCase): def setUp(self): responses.reset() self.snapchat = Snapchat() @responses.activate def test_login(self): login_url = AUTH_URL + 'loq/login' responses.add(responses.POST, login_url, body='{"updates_response": \ {"auth_token":"123","username":"******"}}', status=200, content_type='application/json') self.assertEqual({"updates_response": { "auth_token": "123", "username": "******"}}, self.snapchat.login('eggs', 'spam')) responses.reset() responses.add(responses.POST, login_url, body='{}', status=404, content_type='application/json') self.assertRaises(HTTPError, self.snapchat.login, 'eggs', 'spam') responses.reset() responses.add(responses.POST, login_url, body='', status=200, content_type='application/json') self.assertRaises(ValueError, self.snapchat.login, 'eggs', 'spam') responses.reset() responses.add(responses.POST, login_url, body='{"status":"401","message":"Incorrect password"}', status=200, content_type='application/json') self.assertRaises(Exception, self.snapchat.login, 'eggs', 'spam') @responses.activate def test_logout(self): responses.add(responses.POST, URL + 'bq/logout', body='', status=200, content_type='application/json') self.assertTrue(self.snapchat.logout()) responses.reset() responses.add(responses.POST, URL + 'bq/logout', body='{}', status=404, content_type='application/json') self.assertRaises(HTTPError, self.snapchat.logout) responses.reset() responses.add(responses.POST, URL + 'bq/logout', body='{}', status=200, content_type='application/json') self.assertFalse(self.snapchat.logout())
class SnapchatTestCase(unittest.TestCase): def setUp(self): responses.reset() self.snapchat = Snapchat() @responses.activate def test_login(self): responses.add(responses.POST, URL + 'login', body='{}', status=200, content_type='application/json') self.assertEqual({}, self.snapchat.login('eggs', 'spam')) responses.reset() responses.add(responses.POST, URL + 'login', body='{}', status=404, content_type='application/json') self.assertRaises(HTTPError, self.snapchat.login, 'eggs', 'spam') responses.reset() responses.add(responses.POST, URL + 'login', body='', status=200, content_type='application/json') self.assertRaises(ValueError, self.snapchat.login, 'eggs', 'spam') @responses.activate def test_logout(self): responses.add(responses.POST, URL + 'logout', body='', status=200, content_type='application/json') self.assertTrue(self.snapchat.logout()) responses.reset() responses.add(responses.POST, URL + 'logout', body='{}', status=404, content_type='application/json') self.assertRaises(HTTPError, self.snapchat.logout) responses.reset() responses.add(responses.POST, URL + 'logout', body='{}', status=200, content_type='application/json') self.assertFalse(self.snapchat.logout())