Add support to LDAP for external user linking

test/test_ldap.py

@@ -4,6 +4,7 @@ from app import app
 from initdb import setup_database_for_testing, finished_database_for_testing
 from data.users import LDAPUsers
 from mockldap import MockLdap
+from mock import patch
 
 class TestLDAP(unittest.TestCase):
   def setUp(self):
@@ -198,6 +199,60 @@ class TestLDAP(unittest.TestCase):
     (response, _) = self.ldap.confirm_existing_user('someuser', 'somepass')
     self.assertEquals(response.username, 'someuser')
 
+  def test_link_user(self):
+    # Link someuser.
+    user, error_message = self.ldap.link_user('someuser')
+    self.assertIsNone(error_message)
+    self.assertIsNotNone(user)
+    self.assertEquals('someuser', user.username)
+
+    # Link again. Should return the same user record.
+    user_again, _ = self.ldap.link_user('someuser')
+    self.assertEquals(user_again.id, user.id)
+
+    # Confirm someuser.
+    result, _ = self.ldap.confirm_existing_user('someuser', 'somepass')
+    self.assertIsNotNone(result)
+    self.assertEquals('someuser', result.username)
+
+  def test_query(self):
+    def initializer(uri, trace_level=0):
+      obj = self.mockldap[uri]
+
+      # Seed to "support" wildcard queries, which MockLDAP does not support natively.
+      obj.search_s.seed('ou=employees,dc=quay,dc=io', 2, '(|(uid=cool*)(mail=cool*))')([
+        ('uid=cool.user,ou=employees,dc=quay,dc=io', {
+          'dc': ['quay', 'io'],
+          'ou': 'employees',
+          'uid': ['cool.user', 'referred'],
+          'userPassword': ['somepass'],
+          'mail': ['foo@bar.com']
+        })
+      ])
+
+      obj.search_s.seed('ou=otheremployees,dc=quay,dc=io', 2, '(|(uid=cool*)(mail=cool*))')([])
+
+      obj.search_s.seed('ou=employees,dc=quay,dc=io', 2, '(|(uid=unknown*)(mail=unknown*))')([])
+      obj.search_s.seed('ou=otheremployees,dc=quay,dc=io', 2,
+                        '(|(uid=unknown*)(mail=unknown*))')([])
+      return obj
+
+    with patch('ldap.initialize', new=initializer):
+      # Lookup cool.
+      (response, error_message) = self.ldap.query_users('cool')
+      self.assertIsNone(error_message)
+      self.assertEquals(1, len(response))
+
+      user_info = response[0]
+      self.assertEquals("cool.user", user_info.username)
+      self.assertEquals("foo@bar.com", user_info.email)
+
+      # Lookup unknown.
+      (response, error_message) = self.ldap.query_users('unknown')
+      self.assertIsNone(error_message)
+      self.assertEquals(0, len(response))
+
+
 if __name__ == '__main__':
   unittest.main()