def test_nodewalk_bfs():
cf = CamFlowBackend(open(BASIC_MSGBUF, 'rb'))
it = cf.walk_nodes(TraversalOrder.BFS)
node = next(it)
assert node['id'] == 218451
node = next(it)
assert node['id'] == 359384
node = next(it)
assert node['id'] == 359380
node = next(it)
assert node['id'] == 218450 # child of 218451
node = next(it)
assert node['id'] == 31996 # child of 349384
node = next(it)
assert node['id'] == 359389 # child of 349384
node = next(it)
assert node['id'] == 359379 # child of 359380
try:
next(it)
assert 0, 'iterator should have no more elements'
except StopIteration:
pass
def test_unicode():
cf = CamFlowBackend(open(UNICODE_MSGBUF, 'rb'))
it = cf.walk_nodes(TraversalOrder.BFS)
node = next(it)
assert node['id'] == 218451
assert node['path'] == '/usr/bin/🐎'
node = next(it)
assert node['id'] == 218450
try:
next(it)
assert 0, 'iterator should have no more elements'
except StopIteration:
pass
def test_nodewalk_dfs():
cf = CamFlowBackend(open(BASIC_MSGBUF, 'rb'))
it = cf.walk_nodes(TraversalOrder.DFS)
node = next(it)
assert node['id'] == 218451
assert node['type'] == 'path'
assert node['path'] == '/usr/bin/grep'
node = next(it)
assert node['id'] == 218450 # child of 218451
assert node['type'] == 'file'
assert node['inode'] == 793325
node = next(it)
assert node['id'] == 359384
assert node['type'] == 'file'
node = next(it)
assert node['id'] == 359389 # child of 349384
assert node['type'] == 'file'
node = next(it)
assert node['id'] == 31996 # child of 349384
assert node['type'] == 'task'
node = next(it)
assert node['id'] == 359380
assert node['type'] == 'path'
assert node['path'] == '/525/cgroup'
node = next(it)
assert node['id'] == 359379 # child of 359380
assert node['type'] == 'file'
assert node['inode'] == 144143
assert node['mode'] == 33060
try:
next(it)
assert 0, 'iterator should have no more elements'
except StopIteration:
pass
def test_edgewalk_dfs():
cf = CamFlowBackend(open(EDGES_MSGBUF, 'rb'))
it = cf.walk_edges(TraversalOrder.DFS)
(edge, from_, to) = next(it)
assert edge['id'] == 1
assert from_['id'] == 1
assert to['id'] == 2
assert edge['type'] == 'named'
assert edge['allowed'] == 0
(edge, from_, to) = next(it)
assert edge['id'] == 2
assert from_['id'] == 2
assert to['id'] == 3
(edge, from_, to) = next(it)
assert edge['id'] == 3
assert from_['id'] == 2
assert to['id'] == 4
(edge, from_, to) = next(it)
assert edge['id'] == 4
assert from_['id'] == 1
assert to['id'] == 5
(edge, from_, to) = next(it)
assert edge['id'] == 5
assert from_['id'] == 6
assert to['id'] == 5
(edge, from_, to) = next(it)
assert edge['id'] == 6
assert from_['id'] == 6
assert to['id'] == 7
try:
next(it)
assert 0, 'iterator should have no more elements'
except StopIteration:
pass
def __init__(self):
BASIC_MSGBUF = os.path.join(os.path.dirname(__file__), "..", "test",
"inputs", "basic.txt")
self.graph = CamFlowBackend(open(BASIC_MSGBUF, 'rb'))
def test_empty():
cf = CamFlowBackend(open(EMPTY_MSGBUF, 'rb'))
assert [] == list(cf.walk_nodes(TraversalOrder.DFS))
def test_drop_nulls():
cf = CamFlowBackend(open(HAS_NULL_MSGBUF, 'rb'))
assert len(list(cf.walk_nodes(TraversalOrder.DFS))) == 2
def test_smoke():
"""Simple smoke test to make sure nothing blows up."""
cf = CamFlowBackend()
for _ in cf.walk_nodes(TraversalOrder.DFS):
pass