class GmailRestHandlerTestCase(TestCase):
def setUp(self):
self.gmail = GmailRestHandler('demo_config.py')
def tearDown(self):
self.gmail = None
def test_build_service_returns_resource(self):
"""Test that method returns discovery.Resource"""
service = self.gmail._build_service()
self.assertIsInstance(service, Resource)
def test_get_message(self):
"""Test that method returns message data"""
message = self.gmail.get_message(USER_ID, MESSAGE_ID)
self.assertIsInstance(message, dict)
def test_list_messages_matching_query(self):
"""Test that method returns data"""
message_list = self.gmail.list_messages_matching_query(USER_ID, 'Label_13')
self.assertIsNotNone(message_list)
def test_list_messages_with_labels(self):
"""Test that method returns data"""
message_list = self.gmail.list_messages_with_labels(USER_ID, ['Label_13'])
self.assertIsNotNone(message_list)
def test_modify_message(self):
"""Test that method returns data"""
#message = self.gmail.modify_message(USER_ID, MESSAGE_ID, MESSAGE_LABELS)
# Do not test as can't add/remove same label; don't want to clutter inbox
pass
def test_list_label(self):
"""Test that method returns labels"""
labels = self.gmail.list_labels(USER_ID)
self.assertIsNotNone(labels)
def setUp(self):
self.gmail = GmailRestHandler('demo_config.py')
