class AlbumTest(TestCase):
def setUp(self):
self.user = User(username='******', email='*****@*****.**',
first_name='Sherlock', last_name="Holmes",
password='******')
self.user.full_clean()
self.user.save()
self.photo = Photo(owner=self.user,
image='images/test.png',
name='test',
caption='testing')
self.photo.clean()
self.photo.save()
self.tag = Tag(name='test tag', owner=self.user)
self.tag.clean()
self.tag.save()
self.photo.tags.add(self.tag)
self.album = Album(owner=self.user,
name='test album')
self.album.clean()
self.album.save()
self.album.photos.add(self.photo)
def test_id_creation(self):
self.assertIsNotNone(self.album.id)
def test_owner_entry(self):
self.assertEqual(self.album.name, 'test album')
def test_name_entry(self):
self.assertEqual(self.photo.name, 'test')
def test_album_to_photo_association(self):
photos = Photo.objects.filter(album=self.album.id)
self.assertEqual(photos[0].name, 'test')
class PhotoTest(TestCase):
def setUp(self):
self.user = User(username='******', email='*****@*****.**',
first_name='Sherlock', last_name="Holmes",
password='******')
self.user.full_clean()
self.user.save()
self.photo = Photo(owner=self.user,
image='images/test.png',
name='test',
caption='testing')
self.photo.clean()
self.photo.save()
self.tag = Tag(name='test tag', owner=self.user)
self.tag.clean()
self.tag.save()
self.photo.tags.add(self.tag)
def test_id_creation(self):
self.assertIsNotNone(self.photo.id)
def test_owner_entry(self):
self.assertEqual(self.photo.owner.username, 'sholmes')
def test_image_entry(self):
self.assertEqual(self.photo.image, 'images/test.png')
def test_name_entry(self):
self.assertEqual(self.photo.name, 'test')
def test_caption_entry(self):
self.assertEqual(self.photo.caption, 'testing')
def test_tag_creation(self):
self.assertEqual(self.tag.name, 'test tag')
def test_photo_tag_association(self):
tags = Tag.objects.filter(photo=self.photo.id)
self.assertEqual(tags[0].name, 'test tag')
