Adding users extended API operation.

jkeyes · jkeyes · commit aa876e078482 · 2014-09-25T22:09:59.000+01:00
diff --git a/intercom/company.py b/intercom/company.py
@@ -4,8 +4,9 @@
 from intercom.api_operations.find import Find
 from intercom.api_operations.load import Load
 from intercom.api_operations.save import Save
+from intercom.extended_api_operations.users import Users
 
 
-class Company(Resource, Count, Find, All, Save, Load):
+class Company(Resource, Count, Find, All, Save, Load, Users):
     update_verb = 'post'
     identity_vars = ['id', 'company_id']
diff --git a/intercom/extended_api_operations/__init__.py b/intercom/extended_api_operations/__init__.py
@@ -0,0 +1 @@
+# -*- coding: utf-8 -*-
diff --git a/intercom/extended_api_operations/users.py b/intercom/extended_api_operations/users.py
@@ -0,0 +1,14 @@
+# -*- coding: utf-8 -*-
+
+from intercom import utils
+from intercom.user import User
+from intercom.collection_proxy import CollectionProxy
+
+
+class Users(object):
+
+    @property
+    def users(self):
+        collection = utils.resource_class_to_collection_name(self.__class__)
+        finder_url = "/%s/%s/users" % (collection, self.id)
+        return CollectionProxy(User, "users", finder_url)
diff --git a/tests/integration/test_company.py b/tests/integration/test_company.py
@@ -86,7 +86,8 @@ def test_iterate(self):
         for company in Company.all():
             self.assertTrue(company.id is not None)
 
-    # def test_users(self):
-    #     company = Company.find(id=self.company.id)
-    #     # Get a list of users in a company
-    #     self.assertIsNotNone(company.users)
+    def test_users(self):
+        company = Company.find(id=self.company.id)
+        # Get a list of users in a company
+        for user in company.users:
+            self.assertIsNotNone(user.email)
