diff --git a/tests/test_crawler.py b/tests/test_crawler.py index 9267c50fd8adafc82844167e04d143c6ce0d4d0a..5421a30ca2a42e48637bfc0598cbbcd12ce28539 100644 --- a/tests/test_crawler.py +++ b/tests/test_crawler.py @@ -36,7 +36,7 @@ async def test_check(db_conn, populated_db, session, mocker, coroutine_mock): crawler, "clean_check", return_value={"cleaned": "check"} ) save_check = mocker.patch.object(crawler, "save_check", coroutine_mock()) - await crawler.check(db_conn, session, "test.domain") + await crawler.check(session, "test.domain") fetch_nodeinfo.assert_called_once_with(session, "test.domain") clean_nodeinfo.assert_called_once_with({"hello": "world"}) clean_check.assert_called_once_with( @@ -54,7 +54,7 @@ async def test_check_nodeinfo_connection_error( coroutine_mock(side_effect=aiohttp.client_exceptions.ClientError), ) save_check = mocker.patch.object(crawler, "save_check", coroutine_mock()) - await crawler.check(db_conn, session, "test.domain") + await crawler.check(session, "test.domain") fetch_nodeinfo.assert_called_once_with(session, "test.domain") save_check.assert_called_once_with({"domain": "test.domain", "up": False})