LinkedIn.process_contact returns Friend or None (inherited, just added test)

Signed-off-by: Fredrik Wendt <fredrik@wendt.se>

diff --git a/package/test/unit/test_linkedin.py b/package/test/unit/test_linkedin.py
index 55b7424..63b57f7 100644 (file)
--- a/package/test/unit/test_linkedin.py
+++ b/package/test/unit/test_linkedin.py
@@ -35,6 +35,28 @@ class TestLinkedInService(unittest.TestCase):
         self.linkedInApi = FakeLinkedInApi()
         self.testee = Service(self.linkedInApi)
 
+
+    def test_that_process_contact_returns_None_for_unknown_contact(self):
+        contact = FakeContact('Person Person', [])
+        self._fake_server_response([])
+        
+        self.testee.pre_process_contact(contact);
+        self.testee.process_friends()
+        friend = self.testee.process_contact(contact)
+        assert friend is None
+        
+
+    def test_that_process_contact_returns_friend_object_if_contact_is_known(self):
+        same_name = "A Name"
+        contact = FakeContact(same_name, [])
+        fake_friend = Friend(same_name)
+        self._fake_server_response([fake_friend])
+        
+        self.testee.pre_process_contact(contact);
+        self.testee.process_friends()
+        friend = self.testee.process_contact(contact)
+        assert isinstance(friend, Friend)
+
         
     def test_main_flow_one_match_by_url_one_by_name(self):
         known_url = "http://linkedin/id=1"