Merge pull request #6 from loadsmart/dealWithMissingSubject

Return none if no subject set

Author: luizguilhermefr

gmail_wrapper/entities.py

@@ -79,7 +79,7 @@ def id(self):
 
     @property
     def subject(self):
-        return self.headers["Subject"]
+        return self.headers.get("Subject")
 
     @property
     def date(self):

tests/test_entities.py

@@ -35,6 +35,11 @@ def test_it_fetch_additional_information_when_needed(
         )
         mocked_get_raw_message.assert_called_once_with(raw_incomplete_message["id"])
 
+    def test_it_return_none_if_no_subject_header(self, client, raw_complete_message):
+        del raw_complete_message["payload"]["headers"][2]
+        message = Message(client, raw_complete_message)
+        assert message.subject is None
+
     def test_get_labels(
         self, mocker, raw_complete_message, client, raw_incomplete_message
     ):