from integration_api import USERNAME, get_authed_instance from integration_api import USERMSGS_URL, USERMSGS_COUNT_URL import unittest class UserMsgsApiTestCase(unittest.TestCase): def setUp(self): self.requests = get_authed_instance() self.assertIsNotNone(self.requests) def testListUserMsgsApi(self): res = self.requests.get(USERMSGS_URL) self.assertEqual(res.status_code, 200) json = res.json() self.assertIsNotNone(json) self.assertEqual(json['to_email'], USERNAME) self.assertIsNotNone(json['next_page']) self.assertIsNotNone(json['msgs']) def testReplyUserMsgApi(self): data = { 'id': '0', 'message': 'test' } res = self.requests.post(USERMSGS_URL, data=data) self.assertEqual(res.status_code, 200) json = res.json() self.assertIsNotNone(json) self.assertIsNotNone(json['msgid']) def testCountUnseenMsgsApi(self): data = { 'id': '0', 'message': 'test' } self.requests.post(USERMSGS_URL, data=data) res = self.requests.get(USERMSGS_COUNT_URL, data=data) self.assertEqual(res.status_code, 200) json = res.json() self.assertIsNotNone(json) self.assertGreater(json['count'], 0) if __name__ == '__main__': unittest.main(verbosity=2)