DemonGatanjieuandClaude Opus 5.5
6fb69294f2
On an AutoDL server civitai.com timed out while civitai.red, Civitai's other address with the same API, answered in about a second, so scans there fell back to guessing from the files. civitai_client.api_hosts gives both addresses, the one that answered last first; the scan's lookups, import's and Model Check's (download_sources._civitai_json) and the network test ask the other one when the first does not answer at all. An HTTP error is Civitai's own answer and is not asked again. The test answers which host answered, and Settings says so when it was civitai.red. The pictures stay on image.civitai.com, which such networks often cannot reach either: after COVERS_DOWN_AFTER (2) covers in a row that would not download, a scan fetches no more covers instead of waiting on each. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
62 lines
2.7 KiB
Python
62 lines
2.7 KiB
Python
"""The first-run setup's question: does this computer reach Civitai, and is a set key refused?"""
|
|
import io
|
|
import sys
|
|
import unittest
|
|
import urllib.error
|
|
from pathlib import Path
|
|
from unittest import mock
|
|
|
|
PLUGIN = Path(__file__).resolve().parents[1]
|
|
sys.path[:0] = [str(PLUGIN), str(PLUGIN.parents[1])]
|
|
|
|
from api import local_guard, network_check
|
|
|
|
|
|
class _Answer(io.BytesIO):
|
|
def __enter__(self):
|
|
return self
|
|
|
|
def __exit__(self, *exc):
|
|
return False
|
|
|
|
|
|
def _http_error(code):
|
|
return urllib.error.HTTPError("https://civitai.com" + network_check.PROBE_PATH, code, "x", {}, io.BytesIO(b""))
|
|
|
|
|
|
class CivitaiCheckTests(unittest.TestCase):
|
|
def check(self, *answers, key=None):
|
|
"""The check, with civitai.com answering answers[0] and civitai.red answers[1] (if asked)."""
|
|
headers = {"User-Agent": "t", **({"Authorization": f"Bearer {key}"} if key else {})}
|
|
with mock.patch.object(network_check, "civitai_headers", return_value=headers), \
|
|
mock.patch("civitai_client._answered", ["civitai.com"]), \
|
|
mock.patch.object(network_check.urllib.request, "urlopen", side_effect=list(answers)) as opened:
|
|
result = network_check.check_civitai()
|
|
self.assertEqual(opened.call_args.kwargs["timeout"], network_check.TIMEOUT)
|
|
return result
|
|
|
|
def test_reachable_with_and_without_a_key(self):
|
|
self.assertEqual((lambda r: (r["reachable"], r["key"]))(self.check(_Answer(b"{}"))), (True, "none"))
|
|
self.assertEqual((lambda r: (r["reachable"], r["key"]))(self.check(_Answer(b"{}"), key="k")), (True, "set"))
|
|
|
|
def test_a_refused_key_is_told_apart_from_no_network(self):
|
|
refused = self.check(_http_error(401), key="bad")
|
|
self.assertEqual((refused["reachable"], refused["key"]), (True, "refused"))
|
|
down = self.check(urllib.error.URLError("timed out"), urllib.error.URLError("refused"))
|
|
self.assertEqual((down["reachable"], down["key"], down["host"]), (False, "none", ""))
|
|
self.assertIn("timed out", down["error"]) # why civitai.com, the usual address, failed
|
|
self.assertFalse(self.check(_http_error(503))["reachable"])
|
|
|
|
def test_civitai_red_answering_is_reachable_and_named(self):
|
|
# Some networks (a cloud server in mainland China) reach only Civitai's other address.
|
|
result = self.check(urllib.error.URLError("timed out"), _Answer(b"{}"))
|
|
self.assertEqual((result["reachable"], result["host"]), (True, "civitai.red"))
|
|
self.assertEqual(self.check(_Answer(b"{}"))["host"], "civitai.com")
|
|
|
|
def test_it_answers_this_computer_only(self):
|
|
self.assertIn("/anomalous/civitai_check", local_guard.LOCAL_GETS)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|