fix: Restore expected responses from Lacus, fix tests

Author: Rafiot

poetry.lock

@@ -45,7 +45,7 @@ class CaptureResponse(TypedDict, total=False):
     status: int
     last_redirected_url: str | None
     har: dict[str, Any] | None
-    cookies: list[Cookie] | None
+    cookies: list[dict[str, Any]] | None
     storage: dict[str, Any] | None
     error: str | None
     html: str | None
@@ -65,7 +65,7 @@ class CaptureResponseJson(TypedDict, total=False):
     status: int
     last_redirected_url: str | None
     har: dict[str, Any] | None
-    cookies: list[Cookie] | None
+    cookies: list[dict[str, Any]] | None
     storage: dict[str, Any] | None
     error: str | None
     html: str | None

pylacus/api.py

@@ -5,8 +5,8 @@
 import unittest
 
 from datetime import datetime
+from typing import Any
 
-from lookyloo_models import Cookie
 from pylacus import PyLacus
 from pylacus.api import CaptureStatus
 
@@ -60,12 +60,12 @@ def test_submit_cookies(self) -> None:
         if not response['storage'] or 'cookies' not in response['storage']:
             self.fail("No storage found")
         self.assertEqual(len(response['cookies']), 1)
-        self.assertEqual(response['cookies'][0].domain, 'circl.lu')
+        self.assertEqual(response['cookies'][0]['domain'], 'circl.lu')
         self.assertEqual(response['cookies'], response['storage']['cookies'])
 
-        cookie_from_capture: Cookie = response['cookies'][0]
+        cookie_from_capture = response['cookies'][0]
         expires = datetime.now().timestamp() + 3600
-        cookie_from_capture.expires = expires
+        cookie_from_capture['expires'] = expires
         uuid = self.client.enqueue(url="circl.lu", cookies=[cookie_from_capture], max_retries=0)
         while True:
             status = self.client.get_capture_status(uuid)
@@ -75,12 +75,12 @@ def test_submit_cookies(self) -> None:
         response = self.client.get_capture(uuid)
         if not response['cookies']:
             self.fail("No cookies found")
-        self.assertEqual(response['cookies'][0].expires, expires)
+        self.assertEqual(response['cookies'][0]['expires'], expires)
 
         # Send list
-        cookies_from_capture: list[Cookie] = response['cookies']
+        cookies_from_capture: list[dict[str, Any]] = response['cookies']
         expires = datetime.now().timestamp() + 4000
-        cookies_from_capture[0].expires = expires
+        cookies_from_capture[0]['expires'] = expires
         uuid = self.client.enqueue(url="circl.lu", cookies=cookies_from_capture, max_retries=0)
         while True:
             status = self.client.get_capture_status(uuid)
@@ -90,7 +90,7 @@ def test_submit_cookies(self) -> None:
         response = self.client.get_capture(uuid)
         if not response['cookies']:
             self.fail("No cookies found")
-        self.assertEqual(response['cookies'][0].expires, expires)
+        self.assertEqual(response['cookies'][0]['expires'], expires)
 
         # Send list as json
         cookies_from_capture = response['cookies']
@@ -102,7 +102,7 @@ def test_submit_cookies(self) -> None:
             time.sleep(5)
         response = self.client.get_capture(uuid)
         if response['cookies']:
-            self.assertEqual(response['cookies'][0].expires, expires, response['cookies'])
+            self.assertEqual(response['cookies'][0]['expires'], expires, response['cookies'])
 
     def test_submit_storage(self) -> None:
         uuid = self.client.enqueue(url="https://mdn.github.io/dom-examples/web-storage/",