mirror of
https://github.com/pinry/pinry.git
synced 2025-11-13 16:45:41 +01:00
127 lines
4.6 KiB
Python
127 lines
4.6 KiB
Python
# 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
|
|
|
|
def setUp(self):
|
|
super(ImageResourceTest, self).setUp()
|
|
self.client = Client()
|
|
|
|
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.assertEqual(self.deserialize(response)['id'], 3)
|
|
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.assertEqual(Pin.objects.count(), 3)
|
|
self.assertEquals(Tag.objects.count(), 4)
|
|
|
|
def test_get_list_json_ordered(self):
|
|
pin = Pin.objects.latest('id')
|
|
response = self.api_client.get('/api/v1/pin/', format='json', data={'order_by': '-id'})
|
|
self.assertValidJSONResponse(response)
|
|
self.assertEqual(self.deserialize(response)['objects'][0]['id'], pin.id)
|
|
|
|
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'],
|
|
})
|