def test_AddDivision(self):
n = Network()
companyid = "company"
companyid2 = "company2"
companyid3 = "company3"
divisionid = "division"
divisionid2 = "division2"
divisionid3 = "division3"
c = Company(companyid)
c3 = Company(companyid3)
d = Division(divisionid, companyid)
d2 = Division(divisionid2, companyid2)
n.AddCompany(c)
n.AddCompany(c3)
n.AddDivision(d)
n.AddDivision(d2)
n.AddDivisionByDef(divisionid3, companyid3)
n.AddDivisionByDef(divisionid, companyid2)
with self.assertRaises(Exception):
n.AddDivisionByDef(divisionid, companyid)
self.assertEqual(n.Companies[companyid].Divisions[divisionid].ID, divisionid)
self.assertEqual(n.Companies[companyid2].Divisions[divisionid2].ID, divisionid2)
self.assertEqual(n.Companies[companyid3].Divisions[divisionid3].ID, divisionid3)
self.assertEqual(len(n.Companies[companyid2].Divisions), 2)
self.assertEqual(n.Companies[companyid2].Divisions[divisionid].ID, divisionid)
return
def test_AddCompany(self):
n = Network()
companyid = "company"
companyid2 = "company"
companyid3 = "company3"
c = Company(companyid)
c2 = Company(companyid2)
n.AddCompany(c)
with self.assertRaises(Exception):
n.AddCompany(c2)
n.AddCompanyByDef(companyid3, False, False)
self.assertEqual(n.Companies[companyid].ID, companyid)
self.assertEqual(n.Companies[companyid3].ID, companyid3)
return
def test_FindorAddCompany(self):
n = Network()
companyid = "company"
companyid3 = "company3"
c = Company(companyid)
n.AddCompany(c)
self.assertTrue(n.FindOrAddCompany(companyid).ID, companyid)
self.assertTrue(n.FindOrAddCompany(companyid3).ID, companyid3)
self.assertTrue(n.Companies[companyid3].ID, companyid3)
return
def test_FindorAddDivision(self):
n = Network()
companyid = "company"
companyid3 = "company3"
divisionid = "division"
divisionid3 = "division3"
c = Company(companyid)
d = Division(divisionid, companyid)
n.AddCompany(c)
n.AddDivision(d)
self.assertTrue(n.FindOrAddDivision(divisionid, companyid).ID, divisionid)
self.assertTrue(n.FindOrAddDivision(divisionid3, companyid3).ID, divisionid3)
self.assertTrue(n.Companies[companyid3].Divisions[divisionid3].ID, divisionid3)
return
