diff --git a/unittests/test_crawler.py b/unittests/test_crawler.py index ad69c6f57cbc8d48d194507d7c1aa79c9da7521b..7b19e9aedb4f60ccb6a509103a01c458b1b08a32 100644 --- a/unittests/test_crawler.py +++ b/unittests/test_crawler.py @@ -399,7 +399,8 @@ def test_split_into_inserts_and_updates_with_complex(crawler_mocked_identifiable # TODO write test where the unresoled entity is not part of the identifiable - +@patch("caoscrawler.identifiable_adapters._retrieve_RecordType", + new=Mock(side_effect=lambda id, name: db.RecordType(id=id,name=name))) @patch("caoscrawler.crawl.cached_get_entity_by", new=Mock(side_effect=mock_get_entity_by)) @patch("caoscrawler.identifiable_adapters.cached_query", @@ -594,7 +595,8 @@ def test_replace_entities_with_ids(): assert a.get_property("B").value == 12345 assert a.get_property("C").value == [12345, 233324] - +@patch("caoscrawler.identifiable_adapters._retrieve_RecordType", + new=Mock(side_effect=lambda id, name: db.RecordType(id=id,name=name))) @patch("caoscrawler.crawl.cached_get_entity_by", new=Mock(side_effect=mock_get_entity_by)) @patch("caoscrawler.identifiable_adapters.cached_get_entity_by", @@ -619,7 +621,8 @@ def test_synchronization_no_commit(upmock, insmock): assert len(ins) == 1 assert len(ups) == 1 - +@patch("caoscrawler.identifiable_adapters._retrieve_RecordType", + new=Mock(side_effect=lambda id, name: db.RecordType(id=id,name=name))) @patch("caoscrawler.crawl.cached_get_entity_by", new=Mock(side_effect=mock_get_entity_by)) @patch("caoscrawler.identifiable_adapters.cached_get_entity_by", diff --git a/unittests/test_sync_graph.py b/unittests/test_sync_graph.py index 06f0dfb9eb3d3536d26dcfd354ca27f08ef99a02..3c838d571b40dadb96ccf787a53af72e35277367 100644 --- a/unittests/test_sync_graph.py +++ b/unittests/test_sync_graph.py @@ -139,7 +139,8 @@ def test_create_reference_mapping(): assert forward_references_backref[id(b)] == set([a]) assert backward_references_backref[id(b)] == set() - +@patch("caoscrawler.identifiable_adapters._retrieve_RecordType", + new=Mock(side_effect=lambda id, name: db.RecordType(id=id,name=name))) @patch("caoscrawler.sync_graph.cached_get_entity_by", new=Mock(side_effect=mock_get_entity_by)) def test_SyncGraph_init(): @@ -203,7 +204,8 @@ def test_SyncGraph_init(): if not st._identity_relies_on_unchecked_entity(el): assert el.identifiable is not None or el.id is not None - +@patch("caoscrawler.identifiable_adapters._retrieve_RecordType", + new=Mock(side_effect=lambda id, name: db.RecordType(id=id,name=name))) @patch("caoscrawler.identifiable_adapters.get_children_of_rt", new=Mock(side_effect=lambda x: [x])) def test_merge_into_trivial(simple_adapter): @@ -281,7 +283,8 @@ def test_merge_into_trivial(simple_adapter): assert len(st.backward_references_backref[id(se_c)]) == 1 assert se_b in st.backward_references_backref[id(se_c)] - +@patch("caoscrawler.identifiable_adapters._retrieve_RecordType", + new=Mock(side_effect=lambda id, name: db.RecordType(id=id,name=name))) @patch("caoscrawler.identifiable_adapters.get_children_of_rt", new=Mock(side_effect=lambda x: [x])) def test_merge_into_simple(simple_adapter): @@ -362,7 +365,8 @@ def test_merge_into_simple(simple_adapter): assert len(st.backward_references_backref[id(se_c)]) == 1 se_b in st.backward_references_backref[id(se_c)] - +@patch("caoscrawler.identifiable_adapters._retrieve_RecordType", + new=Mock(side_effect=lambda id, name: db.RecordType(id=id,name=name))) @patch("caoscrawler.identifiable_adapters.get_children_of_rt", new=Mock(side_effect=lambda x: [x])) def test_backward_references_backref(): @@ -483,7 +487,8 @@ def test_set_id_of_node(simple_adapter): assert st.nodes[0].identifiable is not None assert len(st.nodes) == 2 - +@patch("caoscrawler.identifiable_adapters._retrieve_RecordType", + new=Mock(side_effect=lambda id, name: db.RecordType(id=id,name=name))) @patch("caoscrawler.sync_graph.cached_get_entity_by", new=Mock(side_effect=mock_get_entity_by)) def test_merging(simple_adapter):