🧪 moved create_zip to __init__.py

Author: robert-nogueira

tests/__init__.py

@@ -1,8 +1,30 @@
+import io
 import os
+import zipfile
 
 from dotenv import load_dotenv
 
 import squarecloud
 
 load_dotenv()
 client = squarecloud.Client(os.getenv('KEY'))
+
+
+def create_zip(include_squarecloud_app: bool = True):
+    buffer = io.BytesIO()
+
+    with zipfile.ZipFile(buffer, 'w') as zip_file:
+        zip_file.write(
+            r'tests\test_upload\requirements.txt', 'requirements.txt'
+        )
+
+        zip_file.write(r'tests\test_upload\main.py', 'main.py')
+
+        if include_squarecloud_app:
+            zip_file.write(
+                r'tests\test_upload\squarecloud.app', 'squarecloud.app'
+            )
+
+    buffer.seek(0)
+
+    return buffer.getvalue()

tests/test_request_listeners.py

@@ -1,10 +1,11 @@
 import pytest
+import squarecloud
 
 from squarecloud import Endpoint, File
 from squarecloud.data import UploadData
 from squarecloud.http import Response
 
-from . import client
+from . import client, create_zip
 
 
 @pytest.mark.asyncio
@@ -16,8 +17,14 @@ async def test_upload(self):
         async def test(response: Response):
             assert isinstance(response, Response)
 
+        squarecloud.create_config_file(
+            r'tests\test_upload',
+            display_name='normal_test',
+            main='main.py',
+            memory=100,
+        )
         upload_data: UploadData = await client.upload_app(
-            File('tests/test_upload/test_normal_upload.zip')
+            File(create_zip(), filename='file.zip')
         )
         TestRequestListeners.APP_ID = upload_data.id
 

tests/test_upload_app.py

@@ -8,27 +8,7 @@
 from squarecloud import File, errors
 from squarecloud.data import UploadData
 
-from . import client
-
-
-def create_zip(include_squarecloud_app: bool = True):
-    buffer = io.BytesIO()
-
-    with zipfile.ZipFile(buffer, 'w') as zip_file:
-        zip_file.write(
-            r'tests\test_upload\requirements.txt', 'requirements.txt'
-        )
-
-        zip_file.write(r'tests\test_upload\main.py', 'main.py')
-
-        if include_squarecloud_app:
-            zip_file.write(
-                r'tests\test_upload\squarecloud.app', 'squarecloud.app'
-            )
-
-    buffer.seek(0)
-
-    return buffer.getvalue()
+from . import client, create_zip
 
 
 @pytest.mark.asyncio