def test_add_artworks_data(self):
self.artwork1 = Artworks('Leonardo Da Vinci', 'Mona Lisa', 830000000,
True)
self.artwork2 = Artworks('Vincent Van Gogh', 'The Starry Night',
120000000, True)
self.artwork1.save()
self.artwork2.save()
def test_search_available_artwork(self):
self.test_add_artworks_data()
artwork_search1 = Artworks('Leonardo Da Vinci', 'Mona lisa', 830000000,
True)
artwork_db.search_available_artwork(artwork_search1)
artwork_search2 = Artworks('Leonardo Da Vinci', 'Mona lisa', 830000000,
False)
with self.assertRaises(ArtworkDataError):
artwork_db.search_available_artwork(artwork_search2)
def test_save_update_artwork_availibility_changes(self):
artwork = Artworks('Leonardo Da Vinci', 'Mona lisa', 830000000, True)
artwork_db.update_status_artwork(artwork, True)
artwork.save()
artwork.available = False
artwork_db.update_status_artwork(artwork, False)
artwork.save()
self.assertEqual(artwork,
artwork_db.update_status_artwork(artwork.name, False))
def add_new_artwork(name, name_artwork, price, available):
add_artwork = Artworks(artist=name,
artwork_name=name_artwork,
price=price,
available=available)
add_artwork.save()
db.close()
def test_delete_artwork_not_in_db(self):
self.test_add_artworks_data()
self.artwork3 = Artworks('Edvard Munch', 'The Scream', 119900000, True)
with self.assertRaises(ArtworkDataError):
artwork_db.delete_an_artwork(self.artwork3)