erivlis / graphinate

    assert len(builder_with_graph._graph.nodes) == 0

def test_populate_nodes_universe_node_parent_handling(builder_with_graph):

    node_type_absolute_id = (UniverseNode, 'child')

    node = DummyNode(key='n6', value=55)

    def generator(**kwargs):

        yield node

    node_model = DummyNodeModel(generator=generator, parent_type=UniverseNode)

    builder_with_graph.model._node_models = {node_type_absolute_id: [node_model]}

    builder_with_graph._populate_nodes(node_type_absolute_id)

    node_id = (node.key,)

    # Lineage should be (node.key,) and no edge should be created

    assert builder_with_graph._graph.nodes[node_id]['lineage'] == [node.key]

    assert len(builder_with_graph._graph.edges) == 0

    assert builder_with_graph._graph.nodes[node_id]['label'] == "label-7"

def test_populate_nodes_handles_invalid_parent_node_id(builder_with_graph):

    node_type_absolute_id = ('parent', 'child')

    node = DummyNode(key='n1', value=1)

    def generator(**kwargs):

        yield node

    node_model = DummyNodeModel(generator=generator, parent_type=UniverseNode)

    builder_with_graph.model._node_models = {node_type_absolute_id: [node_model]}

    builder_with_graph._populate_nodes(node_type_absolute_id)

    node_id = (node.key,)

    # Should not create any edge since parent_type is UniverseNode

    assert len(builder_with_graph._graph.edges) == 0

    assert node_id in builder_with_graph._graph

def test_populate_nodes_adds_new_node(builder_with_graph):