import pytest


@pytest.mark.usefixtures("state")
def test_read_dict_keys_no_exists(state):
    assert state.read_dict_keys(100000) == None


@pytest.mark.usefixtures("state")
def test_read_dict_keys_simple(state):
    a = state.create_node()
    b = state.create_node()
    c = state.create_nodevalue("f")
    d = state.create_edge(a, b)
    e = state.create_edge(d, c)
    assert a != None
    assert b != None
    assert c != None
    assert d != None
    assert e != None

    l = state.read_dict_keys(a)
    assert l != None
    assert set(l) == set([c])


@pytest.mark.usefixtures("state")
def test_read_dict_keys_multi(state):
    a = state.create_node()
    b = state.create_node()
    c = state.create_nodevalue("f")
    d = state.create_edge(a, b)
    e = state.create_edge(d, c)
    assert a != None
    assert b != None
    assert c != None
    assert d != None
    assert e != None

    g = state.create_node()
    h = state.create_nodevalue("k")
    i = state.create_edge(a, g)
    j = state.create_edge(i, h)
    assert g != None
    assert h != None
    assert i != None
    assert j != None

    l = state.read_dict_keys(a)
    assert l != None
    assert set(l) == set([c, h])