A major RESTful API rewrite

Rewritten API to handle creating pins for both urls
and previously-uploaded images. Added some tests for it.

pinry/api/tests.py

@@ -1,16 +1,118 @@
-from django.test import TestCase
-from django.test.client import Client
-
-
 # pylint: disable-msg=R0904
 # pylint: disable-msg=E1103
 
+from django.test.client import Client
+from django_images.models import Thumbnail
+
+from taggit.models import Tag
+from tastypie.test import ResourceTestCase
+
+from ..pins.models import User, Pin, Image
+
+
+def filter_generator_for(size):
+    def wrapped_func(obj):
+        return Thumbnail.objects.get_or_create_at_size(obj.pk, size)
+    return wrapped_func
+
+
+class ImageResourceTest(ResourceTestCase):
+    fixtures = ['test_resources.json']
+    pass
 
-class RecentPinsTest(TestCase):
     def setUp(self):
+        super(ImageResourceTest, self).setUp()
         self.client = Client()
-        self.url = '/api/pin/?format=json'
 
-    def test_status_code(self):
-        response = self.client.get(self.url)
-        self.assertEqual(response.status_code, 200)
+    def test_list_detail(self):
+        image = Image.objects.get(pk=1)
+        thumbnail = filter_generator_for('thumbnail')(image)
+        standard = filter_generator_for('standard')(image)
+        response = self.api_client.get('/api/v1/image/', format='json')
+        self.assertDictEqual(self.deserialize(response)['objects'][0], {
+            u'image': image.image.url,
+            u'height': image.height,
+            u'width': image.width,
+            u'standard': {
+                u'image': unicode(standard.image.url),
+                u'width': standard.width,
+                u'height': standard.height,
+            },
+            u'thumbnail': {
+                u'image': unicode(thumbnail.image.url),
+                u'width': thumbnail.width,
+                u'height': thumbnail.height,
+            }
+        })
+
+
+class PinResourceTest(ResourceTestCase):
+    fixtures = ['test_resources.json']
+
+    def setUp(self):
+        super(PinResourceTest, self).setUp()
+
+        self.pin_1 = Pin.objects.get(pk=1)
+        self.image_url = 'http://technicallyphilly.com/wp-content/uploads/2013/02/url1.jpeg'
+
+        self.user = User.objects.get(pk=1)
+        self.api_client.client.login(username=self.user.username, password='password')
+
+    def test_post_create_url(self):
+        post_data = {
+            'submitter': '/api/v1/user/1/',
+            'url': self.image_url,
+            'description': 'That\'s an Apple!'
+        }
+        response = self.api_client.post('/api/v1/pin/', data=post_data)
+        self.assertHttpCreated(response)
+        self.assertEqual(Pin.objects.count(), 3)
+        self.assertEqual(Image.objects.count(), 3)
+
+    def test_post_create_obj(self):
+        user = User.objects.get(pk=1)
+        image = Image.objects.get(pk=1)
+        post_data = {
+            'submitter': '/api/v1/user/{}/'.format(user.pk),
+            'image': '/api/v1/image/{}/'.format(image.pk),
+            'description': 'That\'s something else (probably a CC logo)!',
+            'tags': ['random', 'tags'],
+        }
+        response = self.api_client.post('/api/v1/pin/', data=post_data)
+        self.assertHttpCreated(response)
+        # A number of Image objects should stay the same as we are using an existing image
+        self.assertEqual(Image.objects.count(), 2)
+        self.assertEquals(Tag.objects.count(), 4)
+
+    def test_get_list_json(self):
+        user = User.objects.get(pk=1)
+        image = Image.objects.get(pk=1)
+        standard = filter_generator_for('standard')(image)
+        thumbnail = filter_generator_for('thumbnail')(image)
+        response = self.api_client.get('/api/v1/pin/', format='json')
+        self.assertValidJSONResponse(response)
+        self.assertDictEqual(self.deserialize(response)['objects'][0], {
+            u'id': self.pin_1.id,
+            u'submitter': {
+                u'username': user.username,
+                u'gravatar': user.gravatar
+            },
+            u'image': {
+                u'image': unicode(image.image.url),
+                u'width': image.width,
+                u'height': image.height,
+                u'standard': {
+                    u'image': unicode(standard.image.url),
+                    u'width': standard.width,
+                    u'height': standard.height,
+                },
+                u'thumbnail': {
+                    u'image': unicode(thumbnail.image.url),
+                    u'width': thumbnail.width,
+                    u'height': thumbnail.height,
+                }
+            },
+            u'url': self.pin_1.url,
+            u'description': self.pin_1.description,
+            u'tags': [u'creative-commons'],
+        })