diff --git a/unittests/test_crawler.py b/unittests/test_crawler.py index d48333aae40cddf30c29c732e83b3816eb7236ff..484cd360553667ad8588af8476636e8ff81942cf 100644 --- a/unittests/test_crawler.py +++ b/unittests/test_crawler.py @@ -686,7 +686,7 @@ def test_split_into_inserts_and_updates_backref(crawler_mocked_for_backref_test) @patch("caoscrawler.identifiable_adapters.get_children_of_rt", - new=Mock(side_effect=id)) + new=Mock(side_effect=lambda x: [x])) def test_split_into_inserts_and_updates_mult_backref(crawler_mocked_for_backref_test): # test whether multiple references of the same record type are correctly used crawler = crawler_mocked_for_backref_test @@ -708,7 +708,7 @@ def test_split_into_inserts_and_updates_mult_backref(crawler_mocked_for_backref_ @patch("caoscrawler.identifiable_adapters.get_children_of_rt", - new=Mock(side_effect=id)) + new=Mock(side_effect=lambda x: [x])) def test_split_into_inserts_and_updates_diff_backref(crawler_mocked_for_backref_test): # test whether multiple references of the different record types are correctly used crawler = crawler_mocked_for_backref_test