104 lines
4.1 KiB
Python
104 lines
4.1 KiB
Python
from __future__ import unicode_literals
|
|
import mock
|
|
try:
|
|
import unittest2 as unittest # Python2.6
|
|
except ImportError:
|
|
import unittest
|
|
|
|
import openphoto
|
|
|
|
class TestPhotos(unittest.TestCase):
|
|
TEST_HOST = "test.example.com"
|
|
|
|
TEST_PHOTOS = [{"id": "1a",
|
|
"tags": ["tag1", "tag2"],
|
|
"totalPages": 1,
|
|
"totalRows": 2},
|
|
{"id": "2b",
|
|
"tags": ["tag3", "tag4"],
|
|
"totalPages": 1,
|
|
"totalRows": 2}]
|
|
|
|
def setUp(self):
|
|
self.client = openphoto.OpenPhoto(host=self.TEST_HOST)
|
|
|
|
@staticmethod
|
|
def _return_value(result, message="", code=200):
|
|
return {"message": message, "code": code, "result": result}
|
|
|
|
class TestPhotosList(TestPhotos):
|
|
@mock.patch.object(openphoto.OpenPhoto, 'get')
|
|
def test_list(self, mock_get):
|
|
photos = self.TEST_PHOTOS
|
|
mock_get.return_value = self._return_value(photos)
|
|
|
|
result = self.client.photos.list()
|
|
mock_get.assert_called_with("/photos/list.json")
|
|
self.assertEqual(len(result), 2)
|
|
self.assertEqual(result[0].id, "1a")
|
|
self.assertEqual(result[0].tags, ["tag1", "tag2"])
|
|
self.assertEqual(result[1].id, "2b")
|
|
self.assertEqual(result[1].tags, ["tag3", "tag4"])
|
|
|
|
class TestPhotosUpdate(TestPhotos):
|
|
@mock.patch.object(openphoto.OpenPhoto, 'post')
|
|
def test_update(self, mock_post):
|
|
mock_post.return_value = self._return_value(True)
|
|
result = self.client.photos.update(["1a", "2b"], title="Test")
|
|
mock_post.assert_called_with("/photos/update.json",
|
|
ids=["1a", "2b"], title="Test")
|
|
self.assertEqual(result, True)
|
|
|
|
@mock.patch.object(openphoto.OpenPhoto, 'post')
|
|
def test_update_failure_raises_exception(self, mock_post):
|
|
mock_post.return_value = self._return_value(False)
|
|
with self.assertRaises(openphoto.OpenPhotoError):
|
|
self.client.photos.update(["1a", "2b"], title="Test")
|
|
|
|
class TestPhotosDelete(TestPhotos):
|
|
@mock.patch.object(openphoto.OpenPhoto, 'post')
|
|
def test_delete(self, mock_post):
|
|
mock_post.return_value = self._return_value(True)
|
|
result = self.client.photos.delete(["1a", "2b"])
|
|
mock_post.assert_called_with("/photos/delete.json", ids=["1a", "2b"])
|
|
self.assertEqual(result, True)
|
|
|
|
@mock.patch.object(openphoto.OpenPhoto, 'post')
|
|
def test_delete_failure_raises_exception(self, mock_post):
|
|
mock_post.return_value = self._return_value(False)
|
|
with self.assertRaises(openphoto.OpenPhotoError):
|
|
self.client.photos.delete(["1a", "2b"])
|
|
|
|
class TestPhotoDelete(TestPhotos):
|
|
@mock.patch.object(openphoto.OpenPhoto, 'post')
|
|
def test_delete(self, mock_post):
|
|
mock_post.return_value = self._return_value(True)
|
|
result = self.client.photo.delete("1a")
|
|
mock_post.assert_called_with("/photo/1a/delete.json")
|
|
self.assertEqual(result, True)
|
|
|
|
# TODO: photo.delete should raise exception on failure
|
|
@unittest.expectedFailure
|
|
@mock.patch.object(openphoto.OpenPhoto, 'post')
|
|
def test_delete_failure_raises_exception(self, mock_post):
|
|
mock_post.return_value = self._return_value(False)
|
|
with self.assertRaises(openphoto.OpenPhotoError):
|
|
self.client.photo.delete("1a")
|
|
|
|
@mock.patch.object(openphoto.OpenPhoto, 'post')
|
|
def test_object_delete(self, mock_post):
|
|
mock_post.return_value = self._return_value(True)
|
|
photo = openphoto.objects.Photo(self.client, self.TEST_PHOTOS[0])
|
|
result = photo.delete()
|
|
mock_post.assert_called_with("/photo/1a/delete.json")
|
|
self.assertEqual(result, True)
|
|
|
|
# TODO: photo.delete should raise exception on failure
|
|
@unittest.expectedFailure
|
|
@mock.patch.object(openphoto.OpenPhoto, 'post')
|
|
def test_object_delete_failure_raises_exception(self, mock_post):
|
|
mock_post.return_value = self._return_value(False)
|
|
photo = openphoto.objects.Photo(self.client, self.TEST_PHOTOS[0])
|
|
with self.assertRaises(openphoto.OpenPhotoError):
|
|
photo.delete()
|
|
|