class PhotoDetailTest(TestCase):
def setUp(self):
user = User.objects.create_user('adam', '*****@*****.**',
'adampassword')
self.client.login(username='******', password='******')
album = Album.objects.create(owner=user, name='Test')
image_fd, image_path = tempfile.mkstemp(suffix='.jpg')
os.close(image_fd)
Image.new('RGB', (640, 480)).save(image_path, 'JPEG')
self.photo = Photo()
self.photo.owner = user
self.photo.album = album
image = open(image_path)
self.photo.image = ImageFile(image)
self.photo.is_jpeg = True
self.photo.description = 'test image'
self.photo.artist = 'Adam'
self.photo.country = 'USA'
self.photo.province_state = 'VA'
self.photo.city = 'Blacksburg'
self.photo.location = 'Drillfield'
self.photo.time_created = datetime.datetime(2007, 9, 28, 3, 0)
self.photo.save()
os.remove(image_path)
self.photo.keyword_list = ['test', 'photo']
self.photo.save()
def tearDown(self):
self.client.logout()
User.objects.all().delete()
Album.objects.all().delete()
PhotoTag.objects.all().delete()
Photo.objects.all().delete()
def test_view(self):
photo_detail = 'photo_detail'
photo_detail_url = reverse(photo_detail, args=[self.photo.id])
response = self.client.get(photo_detail_url)
self.assertEqual(response.status_code, 200)
def test_photo_in_album(self):
view = 'photo_in_album'
view_url = reverse(view, kwargs={
'object_id': self.photo.id,
})
response = self.client.get(view_url)
self.assertEqual(response.status_code, 200)
def setUp(self):
user = User.objects.create_user('adam', '*****@*****.**',
'adampassword')
self.client.login(username='******', password='******')
album = Album.objects.create(owner=user, name='Test')
image_fd, image_path = tempfile.mkstemp(suffix='.jpg')
os.close(image_fd)
Image.new('RGB', (640, 480)).save(image_path, 'JPEG')
self.photo = Photo()
self.photo.owner = user
self.photo.album = album
image = open(image_path)
self.photo.image = ImageFile(image)
self.photo.is_jpeg = True
self.photo.description = 'test image'
self.photo.artist = 'Adam'
self.photo.country = 'USA'
self.photo.province_state = 'VA'
self.photo.city = 'Blacksburg'
self.photo.location = 'Drillfield'
self.photo.time_created = datetime.datetime(2007, 9, 28, 3, 0)
self.photo.save()
os.remove(image_path)
self.photo.keyword_list = ['test', 'photo']
self.photo.save()
class PhotoTest(TestCase):
def setUp(self):
# Create an image.
image_fd, image_path = tempfile.mkstemp(suffix='.jpg')
os.close(image_fd)
Image.new('RGB', (1, 1)).save(image_path, 'JPEG')
# Create a Photo object.
user = User.objects.create(username="******")
album = Album.objects.create(owner=user, name="Test")
self.photo = Photo()
self.photo.owner = user
image = open(image_path)
self.photo.image = ImageFile(image)
self.photo.album = album
self.photo.is_jpeg = True
self.photo.save()
image.close()
os.remove(image_path)
def tearDown(self):
Photo.objects.all().delete()
Album.objects.all().delete()
User.objects.all().delete()
def test_photo_edit(self):
photo_edit = 'photasm.photos.views.photo_edit'
photo_edit_url = reverse(photo_edit, args=[self.photo.id])
login_url = get_login_url(photo_edit_url)
response = self.client.get(photo_edit_url)
self.assertRedirects(response, login_url)
def test_home(self):
home_url = reverse('photasm.photos.views.home')
login_url = get_login_url(home_url)
response = self.client.get(home_url)
self.assertRedirects(response, login_url)
def test_new_album(self):
home_url = reverse('new_album')
login_url = get_login_url(home_url)
response = self.client.get(home_url)
self.assertRedirects(response, login_url)
def setUp(self):
# Create an image.
image_fd, image_path = tempfile.mkstemp(suffix='.jpg')
os.close(image_fd)
Image.new('RGB', (1, 1)).save(image_path, 'JPEG')
# Create a Photo object.
user = User.objects.create_user('adam', '*****@*****.**',
'adampassword')
self.album = Album.objects.create(owner=user, name="Test")
photo = Photo()
photo.owner = user
image = open(image_path)
photo.image = ImageFile(image)
photo.album = self.album
photo.is_jpeg = True
photo.save()
image.close()
os.remove(image_path)
photo.create_thumbnail()
self.client.login(username='******', password='******')
def setUp(self):
# Create an image.
image_fd, image_path = tempfile.mkstemp(suffix='.jpg')
os.close(image_fd)
Image.new('RGB', (1, 1)).save(image_path, 'JPEG')
# Create a Photo object.
user = User.objects.create(username="******")
album = Album.objects.create(owner=user, name="Test")
self.photo = Photo()
self.photo.owner = user
image = open(image_path)
self.photo.image = ImageFile(image)
self.photo.album = album
self.photo.is_jpeg = True
self.photo.save()
image.close()
os.remove(image_path)
def test_get(self):
# Create a Photo object.
photo = Photo()
owner = User.objects.create(username='******')
album = Album.objects.create(name='test', owner=owner)
test_kw = PhotoTag.objects.create(name="test")
photo_kw = PhotoTag.objects.create(name="photo")
photo.owner = owner
photo.album = album
image = open(self.file_path)
photo.image = ImageFile(image)
photo.save()
image.close()
# Photo has empty list of keywords.
self.assertEqual(photo.keyword_list, [])
# Add some keywords.
photo.keywords.add(test_kw)
photo.keywords.add(photo_kw)
photo.save()
self.assertEqual(photo.keyword_list, [u'test', u'photo'])
def setUp(self):
self.user = User.objects.create_user('adam', '*****@*****.**',
'adampassword')
self.user.is_staff = True
self.user.is_superuser = True
self.user.save()
self.client.login(username='******', password='******')
self.album = Album.objects.create(owner=self.user, name='Test')
self.keyword = PhotoTag.objects.create(name='test')
image_fd, image_path = tempfile.mkstemp(suffix='.jpg')
os.close(image_fd)
Image.new('RGB', (640, 480)).save(image_path, 'JPEG')
self.photo = Photo()
self.photo.owner = self.user
self.photo.album = self.album
image = open(image_path)
self.photo.image = ImageFile(image)
self.photo.is_jpeg = True
self.photo.save()
os.remove(image_path)
class EditPhotoTest(TestCase):
def setUp(self):
self.user = User.objects.create_user('adam', '*****@*****.**',
'adampassword')
self.user.is_staff = True
self.user.is_superuser = True
self.user.save()
self.client.login(username='******', password='******')
self.album = Album.objects.create(owner=self.user, name='Test')
self.keyword = PhotoTag.objects.create(name='test')
image_fd, image_path = tempfile.mkstemp(suffix='.jpg')
os.close(image_fd)
Image.new('RGB', (640, 480)).save(image_path, 'JPEG')
self.photo = Photo()
self.photo.owner = self.user
self.photo.album = self.album
image = open(image_path)
self.photo.image = ImageFile(image)
self.photo.is_jpeg = True
self.photo.save()
os.remove(image_path)
def tearDown(self):
self.client.logout()
User.objects.all().delete()
Album.objects.all().delete()
PhotoTag.objects.all().delete()
Photo.objects.all().delete()
def test_form(self):
photo_edit = 'photasm.photos.views.photo_edit'
photo_edit_url = reverse(photo_edit, args=[0])
response = self.client.get(photo_edit_url)
self.assertEqual(response.status_code, 404)
photo_edit_url = reverse(photo_edit, args=[self.photo.id])
response = self.client.get(photo_edit_url)
self.assertEqual(response.status_code, 200)
post_data = {
'album': self.album.id,
'description': 'test image',
'artist': 'Adam',
'country': 'USA',
'province_state': 'VA',
'city': 'Blacksburg',
'location': 'Drillfield',
'time_created': '2007-09-28 03:00:00',
'keywords': self.keyword.id,
}
response = self.client.post(photo_edit_url, post_data)
self.photo = Photo.objects.get(id=self.photo.id)
self.assertRedirects(response, self.photo.get_absolute_url())
self.assertFalse(self.photo.sync_metadata_to_file())
self.assertEqual(self.photo.description, 'test image')
response = self.client.post(photo_edit_url, {})
self.assertEqual(response.status_code, 200)
photo_edit = 'photo_edit_in_album'
photo_edit_url = reverse(photo_edit, kwargs={
'object_id': self.photo.id,
'in_album': 'albums',
})
post_data['description'] = 'test image in album'
response = self.client.post(photo_edit_url, post_data)
self.photo = Photo.objects.get(id=self.photo.id)
photo_url = reverse("photo_in_album", kwargs={
"object_id": self.photo.id,
})
self.assertRedirects(response, photo_url)
def test_admin_form(self):
photo_edit_url = '/admin/photos/photo/%d/' % self.photo.id
post_data = {
'owner': self.user.id,
'album': self.album.id,
'description': 'test image',
'artist': 'Adam',
'country': 'USA',
'province_state': 'VA',
'city': 'Blacksburg',
'location': 'Drillfield',
'time_created': '2007-09-28 03:00:00',
'keywords': self.keyword.id,
'metadata_sync_enabled': True,
'_save': 'Save',
}
response = self.client.post(photo_edit_url, post_data)
self.photo = Photo.objects.get(id=self.photo.id)
self.assertEqual(response.status_code, 302)
self.assertFalse(self.photo.sync_metadata_to_file())
self.assertEqual(self.photo.description, 'test image')
def runTest(self):
# Create a JPEG with an embedded thumbnail.
image_fd, image_path = tempfile.mkstemp(suffix='.jpg')
os.close(image_fd)
thumb_fd, thumb_path = tempfile.mkstemp(suffix='.jpg')
os.close(thumb_fd)
image = Image.new('RGB', (640, 480))
image.save(image_path, 'JPEG')
thumb = image.copy()
thumb.thumbnail((160, 120))
thumb.save(thumb_path, 'JPEG')
# Write the thumbnail to the image metadata.
metadata = pyexiv2.Image(image_path)
metadata.readMetadata()
metadata.setThumbnailFromJpegFile(thumb_path)
metadata.writeMetadata()
os.remove(thumb_path)
# Create Photo model object.
user = User.objects.create(username="******")
album = Album.objects.create(owner=user, name="Test")
photo = Photo()
photo.owner = user
image = open(image_path)
photo.image = ImageFile(image)
photo.album = album
photo.is_jpeg = True
photo.save()
image.close()
os.remove(image_path)
# Create and verify the thumbnail.
photo.create_thumbnail()
photo.save()
thumb = Image.open(photo.thumbnail.path)
self.assertEqual(thumb.size[0] * thumb.size[1], 19200)
metadata = pyexiv2.Image(photo.image.path)
metadata.readMetadata()
embedded = metadata.getThumbnailData()[1]
external = open(photo.thumbnail.path).read()
self.assertEqual(embedded, external)
# Test JPEG without thumbnail already embedded.
image = Image.new('RGB', (640, 480))
image.save(image_path, 'JPEG')
photo = Photo()
photo.owner = user
image = open(image_path)
photo.image = ImageFile(image)
photo.album = album
photo.is_jpeg = True
photo.save()
image.close()
os.remove(image_path)
photo.create_thumbnail()
photo.save()
thumb = Image.open(photo.thumbnail.path)
self.assertEqual(thumb.size[0] * thumb.size[1], 19200)
metadata = pyexiv2.Image(photo.image.path)
metadata.readMetadata()
embedded = metadata.getThumbnailData()[1]
external = open(photo.thumbnail.path).read()
self.assertEqual(embedded, external)
# Test BMP.
image_fd, image_path = tempfile.mkstemp(suffix='.bmp')
os.close(image_fd)
image = Image.new('RGB', (640, 480))
image.save(image_path, 'BMP')
photo = Photo()
photo.owner = user
image = open(image_path)
photo.image = ImageFile(image)
photo.album = album
photo.is_jpeg = False
photo.save()
image.close()
os.remove(image_path)
photo.create_thumbnail()
photo.save()
thumb = Image.open(photo.thumbnail.path)
self.assertEqual(thumb.size[0] * thumb.size[1], 19200)
# Test small image.
image = Image.new('RGB', (80, 60))
image.save(image_path, 'JPEG')
photo = Photo()
photo.owner = user
image = open(image_path)
photo.image = ImageFile(image)
photo.album = album
photo.is_jpeg = True
photo.save()
image.close()
os.remove(image_path)
self.assertEqual(photo.create_thumbnail(), None)
photo.save()
thumb = Image.open(photo.thumbnail.path)
self.assertEqual(thumb.size, (80, 60))
metadata = pyexiv2.Image(photo.image.path)
metadata.readMetadata()
embedded = metadata.getThumbnailData()[1]
external = open(photo.thumbnail.path).read()
self.assertEqual(embedded, external)
User.objects.all().delete()
Album.objects.all().delete()
Photo.objects.all().delete()
def test_set(self):
# Create a Photo object.
test_kw = PhotoTag.objects.create(name="test")
photo = Photo()
owner = User.objects.create(username='******')
album = Album.objects.create(name='test', owner=owner)
photo.owner = owner
photo.album = album
image = open(self.file_path)
photo.image = ImageFile(image)
photo.save()
image.close()
self.assertEqual(test_kw.photo_set.count(), 0)
# Reminder: Database is assumed to be empty at the start of this test.
self.assertEqual(PhotoTag.objects.count(), 1)
# Set the photo's keywords.
photo.keyword_list = ['Test', 'photo']
photo.save()
self.assertEqual(photo.keyword_list, [u'test', u'photo'])
self.assertEqual(test_kw.photo_set.count(), 1)
self.assertEqual(PhotoTag.objects.count(), 2)
# Set the photo's keywords to something completely different.
photo.keyword_list = ['foo', 'bar']
photo.save()
self.assertEqual(photo.keyword_list, [u'foo', u'bar'])
self.assertEqual(test_kw.photo_set.count(), 0)
self.assertEqual(PhotoTag.objects.count(), 4)
# Set the photo's keywords to an empty list.
photo.keyword_list = []
photo.save()
self.assertEqual(photo.keyword_list, [])
def runTest(self):
# Create an image with metadata.
file_descriptor, file_path = tempfile.mkstemp(suffix='.jpg')
os.close(file_descriptor)
Image.new('RGB', (1, 1)).save(file_path, 'JPEG')
metadata = pyexiv2.Image(file_path)
metadata.readMetadata()
metadata['Exif.Image.ImageDescription'] = "Test file"
metadata['Exif.Image.Artist'] = "Adam"
metadata['Iptc.Application2.CountryName'] = "USA"
metadata['Iptc.Application2.ProvinceState'] = "Virginia"
metadata['Iptc.Application2.City'] = "Blacksburg"
metadata['Iptc.Application2.SubLocation'] = "Dreamland"
try:
metadata['Exif.Photo.DateTimeOriginal'] = datetime.datetime(
2007, 9, 28, 3, 0)
except TypeError:
pass
metadata['Iptc.Application2.Keywords'] = ['test', 'photo']
metadata.writeMetadata()
# Create a Photo object.
user = User.objects.create(username="******")
album = Album.objects.create(owner=user, name="Test")
photo = Photo()
photo.owner = user
image = open(file_path)
photo.image = ImageFile(image)
photo.album = album
photo.is_jpeg = True
photo.save()
image.close()
os.remove(file_path)
# Synchronize.
performed_sync = photo.sync_metadata_from_file()
self.assertTrue(performed_sync)
# Metadata should exist in the database.
self.assertEqual(photo.description, 'Test file')
self.assertEqual(photo.artist, 'Adam')
self.assertEqual(photo.country, 'USA')
self.assertEqual(photo.province_state, 'Virginia')
self.assertEqual(photo.city, 'Blacksburg')
self.assertEqual(photo.location, 'Dreamland')
self.assertEqual(str(photo.time_created), '2007-09-28 03:00:00')
self.assertEqual(photo.keyword_list, [u'test', u'photo'])
# Attempt to synchronize again...
self.assertFalse(photo.sync_metadata_from_file())
# Nothing should have changed.
self.assertEqual(photo.description, 'Test file')
self.assertEqual(photo.artist, 'Adam')
self.assertEqual(photo.country, 'USA')
self.assertEqual(photo.province_state, 'Virginia')
self.assertEqual(photo.city, 'Blacksburg')
self.assertEqual(photo.location, 'Dreamland')
self.assertEqual(str(photo.time_created), '2007-09-28 03:00:00')
self.assertEqual(photo.keyword_list, [u'test', u'photo'])
# Synchronize only one field.
metadata = pyexiv2.Image(photo.image.path)
metadata.readMetadata()
metadata['Exif.Image.ImageDescription'] = "Image for testing"
metadata.writeMetadata()
performed_sync = photo.sync_metadata_from_file()
self.assertTrue(performed_sync)
self.assertEqual(photo.description, 'Image for testing')
# Disable synchronization.
photo.metadata_sync_enabled = False
photo.description = "Image for testing"
photo.save()
self.assertFalse(photo.sync_metadata_from_file())
# Sometimes metadata can't be read from the image.
file_descriptor, file_path = tempfile.mkstemp(suffix='.pcx')
os.close(file_descriptor)
Image.new('RGB', (1, 1)).save(file_path, 'PCX')
photo = Photo()
photo.owner = user
image = open(file_path)
photo.image = ImageFile(image)
photo.album = album
photo.save()
image.close()
os.remove(file_path)
self.assertFalse(photo.sync_metadata_from_file())
self.assertFalse(photo.metadata_sync_enabled)
User.objects.all().delete()
Album.objects.all().delete()
PhotoTag.objects.all().delete()
Photo.objects.all().delete()
def runTest(self):
# Create an image.
file_descriptor, file_path = tempfile.mkstemp(suffix='.jpg')
os.close(file_descriptor)
Image.new('RGB', (1, 1)).save(file_path, 'JPEG')
# Create a Photo object.
user = User.objects.create(username="******")
album = Album.objects.create(owner=user, name="Test")
photo = Photo()
photo.owner = user
image = open(file_path)
photo.image = ImageFile(image)
photo.description = "Test file"
photo.artist = "Adam"
photo.country = "USA"
photo.province_state = "Virginia"
photo.city = "Blacksburg"
photo.location = "Dreamland"
photo.time_created = datetime.datetime(2007, 9, 28, 3, 0)
photo.album = album
photo.is_jpeg = True
photo.save()
image.close()
os.remove(file_path)
photo.keyword_list = ['test', 'photo']
photo.save()
# Synchronize.
performed_sync = photo.sync_metadata_to_file()
self.assertTrue(performed_sync)
# Metadata should exist in the file itself.
metadata = pyexiv2.Image(photo.image.path)
metadata.readMetadata()
self.assertEqual(metadata['Exif.Image.ImageDescription'], 'Test file')
self.assertEqual(metadata['Exif.Image.Artist'], 'Adam')
self.assertEqual(metadata['Iptc.Application2.CountryName'], 'USA')
self.assertEqual(metadata['Iptc.Application2.ProvinceState'],
'Virginia')
self.assertEqual(metadata['Iptc.Application2.City'], 'Blacksburg')
self.assertEqual(metadata['Iptc.Application2.SubLocation'],
'Dreamland')
self.assertEqual(str(metadata['Exif.Photo.DateTimeOriginal']),
'2007-09-28 03:00:00')
self.assertEqual(metadata['Iptc.Application2.Keywords'],
('test', 'photo'))
self.assertEqual(metadata['Exif.Photo.PixelXDimension'], 1)
self.assertEqual(metadata['Exif.Photo.PixelYDimension'], 1)
# Attempt to synchronize again...
self.assertFalse(photo.sync_metadata_to_file())
# Nothing should have changed.
metadata = pyexiv2.Image(photo.image.path)
metadata.readMetadata()
self.assertEqual(metadata['Exif.Image.ImageDescription'], 'Test file')
self.assertEqual(metadata['Exif.Image.Artist'], 'Adam')
self.assertEqual(metadata['Iptc.Application2.CountryName'], 'USA')
self.assertEqual(metadata['Iptc.Application2.ProvinceState'],
'Virginia')
self.assertEqual(metadata['Iptc.Application2.City'], 'Blacksburg')
self.assertEqual(metadata['Iptc.Application2.SubLocation'],
'Dreamland')
self.assertEqual(str(metadata['Exif.Photo.DateTimeOriginal']),
'2007-09-28 03:00:00')
self.assertEqual(metadata['Iptc.Application2.Keywords'],
('test', 'photo'))
self.assertEqual(metadata['Exif.Photo.PixelXDimension'], 1)
self.assertEqual(metadata['Exif.Photo.PixelYDimension'], 1)
# Synchronize only one field.
photo.description = "Image for testing"
photo.save()
performed_sync = photo.sync_metadata_to_file()
self.assertTrue(performed_sync)
metadata = pyexiv2.Image(photo.image.path)
metadata.readMetadata()
self.assertEqual(metadata['Exif.Image.ImageDescription'],
'Image for testing')
# Disable synchronization.
photo.metadata_sync_enabled = False
photo.description = "Sample image for testing purposes"
photo.save()
self.assertFalse(photo.sync_metadata_to_file())
# Sometimes the image isn't a JPEG.
file_descriptor, file_path = tempfile.mkstemp(suffix='.tif')
os.close(file_descriptor)
Image.new('RGB', (1, 1)).save(file_path, 'TIFF')
photo = Photo()
photo.owner = user
photo.description = 'Test file'
image = open(file_path)
photo.image = ImageFile(image)
photo.album = album
photo.is_jpeg = False
photo.save()
image.close()
os.remove(file_path)
performed_sync = photo.sync_metadata_to_file()
self.assertTrue(performed_sync)
metadata = pyexiv2.Image(photo.image.path)
metadata.readMetadata()
self.assertEqual(metadata['Exif.Image.ImageWidth'], 1)
self.assertEqual(metadata['Exif.Image.ImageLength'], 1)
# Sometimes metadata can't be read from the image.
file_descriptor, file_path = tempfile.mkstemp(suffix='.pcx')
os.close(file_descriptor)
Image.new('RGB', (1, 1)).save(file_path, 'PCX')
photo = Photo()
photo.owner = user
photo.description = 'Test file'
image = open(file_path)
photo.image = ImageFile(image)
photo.album = album
photo.save()
image.close()
os.remove(file_path)
self.assertFalse(photo.sync_metadata_to_file())
self.assertFalse(photo.metadata_sync_enabled)
# Sometimes metadata can't be written to the image.
file_descriptor, file_path = tempfile.mkstemp(suffix='.bmp')
os.close(file_descriptor)
Image.new('RGB', (1, 1)).save(file_path, 'BMP')
photo = Photo()
photo.owner = user
photo.description = 'Test file'
image = open(file_path)
photo.image = ImageFile(image)
photo.album = album
photo.save()
image.close()
os.remove(file_path)
self.assertFalse(photo.sync_metadata_to_file())
self.assertFalse(photo.metadata_sync_enabled)
User.objects.all().delete()
Album.objects.all().delete()
PhotoTag.objects.all().delete()
Photo.objects.all().delete()
