def setUp(self) -> None:
self.basic_tier = self.create_basic_tier()
self.user = CustomUserFactory(account_tier=self.basic_tier)
self.url = reverse('detail', args=(self.user.pk, ))
self.user_image_1 = UserImageFactory(user=self.user,
image__filename='test_1.jpg')
self.user_image_2 = UserImageFactory(user=self.user,
image__filename='test_2.jpg')
def setUp(self) -> None:
self.user = CustomUserFactory()
self.image_1 = UserImageFactory()
self.image_2 = UserImageFactory(user=self.user)
self.image_3 = UserImageFactory()
self.request = RequestFactory()
self.request.user = self.user
self.queryset = UserImage.objects.all()
self.user_filtered_primary_key = UserFilteredPrimaryKeyRelatedField(
queryset=self.queryset)
self.context = {'request': self.request}
self.temporary_image_serializer = TemporaryImageLinkSerializer
def setUp(self) -> None:
self.user = CustomUserFactory()
self.user_image = UserImageFactory(user=self.user)
self.serializer = TemporaryImageLinkSerializer
self.data = {'user_image': self.user_image.pk, 'time_expiration': 1234}
self.request = RequestFactory()
self.request.user = self.user
def setUp(self) -> None:
self.enterprise_tier = self.create_enterprise_tier()
self.user = CustomUserFactory(account_tier=self.enterprise_tier)
self.user_image = UserImageFactory(user=self.user)
self.permission = HasGenerateLinkPermission()
self.request = RequestFactory()
self.request.user = self.user
self.view = APIView.as_view()
def test_image_upload_to(self):
filename = 'test_name.jpg'
today = timezone.now()
user_image = UserImageFactory()
result = image_upload_to(user_image, filename)
expected_result = os.path.join('userdata', 'images', 'user', str(user_image.user.pk), f'{today.strftime("%Y")}',
f'{today.strftime("%m")}', f'{today.strftime("%d")}', filename)
self.assertEqual(expected_result, result)
def test_create_image_not_belong_to_user(self):
self.assertFalse(TemporaryImageLink.objects.exists())
user_image = UserImageFactory()
self.data['user_image'] = user_image.pk
serializer = self.serializer(data=self.data,
context={'request': self.request})
self.assertFalse(serializer.is_valid())
self.assertFalse(TemporaryImageLink.objects.exists())
def test_generate_thumbnails(self):
file_name = 'test_img.jpg'
# instance is created signal is executed
user = CustomUserFactory(account_tier=self.create_premium_tier())
user_image = UserImageFactory(image__witdh=1600, image__height=800, image__filename=file_name, user=user)
base_dir = os.path.dirname(user_image.image.path)
for thumbnail in user.account_tier.thumbnails.all():
file_path = os.path.join(base_dir, thumbnail.thumbnail_prefix + file_name)
self.assertTrue(os.path.exists(file_path))
def test_cache_not_activated(self):
base_images_quantity = self.user.images.count()
self.login(self.user)
resp_1 = self.client.get(self.url)
data_1 = resp_1.json()
self.assertEqual(len(data_1['images']), base_images_quantity)
# add UserImage - response should not be get from cache
self.user_image_3 = UserImageFactory(user=self.user,
image__filename='test_3.jpg')
resp_2 = self.client.get(self.url)
data_2 = resp_2.json()
self.assertNotEqual(data_1, data_2)
def setUp(self) -> None:
self.image = UserImageFactory()
self.serializer = UserImageSerializer