mirror of
https://github.com/Horizontal42/netsleuth.git
synced 2026-11-11 02:12:04 +03:00
Merge 1c9fd17d0f into 8c99ac78b7
This commit is contained in:
commit
85f7ca563d
2 changed files with 12 additions and 0 deletions
|
|
@ -124,6 +124,9 @@ async def tier_ookla(
|
|||
iface_name: str | None = None,
|
||||
ipv4: str | None = None,
|
||||
) -> SpeedResult:
|
||||
if server and server.startswith("-"):
|
||||
raise ValueError("Invalid server argument: cannot start with '-'")
|
||||
|
||||
args = [binary, "--format=json", "--accept-license", "--accept-gdpr"]
|
||||
args += ookla_interface_args(iface_name, ipv4)
|
||||
if server:
|
||||
|
|
|
|||
|
|
@ -287,6 +287,15 @@ async def test_tier_ookla_success_parses_json():
|
|||
assert result.idle_rtt_ms == 15.5
|
||||
|
||||
|
||||
|
||||
|
||||
async def test_tier_ookla_server_argument_injection():
|
||||
with pytest.raises(
|
||||
ValueError, match="Invalid server argument: cannot start with '-'"
|
||||
):
|
||||
await tier_ookla("speedtest", "--hacked", 10.0)
|
||||
|
||||
|
||||
async def test_tier_ookla_server_args():
|
||||
mock_proc = AsyncMock()
|
||||
mock_payload = {}
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue