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