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_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