class TestImage(TestCase):
def setUp(self):
self.category = Category(name="Test cat")
self.category.save_category()
self.location = Location(name="Test Loc")
self.location.save_location()
self.image = Image(
name="Test Img",
location_id=self.location,
category_id=self.category,
image=SimpleUploadedFile(
name='image_test.jpg',
content=open(
'photo_gallery/static/images/default_location.jpg',
'rb').read(),
content_type='image/jpeg'))
def test_save_image(self):
self.image.save_image()
images = Image.objects.all()
self.assertTrue(len(images) > 0)
def test_object_instance(self):
self.image.save_image()
images = Image.objects.all()
self.assertIsInstance(images[0], Image)
def test_del_image(self):
self.image.save_image()
self.image.delete_image()
images = Image.objects.all()
self.assertTrue(len(images) == 0)
def test_updated_image(self):
self.image.save_image()
data = dict(name="New Name")
self.image.update_image(self.image.id, data)
self.assertTrue(self.image.name, "New Name")
def test_get_img_by_id(self):
self.image.save_image()
print(self.image)
res = self.image.get_image_by_id(image_id=self.image.id)
self.assertIsInstance(res, Image)
def test_filter_img_by_loc(self):
self.image.save_image()
res = self.image.filter_by_location(location_id=self.location.id)
self.assertTrue(len(res), 1)
self.assertEqual(res[0].location_id.name, self.location.name)
def test_search_image(self):
self.image.save_image()
res = self.image.search_image(category_id=self.category.id)
self.assertTrue(len(res), 1)
self.assertEqual(res[0].category_id.name, self.category.name)
class TestLocation(TestCase):
def setUp(self):
self.location = Location(name="Test Location")
def test_save_location(self):
self.location.save_location()
locations = Location.objects.all()
self.assertTrue(len(locations) > 0)
def test_object_instance(self):
self.location.save_location()
locations = Location.objects.all()
self.assertIsInstance(locations[0], Location)
def test_del_location(self):
self.location.save_location()
self.location.delete_location()
locations = Location.objects.all()
self.assertTrue(len(locations) == 0)
def test_updated_location(self):
self.location.save_location()
self.location.update_location(name="New Name")
self.assertTrue(self.location.name, "New Name")