Adding Save and All ops to Company.

jkeyes · jkeyes · commit 9adaa4e75310 · 2014-09-25T00:56:13.000+01:00
Adding tests for examples in the README.md
diff --git a/intercom/company.py b/intercom/company.py
@@ -1,9 +1,11 @@
 from intercom.user import Resource
+from intercom.api_operations.all import All
 from intercom.api_operations.count import Count
 from intercom.api_operations.find import Find
 from intercom.api_operations.load import Load
+from intercom.api_operations.save import Save
 
 
-class Company(Resource, Count, Find, Load):
+class Company(Resource, Count, Find, All, Save, Load):
     update_verb = 'post'
     identity_vars = ['id', 'company_id']
diff --git a/intercom/utils.py b/intercom/utils.py
@@ -22,7 +22,7 @@ def constantize_singular_resource_name(resource_name):
 
 
 def resource_class_to_collection_name(cls):
-    return cls.__name__.lower() + "s"
+    return pluralize(cls.__name__.lower())
 
 
 def resource_class_to_name(cls):
diff --git a/tests/integration/test_company.py b/tests/integration/test_company.py
@@ -0,0 +1,92 @@
+# -*- coding: utf-8 -*-
+
+from datetime import datetime
+import os
+import time
+import unittest
+from intercom import Intercom
+from intercom.company import Company
+from intercom.user import User
+
+Intercom.app_id = os.environ.get('INTERCOM_APP_ID')
+Intercom.app_api_key = os.environ.get('INTERCOM_APP_API_KEY')
+
+
+class CompanyTest(unittest.TestCase):
+    email = "ada@example.com"
+
+    @classmethod
+    def setup_class(cls):
+        # get company
+        cls.company = Company.all()[0]
+        # get user
+        cls.user = User.find(email=cls.email)
+        if not hasattr(cls.user, 'user_id'):
+            # Create a user
+            cls.user = User.create(
+                email=cls.email,
+                user_id="ada",
+                name="Ada Lovelace")
+
+    def test_add_user(self):
+        user = User.find(email=self.email)
+        user.companies = [
+            {"company_id": 6, "name": "Intercom"},
+            {"company_id": 9, "name": "Test Company"}
+        ]
+        user.save()
+        user = User.find(email=self.email)
+        self.assertEqual(len(user.companies), 2)
+        self.assertEqual(user.companies[0].company_id, "9")
+
+    def test_add_user_custom_attributes(self):
+        user = User.find(email=self.email)
+        user.companies = [
+            {
+                "id": 6,
+                "name": "Intercom",
+                "custom_attributes": {
+                    "referral_source": "Google"
+                }
+            }
+        ]
+        user.save()
+        user = User.find(email=self.email)
+        self.assertEqual(len(user.companies), 2)
+        self.assertEqual(user.companies[0].company_id, "9")
+
+    def test_find_by_company_id(self):
+        # Find a company by company_id
+        company = Company.find(company_id=self.company.company_id)
+        self.assertEqual(company.company_id, self.company.company_id)
+
+    def test_find_by_company_name(self):
+        # Find a company by name
+        company = Company.find(name=self.company.name)
+        self.assertEqual(company.name, self.company.name)
+
+    def test_find_by_id(self):
+        # Find a company by _id
+        company = Company.find(id=self.company.id)
+        self.assertEqual(company.company_id, self.company.company_id)
+
+    def test_update(self):
+        # Find a company by id
+        company = Company.find(id=self.company.id)
+        # Update a company
+        now = datetime.utcnow()
+        updated_name = 'Company %s' % (time.mktime(now.timetuple()))
+        company.name = updated_name
+        company.save()
+        company = Company.find(id=self.company.id)
+        self.assertEqual(company.name, updated_name)
+
+    def test_iterate(self):
+        # Iterate over all companies
+        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)
