234 lines
10 KiB
Python
234 lines
10 KiB
Python
from __future__ import unicode_literals
|
|
import mock
|
|
try:
|
|
import unittest2 as unittest # Python2.6
|
|
except ImportError:
|
|
import unittest
|
|
|
|
import openphoto
|
|
|
|
class TestAlbums(unittest.TestCase):
|
|
TEST_HOST = "test.example.com"
|
|
TEST_ALBUMS_DICT = [{"cover": {"id": "1a", "tags": ["tag1", "tag2"]},
|
|
"id": "1",
|
|
"name": "Album 1",
|
|
"totalRows": 2},
|
|
{"cover": {"id": "2b", "tags": ["tag3", "tag4"]},
|
|
"id": "2",
|
|
"name": "Album 2",
|
|
"totalRows": 2}]
|
|
def setUp(self):
|
|
self.client = openphoto.OpenPhoto(host=self.TEST_HOST)
|
|
self.TEST_ALBUMS = [openphoto.objects.Album(self.client, album)
|
|
for album in self.TEST_ALBUMS_DICT]
|
|
|
|
@staticmethod
|
|
def _return_value(result, message="", code=200):
|
|
return {"message": message, "code": code, "result": result}
|
|
|
|
class TestAlbumsList(TestAlbums):
|
|
@mock.patch.object(openphoto.OpenPhoto, 'get')
|
|
def test_albums_list(self, mock):
|
|
mock.return_value = self._return_value(self.TEST_ALBUMS_DICT)
|
|
result = self.client.albums.list()
|
|
mock.assert_called_with("/albums/list.json")
|
|
self.assertEqual(len(result), 2)
|
|
self.assertEqual(result[0].id, "1")
|
|
self.assertEqual(result[0].name, "Album 1")
|
|
self.assertEqual(result[1].id, "2")
|
|
self.assertEqual(result[1].name, "Album 2")
|
|
|
|
# TODO: cover should be updated to Photo object
|
|
@unittest.expectedFailure
|
|
@mock.patch.object(openphoto.OpenPhoto, 'get')
|
|
def test_albums_list_returns_cover_photos(self, mock):
|
|
mock.return_value = self._return_value(self.TEST_ALBUMS_DICT)
|
|
result = self.client.albums.list()
|
|
mock.assert_called_with("/albums/list.json")
|
|
self.assertEqual(len(result), 2)
|
|
self.assertEqual(result[0].id, "1")
|
|
self.assertEqual(result[0].name, "Album 1")
|
|
self.assertEqual(result[0].cover.id, "1a")
|
|
self.assertEqual(result[0].cover.tags, ["tag1", "tag2"])
|
|
self.assertEqual(result[1].id, "2")
|
|
self.assertEqual(result[0].name, "Album 2")
|
|
self.assertEqual(result[1].cover.id, "2b")
|
|
self.assertEqual(result[1].cover.tags, ["tag3", "tag4"])
|
|
|
|
class TestAlbumCreate(TestAlbums):
|
|
# TODO: cover should be updated to Photo object
|
|
@mock.patch.object(openphoto.OpenPhoto, 'post')
|
|
def test_album_create(self, mock):
|
|
mock.return_value = self._return_value(self.TEST_ALBUMS_DICT[0])
|
|
result = self.client.album.create(name="Test", foo="bar")
|
|
mock.assert_called_with("/album/create.json", name="Test", foo="bar")
|
|
self.assertEqual(result.id, "1")
|
|
self.assertEqual(result.name, "Album 1")
|
|
# self.assertEqual(result.cover.id, "1a")
|
|
# self.assertEqual(result.cover.tags, ["tag1", "tag2"])
|
|
|
|
class TestAlbumDelete(TestAlbums):
|
|
@mock.patch.object(openphoto.OpenPhoto, 'post')
|
|
def test_album_delete(self, mock):
|
|
mock.return_value = self._return_value(True)
|
|
result = self.client.album.delete(self.TEST_ALBUMS[0])
|
|
mock.assert_called_with("/album/1/delete.json")
|
|
self.assertEqual(result, True)
|
|
|
|
@mock.patch.object(openphoto.OpenPhoto, 'post')
|
|
def test_album_delete_id(self, mock):
|
|
mock.return_value = self._return_value(True)
|
|
result = self.client.album.delete("1")
|
|
mock.assert_called_with("/album/1/delete.json")
|
|
self.assertEqual(result, True)
|
|
|
|
# TODO: album.delete should raise exception on failure
|
|
@unittest.expectedFailure
|
|
@mock.patch.object(openphoto.OpenPhoto, 'post')
|
|
def test_album_delete_failure_raises_exception(self, mock):
|
|
mock.return_value = self._return_value(False)
|
|
with self.assertRaises(openphoto.OpenPhotoError):
|
|
self.client.album.delete(self.TEST_ALBUMS[0])
|
|
|
|
# TODO: after deleting object fields, name and id should be set to None
|
|
@mock.patch.object(openphoto.OpenPhoto, 'post')
|
|
def test_album_object_delete(self, mock):
|
|
mock.return_value = self._return_value(True)
|
|
album = self.TEST_ALBUMS[0]
|
|
result = album.delete()
|
|
mock.assert_called_with("/album/1/delete.json")
|
|
self.assertEqual(result, True)
|
|
self.assertEqual(album.get_fields(), {})
|
|
# self.assertEqual(album.id, None)
|
|
# self.assertEqual(album.name, None)
|
|
|
|
# TODO: album.delete should raise exception on failure
|
|
@unittest.expectedFailure
|
|
@mock.patch.object(openphoto.OpenPhoto, 'post')
|
|
def test_album_object_delete_failure_raises_exception(self, mock):
|
|
mock.return_value = self._return_value(False)
|
|
with self.assertRaises(openphoto.OpenPhotoError):
|
|
self.TEST_ALBUMS[0].delete()
|
|
|
|
class TestAlbumForm(TestAlbums):
|
|
@mock.patch.object(openphoto.OpenPhoto, 'post')
|
|
def test_album_form(self, mock):
|
|
with self.assertRaises(NotImplementedError):
|
|
self.client.album.form(self.TEST_ALBUMS[0])
|
|
|
|
@mock.patch.object(openphoto.OpenPhoto, 'post')
|
|
def test_album_form_id(self, mock):
|
|
with self.assertRaises(NotImplementedError):
|
|
self.client.album.form("1")
|
|
|
|
@mock.patch.object(openphoto.OpenPhoto, 'post')
|
|
def test_album_object_form(self, mock):
|
|
with self.assertRaises(NotImplementedError):
|
|
self.TEST_ALBUMS[0].form()
|
|
|
|
class TestAlbumAddPhotos(TestAlbums):
|
|
@mock.patch.object(openphoto.OpenPhoto, 'post')
|
|
def test_album_add_photos(self, mock):
|
|
with self.assertRaises(NotImplementedError):
|
|
self.client.album.add_photos(self.TEST_ALBUMS[0], ["Photo Objects"])
|
|
|
|
@mock.patch.object(openphoto.OpenPhoto, 'post')
|
|
def test_album_add_photos_id(self, mock):
|
|
with self.assertRaises(NotImplementedError):
|
|
self.client.album.add_photos("1", ["Photo Objects"])
|
|
|
|
# TODO: object.add_photos should accept photos list as first parameter
|
|
@unittest.expectedFailure
|
|
@mock.patch.object(openphoto.OpenPhoto, 'post')
|
|
def test_album_object_add_photos(self, mock):
|
|
with self.assertRaises(NotImplementedError):
|
|
self.TEST_ALBUMS[0].add_photos(["Photo Objects"])
|
|
|
|
class TestAlbumRemovePhotos(TestAlbums):
|
|
@mock.patch.object(openphoto.OpenPhoto, 'post')
|
|
def test_album_remove_photos(self, mock):
|
|
with self.assertRaises(NotImplementedError):
|
|
self.client.album.remove_photos(self.TEST_ALBUMS[0], ["Photo Objects"])
|
|
|
|
@mock.patch.object(openphoto.OpenPhoto, 'post')
|
|
def test_album_remove_photos_id(self, mock):
|
|
with self.assertRaises(NotImplementedError):
|
|
self.client.album.remove_photos("1", ["Photo Objects"])
|
|
|
|
# TODO: object.remove_photos should accept photos list as first parameter
|
|
@unittest.expectedFailure
|
|
@mock.patch.object(openphoto.OpenPhoto, 'post')
|
|
def test_album_object_remove_photos(self, mock):
|
|
with self.assertRaises(NotImplementedError):
|
|
self.TEST_ALBUMS[0].remove_photos(["Photo Objects"])
|
|
|
|
class TestAlbumUpdate(TestAlbums):
|
|
# TODO: cover should be updated to Photo object
|
|
@mock.patch.object(openphoto.OpenPhoto, 'post')
|
|
def test_album_update(self, mock):
|
|
mock.return_value = self._return_value(self.TEST_ALBUMS_DICT[1])
|
|
result = self.client.album.update(self.TEST_ALBUMS[0], name="Test")
|
|
mock.assert_called_with("/album/1/update.json", name="Test")
|
|
self.assertEqual(result.id, "2")
|
|
self.assertEqual(result.name, "Album 2")
|
|
# self.assertEqual(result.cover.id, "2b")
|
|
# self.assertEqual(result.cover.tags, ["tag3", "tag4"])
|
|
|
|
# TODO: cover should be updated to Photo object
|
|
@mock.patch.object(openphoto.OpenPhoto, 'post')
|
|
def test_album_update_id(self, mock):
|
|
mock.return_value = self._return_value(self.TEST_ALBUMS_DICT[1])
|
|
result = self.client.album.update("1", name="Test")
|
|
mock.assert_called_with("/album/1/update.json", name="Test")
|
|
self.assertEqual(result.id, "2")
|
|
self.assertEqual(result.name, "Album 2")
|
|
# self.assertEqual(result.cover.id, "2b")
|
|
# self.assertEqual(result.cover.tags, ["tag3", "tag4"])
|
|
|
|
# TODO: cover should be updated to Photo object
|
|
@mock.patch.object(openphoto.OpenPhoto, 'post')
|
|
def test_album_object_update(self, mock):
|
|
mock.return_value = self._return_value(self.TEST_ALBUMS_DICT[1])
|
|
album = self.TEST_ALBUMS[0]
|
|
album.update(name="Test")
|
|
mock.assert_called_with("/album/1/update.json", name="Test")
|
|
self.assertEqual(album.id, "2")
|
|
self.assertEqual(album.name, "Album 2")
|
|
# self.assertEqual(album.cover.id, "2b")
|
|
# self.assertEqual(album.cover.tags, ["tag3", "tag4"])
|
|
|
|
class TestAlbumView(TestAlbums):
|
|
# TODO: cover should be updated to Photo object
|
|
@mock.patch.object(openphoto.OpenPhoto, 'get')
|
|
def test_album_view(self, mock):
|
|
mock.return_value = self._return_value(self.TEST_ALBUMS_DICT[1])
|
|
result = self.client.album.view(self.TEST_ALBUMS[0], name="Test")
|
|
mock.assert_called_with("/album/1/view.json", name="Test")
|
|
self.assertEqual(result.id, "2")
|
|
self.assertEqual(result.name, "Album 2")
|
|
# self.assertEqual(result.cover.id, "2b")
|
|
# self.assertEqual(result.cover.tags, ["tag3", "tag4"])
|
|
|
|
# TODO: cover should be updated to Photo object
|
|
@mock.patch.object(openphoto.OpenPhoto, 'get')
|
|
def test_album_view_id(self, mock):
|
|
mock.return_value = self._return_value(self.TEST_ALBUMS_DICT[1])
|
|
result = self.client.album.view("1", name="Test")
|
|
mock.assert_called_with("/album/1/view.json", name="Test")
|
|
self.assertEqual(result.id, "2")
|
|
self.assertEqual(result.name, "Album 2")
|
|
# self.assertEqual(result.cover.id, "2b")
|
|
# self.assertEqual(result.cover.tags, ["tag3", "tag4"])
|
|
|
|
# TODO: cover should be updated to Photo object
|
|
@mock.patch.object(openphoto.OpenPhoto, 'get')
|
|
def test_album_object_view(self, mock):
|
|
mock.return_value = self._return_value(self.TEST_ALBUMS_DICT[1])
|
|
album = self.TEST_ALBUMS[0]
|
|
album.view(name="Test")
|
|
mock.assert_called_with("/album/1/view.json", name="Test")
|
|
self.assertEqual(album.id, "2")
|
|
self.assertEqual(album.name, "Album 2")
|
|
# self.assertEqual(album.cover.id, "2b")
|
|
# self.assertEqual(album.cover.tags, ["tag3", "tag4"])
|
|
|