From 3409234a4d248f7077ff11c32c2f37bc813017c2 Mon Sep 17 00:00:00 2001 From: Kekoa Kaaikala Date: Fri, 30 Sep 2022 17:25:49 +0000 Subject: [PATCH] UT: Address mypy errors due to get_url_for_resource --- monkey/tests/unit_tests/monkey_island/conftest.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/monkey/tests/unit_tests/monkey_island/conftest.py b/monkey/tests/unit_tests/monkey_island/conftest.py index 094bbb2aa..6b47ccc34 100644 --- a/monkey/tests/unit_tests/monkey_island/conftest.py +++ b/monkey/tests/unit_tests/monkey_island/conftest.py @@ -1,7 +1,7 @@ import os import re from collections.abc import Callable -from typing import Set +from typing import Set, Type import flask_restful import pytest @@ -47,7 +47,7 @@ def mock_flask_resource_manager(container): return flask_resource_manager -def get_url_for_resource(resource: AbstractResource, **kwargs): +def get_url_for_resource(resource: Type[AbstractResource], **kwargs): chosen_url = None for url in resource.urls: if _get_url_keywords(url) == set(kwargs.keys()):