def testSetGetStarts(self, procs, starts):
for p in procs:
ProcTree.register(p)
pt = ProcTree()
pt.setStarts(starts)
for proc in procs:
if proc in starts:
self.assertTrue(ProcTree.NODES[proc].start)
else:
self.assertFalse(ProcTree.NODES[proc].start)
s = pt.getStarts()
self.assertCountEqual(s, starts)
self.assertCountEqual(pt.starts, starts)
def test_proctree_setgetstarts():
p12 = Proc()
p13 = Proc()
p14 = Proc()
#p14.hide = True
p12.depends = p13
ProcTree.register(p13)
ProcTree.register(p14)
pt = ProcTree()
pt.init()
# with pytest.raises(ProcHideError):
# pt.setStarts([p14])
pt.ends = [1, 2, 3]
pt.setStarts([p13])
assert ProcTree.NODES[p12].start == False
assert ProcTree.NODES[p13].start == True
assert ProcTree.NODES[p14].start == False
assert pt.starts == [p13]
assert pt.ends == []
pt.starts = []
assert pt.getStarts() == [p13]
assert pt.starts == [p13]
assert pt.getStarts() == [p13]
