Rounding out tag support.

jkeyes · jkeyes · commit 0285ddd6a9f0 · 2015-03-27T14:19:05.000Z
diff --git a/intercom/tag.py b/intercom/tag.py
@@ -48,12 +48,21 @@ def untag_companies(cls, name, companies):
         return cls._tag_collection(name, 'companies', companies, untag=True)
 
     @classmethod
-    def find_all_for_user(cls, id=None, email=None, user_id=None):
-        params = {}
-        if id:
-            params['id'] = id
-        if email:
-            params['email'] = email
-        if user_id:
-            params['user_id'] = user_id
+    def find_all_for_user(cls, **kwargs):
+        return cls.find_all_for('user', **kwargs)
+
+    @classmethod
+    def find_all_for_company(cls, **kwargs):
+        return cls.find_all_for('company', **kwargs)
+
+    @classmethod
+    def find_all_for(cls, taggable_type, **kwargs):
+        params = {
+            'taggable_type': taggable_type
+        }
+        res_id = kwargs.pop('id', None)
+        if res_id:
+            params['taggable_id'] = res_id
+        params.update(kwargs)
+
         return Tag.find_all(**params)
diff --git a/tests/integration/test_tags.py b/tests/integration/test_tags.py
@@ -4,6 +4,8 @@
 import unittest
 from intercom import Intercom
 from intercom import Tag
+from intercom import User
+from intercom import Company
 from . import delete
 from . import get_or_create_company
 from . import get_or_create_user
@@ -32,47 +34,68 @@ def teardown_class(cls):
 
     def test_tag_users(self):
         # Tag users
-        tag = Tag.tag_users("blue", [self.user.id])
-        self.assertEqual(tag.name, "blue")
+        tag = Tag.tag_users('blue', [self.user.id])
+        self.assertEqual(tag.name, 'blue')
+        user = User.find(email=self.user.email)
+        self.assertEqual(1, len(user.tags))
 
     def test_untag_users(self):
         # Untag users
-        tag = Tag.untag_users("blue", [self.user.id])
-        self.assertEqual(tag.name, "blue")
+        tag = Tag.untag_users('blue', [self.user.id])
+        self.assertEqual(tag.name, 'blue')
+        user = User.find(email=self.user.email)
+        self.assertEqual(0, len(user.tags))
 
     def test_all(self):
         # Iterate over all tags
         for tag in Tag.all():
             self.assertIsNotNone(tag.id)
 
-    # def test_all_for_user_by_id(self):
-    #     # Iterate over all tags for user
-    #     tags = Tag.find_all_for_user(id=self.user.id)
-    #     for tag in tags:
-    #         self.assertIsNotNone(tag.id)
+    def test_all_for_user_by_id(self):
+        # Iterate over all tags for user
+        tags = Tag.find_all_for_user(id=self.user.id)
+        for tag in tags:
+            self.assertIsNotNone(tag.id)
 
-    # def test_all_for_user_by_email(self):
-    #     # Iterate over all tags for user
-    #     tags = Tag.find_all_for_user(email=self.user.email)
-    #     for tag in tags:
-    #         self.assertIsNotNone(tag.id)
+    def test_all_for_user_by_email(self):
+        # Iterate over all tags for user
+        tags = Tag.find_all_for_user(email=self.user.email)
+        for tag in tags:
+            self.assertIsNotNone(tag.id)
 
-    # def test_all_for_user_by_user_id(self):
-    #     # Iterate over all tags for user
-    #     tags = Tag.find_all_for_user(user_id=self.user.user_id)
-    #     for tag in tags:
-    #         self.assertIsNotNone(tag.id)
+    def test_all_for_user_by_user_id(self):
+        # Iterate over all tags for user
+        tags = Tag.find_all_for_user(user_id=self.user.user_id)
+        for tag in tags:
+            self.assertIsNotNone(tag.id)
 
     def test_tag_companies(self):
         # Tag companies
         tag = Tag.tag_companies("red", [self.user.companies[0].id])
         self.assertEqual(tag.name, "red")
+        company = Company.find(id=self.user.companies[0].id)
+        self.assertEqual(1, len(company.tags))
 
     def test_untag_companies(self):
         # Untag companies
-        tag = Tag.untag_companies("blue", [self.user.companies[0].id])
-        self.assertEqual(tag.name, "blue")
+        tag = Tag.untag_companies("red", [self.user.companies[0].id])
+        self.assertEqual(tag.name, "red")
+        company = Company.find(id=self.user.companies[0].id)
+        self.assertEqual(0, len(company.tags))
+
+    # Iterate over all tags for company
+    def test_all_for_company_by_id(self):
+        # Iterate over all tags for user
+        red_tag = Tag.tag_companies("red", [self.company.id])
+        tags = Tag.find_all_for_company(id=self.company.id)
+        for tag in tags:
+            self.assertEqual(red_tag.id, tag.id)
+        Tag.untag_companies("red", [self.company.id])
 
-    # # Iterate over all tags for company
-    # Tag.find_all_for_company(id='43357e2c3c77661e25000026')
-    # Tag.find_all_for_company(company_id='6')
+    def test_all_for_company_by_company_id(self):
+        # Iterate over all tags for user
+        red_tag = Tag.tag_companies("red", [self.company.id])
+        tags = Tag.find_all_for_company(company_id=self.company.id)
+        for tag in tags:
+            self.assertEqual(red_tag.id, tag.id)
+        Tag.untag_companies("red", [self.company.id])
