Adding Tag tests (not complete).

jkeyes · jkeyes · commit 1f4fb5caaa83 · 2014-09-26T22:57:48.000+01:00
diff --git a/tests/integration/test_tags.py b/tests/integration/test_tags.py
@@ -0,0 +1,77 @@
+# -*- coding: utf-8 -*-
+
+import os
+import unittest
+from intercom import Intercom
+from intercom.tag import Tag
+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 TagTest(unittest.TestCase):
+    email = "ada@example.com"
+
+    @classmethod
+    def setup_class(cls):
+        # 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")
+            cls.user.companies = [
+                {"company_id": 6, "name": "Intercom"},
+                {"company_id": 9, "name": "Test Company"}
+            ]
+            cls.user.save()
+
+    def test_tag_users(self):
+        # Tag users
+        tag = Tag.tag_users("blue", [self.user.id])
+        self.assertEqual(tag.name, "blue")
+
+    def test_untag_users(self):
+        # Untag users
+        tag = Tag.untag_users("blue", [self.user.id])
+        self.assertEqual(tag.name, "blue")
+
+    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_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_tag_companies(self):
+        # Tag companies
+        tag = Tag.tag_companies("red", [self.user.companies[0].id])
+        self.assertEqual(tag.name, "red")
+
+    def test_untag_companies(self):
+        # Untag companies
+        tag = Tag.untag_companies("blue", [self.user.companies[0].id])
+        self.assertEqual(tag.name, "blue")
+
+    # # Iterate over all tags for company
+    # Tag.find_all_for_company(id='43357e2c3c77661e25000026')
+    # Tag.find_all_for_company(company_id='6')
