8
8
from squarecloud .data import (
9
9
AppData ,
10
10
Backup ,
11
+ BackupInfo ,
11
12
DeployData ,
13
+ DNSRecord ,
12
14
DomainAnalytics ,
13
15
FileInfo ,
14
16
LogsData ,
17
+ ResumedStatus ,
15
18
StatusData ,
16
19
UserData ,
17
20
)
@@ -25,7 +28,7 @@ def decorator(func):
25
28
async def wrapper (self , client : Client , app : Application ):
26
29
if client .get_listener (endpoint ):
27
30
client .remove_listener (endpoint )
28
- return await func (self , client = client , app = app )
31
+ return await func (self , client , app )
29
32
return wrapper
30
33
return decorator
31
34
@@ -298,16 +301,6 @@ async def test_listener(response: Response):
298
301
)
299
302
@_clear_listener_on_rerun (Endpoint .custom_domain ())
300
303
async def test_set_custom_domain (self , client : Client , app : Application ):
301
- endpoint : Endpoint = Endpoint .custom_domain ()
302
-
303
- if not client .get_request_listener (endpoint ):
304
- @client .on_request (endpoint )
305
- async def test (response : Response ):
306
- assert isinstance (response , Response )
307
-
308
- await client .set_custom_domain (
309
- '6c8e9b785cce4f99984f9ca1c5470d51' , 'test.com.br'
310
- )
311
304
endpoint : Endpoint = Endpoint .github_integration ()
312
305
expected_result : str | None
313
306
expected_response : Response | None = None
@@ -324,6 +317,71 @@ async def test_listener(response: Response):
324
317
assert isinstance (expected_result , str )
325
318
assert isinstance (expected_response , Response )
326
319
320
+ @_clear_listener_on_rerun (Endpoint .all_backups ())
321
+ async def test_all_backups (self , client : Client , app : Application ):
322
+ endpoint : Endpoint = Endpoint .all_backups ()
323
+ expected_result : list [BackupInfo ] | None
324
+ expected_response : Response | None = None
325
+
326
+ @client .on_request (endpoint )
327
+ async def test_listener (response : Response ):
328
+ nonlocal expected_response
329
+ expected_response = response
330
+
331
+ expected_result = await client .all_app_backups (app .id )
332
+ assert isinstance (expected_result , list )
333
+ assert isinstance (expected_result [0 ], BackupInfo )
334
+ assert isinstance (expected_response , Response )
335
+
336
+ @_clear_listener_on_rerun (Endpoint .all_apps_status ())
337
+ async def test_all_apps_status (self , client : Client , app : Application ):
338
+ endpoint : Endpoint = Endpoint .all_apps_status ()
339
+ expected_result : list [ResumedStatus ] | None
340
+ expected_response : Response | None = None
341
+
342
+ @client .on_request (endpoint )
343
+ async def test_listener (response : Response ):
344
+ nonlocal expected_response
345
+ expected_response = response
346
+
347
+ expected_result = await client .all_apps_status ()
348
+ assert isinstance (expected_result , list )
349
+ assert isinstance (expected_result [0 ], ResumedStatus )
350
+ assert isinstance (expected_response , Response )
351
+
352
+ @_clear_listener_on_rerun (Endpoint .move_file ())
353
+ async def test_move_app_file (self , client : Client , app : Application ):
354
+ endpoint : Endpoint = Endpoint .move_file ()
355
+ expected_result : Response | None
356
+ expected_response : Response | None = None
357
+
358
+ @client .on_request (endpoint )
359
+ async def test_listener (response : Response ):
360
+ nonlocal expected_response
361
+ expected_response = response
362
+
363
+ expected_result = await client .move_app_file (
364
+ app .id , 'test.txt' , 'test_move.txt'
365
+ )
366
+ assert isinstance (expected_result , Response )
367
+ assert isinstance (expected_response , Response )
368
+
369
+ @_clear_listener_on_rerun (Endpoint .dns_records ())
370
+ async def test_move_app_file (self , client : Client , app : Application ):
371
+ endpoint : Endpoint = Endpoint .dns_records ()
372
+ expected_result : list [DNSRecord ] | None
373
+ expected_response : Response | None = None
374
+
375
+ @client .on_request (endpoint )
376
+ async def test_listener (response : Response ):
377
+ nonlocal expected_response
378
+ expected_response = response
379
+
380
+ expected_result = await client .dns_records (app .id )
381
+ assert isinstance (expected_result , list )
382
+ assert isinstance (expected_result [0 ], DNSRecord )
383
+ assert isinstance (expected_response , Response )
384
+
327
385
@_clear_listener_on_rerun (endpoint = Endpoint .app_status ())
328
386
async def test_pydantic_cast (self , client : Client , app : Application ):
329
387
class Person (BaseModel ):
0 commit comments