Adding custom attribute support to Company.

jkeyes · jkeyes · commit 1ec07c141f94 · 2015-03-27T14:17:17.000Z
diff --git a/intercom/company.py b/intercom/company.py
@@ -13,3 +13,7 @@
 class Company(Resource, Delete, Count, Find, All, Save, Load, Users):
     update_verb = 'post'
     identity_vars = ['id', 'company_id']
+
+    @property
+    def flat_store_attributes(self):
+        return ['custom_attributes']
diff --git a/tests/integration/test_company.py b/tests/integration/test_company.py
@@ -54,6 +54,11 @@ def test_add_user_custom_attributes(self):
         self.assertEqual(len(user.companies), 2)
         self.assertEqual(user.companies[0].company_id, "9")
 
+        # check the custom attributes
+        company = Company.find(company_id=6)
+        self.assertEqual(
+            company.custom_attributes['referral_source'], "Google")
+
     def test_find_by_company_id(self):
         # Find a company by company_id
         company = Company.find(company_id=self.company.company_id)
