updated and cleaned up Gravatar

[hermes] / package / test / integration / test_gravatar.py

diff --git a/package/test/integration/test_gravatar.py b/package/test/integration/test_gravatar.py
index 5f828e6..f4b3391 100644 (file)
--- a/package/test/integration/test_gravatar.py
+++ b/package/test/integration/test_gravatar.py
@@ -31,11 +31,14 @@ class IntegrationTestGravatarService(unittest.TestCase):
         self.testee.pre_process_contact(self.existing_contact)
         self.testee.pre_process_contact(self.missing_contact)
         self.testee.process_friends()
-        self.testee.process_contact(self.existing_contact)
-        self.testee.process_contact(self.missing_contact)
+        existing_friend = self.testee.process_contact(self.existing_contact)
+        missing_friend = self.testee.process_contact(self.missing_contact)
         
-        friends = self.testee.get_friends()
-        contacts = self.testee.get_contacts_with_match()
+        assert isinstance(existing_friend, Friend)
+        assert missing_friend is None
+        
+        friends = self.testee._get_friends()
+        contacts = self.testee._get_contacts_with_match()
         assert len(friends) == 1
         assert len(contacts) == 1
         assert self.missing_contact not in contacts.keys()