def testSingleModule(self):
session = self.createSurveySession()
session.addChild(model.Module(title=u"Root", module_id="1", zodb_path="1"))
tree = update.getSessionTree(session)
self.assertEqual(len(tree), 1)
self.assertEqual(tree[0].zodb_path, "1")
self.assertEqual(tree[0].type, "module")
self.assertEqual(tree[0].path, "001")
def testSingleModule(self):
session = self.createSurveySession()
session.addChild(
model.Module(title="Root", module_id="1", zodb_path="1"))
tree = update.getSessionTree(session)
self.assertEqual(len(tree), 1)
self.assertEqual(tree[0].zodb_path, "1")
self.assertEqual(tree[0].type, "module")
self.assertEqual(tree[0].path, "001")
def testModuleAndRisk(self):
session = self.createSurveySession()
root = model.Module(title=u"Root", module_id="1", zodb_path="1")
session.addChild(root)
root.addChild(model.Risk(title=u"Risk 1", risk_id="1", zodb_path="1/1", type="risk", identification="no"))
tree = update.getSessionTree(session)
self.assertEqual(len(tree), 2)
self.assertEqual(tree[0].zodb_path, "1")
self.assertEqual(tree[0].type, "module")
self.assertEqual(tree[0].path, "001")
self.assertEqual(tree[1].zodb_path, "1/1")
self.assertEqual(tree[1].type, "risk")
self.assertEqual(tree[1].path, "001001")
def testModuleAndRisk(self):
session = self.createSurveySession()
root = model.Module(title="Root", module_id="1", zodb_path="1")
session.addChild(root)
root.addChild(
model.Risk(
title="Risk 1",
risk_id="1",
zodb_path="1/1",
type="risk",
identification="no",
))
tree = update.getSessionTree(session)
self.assertEqual(len(tree), 2)
self.assertEqual(tree[0].zodb_path, "1")
self.assertEqual(tree[0].type, "module")
self.assertEqual(tree[0].path, "001")
self.assertEqual(tree[1].zodb_path, "1/1")
self.assertEqual(tree[1].type, "risk")
self.assertEqual(tree[1].path, "001001")
def testEmpty(self):
session = self.createSurveySession()
self.assertEqual(update.getSessionTree(session), [])
def testEmpty(self):
session = self.createSurveySession()
self.assertEqual(update.getSessionTree(session), [])
