Adding UserTest based on User examples in README.md.

jkeyes · jkeyes · commit 4a3475f9d420 · 2014-09-24T23:23:26.000+01:00
diff --git a/tests/integration/__init__.py b/tests/integration/__init__.py
@@ -0,0 +1 @@
+# -*- coding: utf-8 -*-
diff --git a/tests/integration/test_user.py b/tests/integration/test_user.py
@@ -0,0 +1,66 @@
+# -*- coding: utf-8 -*-
+
+import os
+import unittest
+from intercom import Intercom
+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 UserTest(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")
+        print cls.user.id
+
+    def test_find_by_email(self):
+        # Find user by email
+        user = User.find(email=self.email)
+        print user
+        self.assertEqual(self.email, user.email)
+
+    def test_find_by_user_id(self):
+        # Find user by email
+        user = User.find(user_id="ada")
+        self.assertEqual(self.email, user.email)
+
+    def test_find_by_id(self):
+        # Find user by id
+        user = User.find(id=self.user.id)
+        self.assertEqual(self.email, user.email)
+
+    def test_custom_attributes(self):
+        # Update custom_attributes for a user
+        user = User.find(id=self.user.id)
+        user.custom_attributes["average_monthly_spend"] = 1234.56
+        user.save()
+        user = User.find(id=self.user.id)
+        self.assertEqual(
+            user.custom_attributes["average_monthly_spend"], 1234.56)
+
+    def test_increment(self):
+        # Perform incrementing
+        user = User.find(id=self.user.id)
+        karma = user.custom_attributes.get('karma', 0)
+        user.increment('karma')
+        user.save()
+        self.assertEqual(user.custom_attributes["karma"], karma + 1)
+        user.increment('karma')
+        user.save()
+        self.assertEqual(user.custom_attributes["karma"], karma + 2)
+
+    def test_iterate(self):
+        # Iterate over all users
+        for user in User.all():
+            self.assertTrue(user.id is not None)
