Code improvements

Author: LinaSSavova

styleguide_example/files/tests/flows/test_standard_upload.py

@@ -1,3 +1,4 @@
+import shutil
 from django.conf import settings
 from django.test import TestCase, override_settings
 from django.urls import reverse
@@ -34,7 +35,7 @@ def test_standard_upload(self):
 
         # Create a user
         credentials = {
-            "email": "some_email@hacksoft.io",
+            "email": "test@hacksoft.io",
             "password": "123456"
         }
         user_create(**credentials)
@@ -57,9 +58,7 @@ def test_standard_upload(self):
         )
 
         with self.subTest("1. Upload a file, below the size limit, assert models gets created accordingly"):
-            response = self.client.post(
-                self.standard_upload_url, {"file": file_1}, enctype="multipart/form-data", **auth_headers
-            )
+            response = self.client.post(self.standard_upload_url, {"file": file_1}, **auth_headers)
 
             self.assertEqual(201, response.status_code)
             self.assertEqual(1, File.objects.count())
@@ -70,26 +69,25 @@ def test_standard_upload(self):
         )
 
         with self.subTest("2. Upload a file, above the size limit, assert API error, nothing gets created"):
-            response = self.client.post(
-                self.standard_upload_url, {"file": file_2}, enctype="multipart/form-data", **auth_headers
-            )
+            response = self.client.post(self.standard_upload_url, {"file": file_2}, **auth_headers)
 
             self.assertEqual(400, response.status_code)
             self.assertEqual(1, File.objects.count())
 
         # Create a file equal to the size limit
         file_3 = SimpleUploadedFile(
-            name="file_equal.txt", content=file_max_size * "b".encode(), content_type="text/plain"
+            name="file_equal.txt", content=file_max_size * "a".encode(), content_type="text/plain"
         )
 
         with self.subTest("3. Upload a file, equal to the size limit, assert models gets created accordingly"):
-            response = self.client.post(
-                self.standard_upload_url, {"file": file_3}, enctype="multipart/form-data", **auth_headers
-            )
+            response = self.client.post(self.standard_upload_url, {"file": file_3}, **auth_headers)
 
             self.assertEqual(201, response.status_code)
             self.assertEqual(2, File.objects.count())
 
+    def tearDown(self):
+        shutil.rmtree(settings.MEDIA_ROOT, ignore_errors=True)
+
 
 class StandardUploadAdminTests(TestCase):
     """
@@ -123,7 +121,7 @@ def test_standard_admin_upload_and_update(self):
 
         # Create a superuser
         credentials = {
-            "email": "admin_email@hacksoft.io",
+            "email": "test@hacksoft.io",
             "password": "123456",
             "is_admin": True,
             "is_superuser": True
@@ -171,7 +169,7 @@ def test_standard_admin_upload_and_update(self):
             self.assertEqual(file_2.name, File.objects.last().original_file_name)
 
         file_3 = SimpleUploadedFile(
-            name="oversized_file.txt", content=(file_max_size + 10) * "b".encode(), content_type="text/plain"
+            name="oversized_file.txt", content=(file_max_size + 1) * "a".encode(), content_type="text/plain"
         )
 
         data_oversized_file = {
@@ -188,14 +186,14 @@ def test_standard_admin_upload_and_update(self):
             self.assertEqual(file_2.name, File.objects.last().original_file_name)
 
         file_4 = SimpleUploadedFile(
-            name="new_oversized_file.txt", content=(file_max_size + 20) * "c".encode(), content_type="text/plain"
+            name="new_oversized_file.txt", content=(file_max_size + 1) * "a".encode(), content_type="text/plain"
         )
 
         data_new_oversized_file = {
             "file": file_4,
             "uploaded_by": user.id
         }
-
+        # Comment here
         with self.subTest(
             "4. Update an existing file with an oversized one via the Django admin, assert error, nothing gets created"
         ):
@@ -205,3 +203,6 @@ def test_standard_admin_upload_and_update(self):
             self.assertContains(response_2, "File is too large")
             self.assertEqual(1, File.objects.count())
             self.assertEqual(file_2.name, File.objects.last().original_file_name)
+
+    def tearDown(self):
+        shutil.rmtree(settings.MEDIA_ROOT, ignore_errors=True)