From c97074f11c43ebf813cb50f1117972c34b3c4c47 Mon Sep 17 00:00:00 2001 From: Zafer Balkan Date: Wed, 28 Feb 2024 21:46:25 +0200 Subject: [PATCH] Fixed UrlHaus test --- tests/conftest.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/conftest.py b/tests/conftest.py index d3fb064..83706b0 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -60,7 +60,7 @@ def shodan_get_ip(ip, pool) -> ShodanIp: return ShodanIp.model_validate(pool[ip]) -def urlhaus_get_host(entity, pool) -> ShodanIp: +def urlhaus_get_host(entity, pool) -> UrlHaus: """ Mock replacement for Urlhaus()._get_host() """ return UrlHaus.model_validate(pool[entity])