Adding Notes test (and fixing load).

jkeyes · jkeyes · commit 512bdc9fe844 · 2014-09-27T00:12:09.000+01:00
diff --git a/intercom/api_operations/load.py b/intercom/api_operations/load.py
@@ -4,11 +4,11 @@
 
 class Load(object):
 
-    @classmethod
-    def load(cls, **params):
+    def load(self):
+        cls = self.__class__
         collection = utils.resource_class_to_collection_name(cls)
-        if 'id' in params:
-            response = Intercom.get("/%s/%s" % (collection, params['id']))
+        if hasattr(self, 'id'):
+            response = Intercom.get("/%s/%s" % (collection, self.id))
         else:
             raise Exception(
                 "Cannot load %s as it does not have a valid id." % (cls))
diff --git a/intercom/note.py b/intercom/note.py
@@ -1,8 +1,9 @@
 from intercom.user import Resource
+from intercom.api_operations.find_all import FindAll
 from intercom.api_operations.find import Find
 from intercom.api_operations.save import Save
 from intercom.api_operations.load import Load
 
 
-class Note(Resource, Find, Load, Save):
+class Note(Resource, Find, FindAll, Load, Save):
     pass
diff --git a/intercom/utils.py b/intercom/utils.py
@@ -33,6 +33,7 @@ def resource_class_to_name(cls):
 
 
 def create_class_instance(class_name):
+    from intercom.api_operations.load import Load
     from intercom.traits.api_resource import Resource
 
     if class_name in CLASS_REGISTRY:
@@ -43,7 +44,7 @@ def __new__(mcs, name, bases, attributes):
             return super(Meta, mcs).__new__(
                 mcs, str(class_name), bases, attributes)
 
-    class DynamicClass(Resource):
+    class DynamicClass(Resource, Load):
         __metaclass__ = Meta
 
     dyncls = DynamicClass()
diff --git a/tests/integration/test_notes.py b/tests/integration/test_notes.py
@@ -0,0 +1,45 @@
+# -*- coding: utf-8 -*-
+
+import os
+import unittest
+from intercom import Intercom
+from intercom.note import Note
+
+Intercom.app_id = os.environ.get('INTERCOM_APP_ID')
+Intercom.app_api_key = os.environ.get('INTERCOM_APP_API_KEY')
+
+
+class NoteTest(unittest.TestCase):
+    email = "ada@example.com"
+
+    def test_create_note(self):
+        # Create a note for a user
+        note = Note.create(
+            body="<p>Text for the note</p>",
+            email=self.email)
+        self.assertIsNotNone(note.id)
+
+    def test_find_note(self):
+        # Find a note by id
+        orig_note = Note.create(
+            body="<p>Text for the note</p>",
+            email=self.email)
+        note = Note.find(id=orig_note.id)
+        self.assertEqual(note.body, orig_note.body)
+
+    def test_find_all_email(self):
+        # Iterate over all notes for a user via their email address
+        for note in Note.find_all(email=self.email):
+            self.assertTrue(note.id is not None)
+            user = note.user.load()
+            self.assertEqual(user.email, self.email)
+
+    def test_find_all_id(self):
+        from intercom.user import User
+        user = User.find(email=self.email)
+
+        # Iterate over all notes for a user via their email address
+        for note in Note.find_all(user_id=user.user_id):
+            self.assertTrue(note.id is not None)
+            user = note.user.load()
+            self.assertEqual(user.email, self.email)
