Update tests

deedy5 · Jan 26, 2025 · e5c4fbd · e5c4fbd
1 parent 08b3c9b
commit e5c4fbd
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 37 deletions.
diff --git a/tests/test_client.py b/tests/test_client.py
@@ -240,29 +240,23 @@ def test_client_post_files(test_files):
     assert json_data["files"] == {"file1": "aaa111", "file2": "bbb222"}
 
 
-@retry()
-def test_client_impersonate_chrome130():
-    client = primp.Client(
-        impersonate="chrome_130",
-    )
-    # response = client.get("https://tls.peet.ws/api/all")
-    response = client.get("https://tls.http.rw/api/clean")
-    assert response.status_code == 200
-    json_data = response.json()
-    assert json_data["ja4"] == "t13d1516h2_8daaf6152771_b1ff8ab2d16f"
-    assert json_data["akamai_hash"] == "90224459f8bf70b7d0a8797eb916dbc9"
-    assert json_data["peetprint_hash"] == "b8ce945a4d9a7a9b5b6132e3658fe033"
-
-
 @retry()
 def test_client_impersonate_chrome131():
     client = primp.Client(
         impersonate="chrome_131",
+        impersonate_os="windows",
     )
     # response = client.get("https://tls.peet.ws/api/all")
-    response = client.get("https://tls.http.rw/api/clean")
+    response = client.get("https://tls.http.rw/api/all")
     assert response.status_code == 200
     json_data = response.json()
-    assert json_data["ja4"] == "t13d1516h2_8daaf6152771_b1ff8ab2d16f"
-    assert json_data["akamai_hash"] == "90224459f8bf70b7d0a8797eb916dbc9"
-    assert json_data["peetprint_hash"] == "7466733991096b3f4e6c0e79b0083559"
+    assert (
+        json_data["user_agent"]
+        == "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36"
+    )
+    assert json_data["tls"]["ja4"] == "t13d1516h2_8daaf6152771_b1ff8ab2d16f"
+    assert (
+        json_data["http2"]["akamai_fingerprint_hash"]
+        == "90224459f8bf70b7d0a8797eb916dbc9"
+    )
+    assert json_data["tls"]["peetprint_hash"] == "7466733991096b3f4e6c0e79b0083559"
diff --git a/tests/test_defs.py b/tests/test_defs.py
@@ -286,28 +286,22 @@ def test_put():
 
 
 @retry()
-def test_get_impersonate_chrome130():
+def test_get_impersonate_firefox133():
     response = primp.get(
         # "https://tls.peet.ws/api/clean",
-        "https://tls.http.rw/api/clean",
-        impersonate="chrome_130",
+        "https://tls.http.rw/api/all",
+        impersonate="firefox_133",
+        impersonate_os="linux",
     )
     assert response.status_code == 200
     json_data = response.json()
-    assert json_data["ja4"] == "t13d1516h2_8daaf6152771_b1ff8ab2d16f"
-    assert json_data["akamai_hash"] == "90224459f8bf70b7d0a8797eb916dbc9"
-    assert json_data["peetprint_hash"] == "b8ce945a4d9a7a9b5b6132e3658fe033"
-
-
-@retry()
-def test_get_impersonate_chrome131():
-    response = primp.get(
-        # "https://tls.peet.ws/api/clean",
-        "https://tls.http.rw/api/clean",
-        impersonate="chrome_131",
+    assert (
+        json_data["user_agent"]
+        == "Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:133.0) Gecko/20100101 Firefox/133.0"
     )
-    assert response.status_code == 200
-    json_data = response.json()
-    assert json_data["ja4"] == "t13d1516h2_8daaf6152771_b1ff8ab2d16f"
-    assert json_data["akamai_hash"] == "90224459f8bf70b7d0a8797eb916dbc9"
-    assert json_data["peetprint_hash"] == "7466733991096b3f4e6c0e79b0083559"
+    assert json_data["tls"]["ja4"] == "t13d1716h2_5b57614c22b0_bed828528d07"
+    assert (
+        json_data["http2"]["akamai_fingerprint_hash"]
+        == "87a68d774464f75f8b88395030c79a80"
+    )
+    assert json_data["tls"]["peetprint_hash"] == "199f9cf4a47bfc51995a9f3942190094"