test: fix smolmachines proxy assertions
lint / lint (push) Successful in 1m52s
test / unit (pull_request) Successful in 41s
test / integration (pull_request) Successful in 25s

This commit is contained in:
2026-06-08 23:22:47 -04:00
parent fabcd026af
commit cc1d986a74
@@ -134,19 +134,14 @@ class TestSmolmachinesLaunch(unittest.TestCase):
) )
def test_egress_proxy_reachable_through_tsi_loopback_alias(self): def test_egress_proxy_reachable_through_tsi_loopback_alias(self):
self.assertTrue(
self.plan.agent_proxy_url.startswith("http://127."),
self.plan.agent_proxy_url,
)
r = self.bottle.exec( r = self.bottle.exec(
"printf '%s\n' \"$HTTPS_PROXY\" \"$HTTP_PROXY\"" "printf '%s\n' \"$HTTPS_PROXY\" \"$HTTP_PROXY\""
) )
self.assertEqual(0, r.returncode, msg=r.stderr) self.assertEqual(0, r.returncode, msg=r.stderr)
proxies = [line.strip() for line in r.stdout.splitlines()] proxies = [line.strip() for line in r.stdout.splitlines()]
self.assertEqual( self.assertEqual(2, len(proxies), proxies)
[self.plan.agent_proxy_url, self.plan.agent_proxy_url], self.assertEqual(proxies[0], proxies[1], proxies)
proxies, self.assertTrue(proxies[0].startswith("http://127."), proxies[0])
)
r = self.bottle.exec( r = self.bottle.exec(
"curl -fsS --max-time 20 https://example.com >/dev/null && echo OK" "curl -fsS --max-time 20 https://example.com >/dev/null && echo OK"
@@ -178,6 +173,7 @@ class TestSmolmachinesLaunch(unittest.TestCase):
or "502" in r.stdout or "502" in r.stdout
or "blocked" in r.stdout.lower() or "blocked" in r.stdout.lower()
or "not allowed" in r.stdout.lower() or "not allowed" in r.stdout.lower()
or "not in the bottle's egress.routes allowlist" in r.stdout.lower()
or "forbidden" in r.stdout.lower() or "forbidden" in r.stdout.lower()
or "failed" in r.stdout.lower(), or "failed" in r.stdout.lower(),
f"expected non-allowlisted proxy request to fail; got: {r.stdout!r}", f"expected non-allowlisted proxy request to fail; got: {r.stdout!r}",