Add test for callable saml attribute mapping

Author: joetsoi

tests/testprofiles/tests.py

@@ -68,3 +68,24 @@ def test_update_user(self):
             self.assertEquals(user.get_profile().age, '22')
         else:
             self.assertEquals(user.age, '22')
+
+    def test_update_user_callable_attributes(self):
+        user = User.objects.create(username='john')
+
+        backend = Saml2Backend()
+        attribute_mapping = {
+            'uid': ('username', ),
+            'mail': ('email', ),
+            'cn': ('process_first_name', ),
+            'sn': ('last_name', ),
+            }
+        attributes = {
+            'uid': ('john', ),
+            'mail': ('[email protected]', ),
+            'cn': ('John', ),
+            'sn': ('Doe', ),
+            }
+        backend.update_user(user, attributes, attribute_mapping)
+        self.assertEquals(user.email, '[email protected]')
+        self.assertEquals(user.first_name, 'John')
+        self.assertEquals(user.last_name, 'Doe')