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)