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 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()
