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