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