def test_addProjects(self):
adb = AstridDB(dbname = "turtle_sim", test = True, quiet = True)
self.assertEquals(adb.host, "gbtdata.gbt.nrao.edu")
self.assertEquals(adb.user, "turtle_admin")
self.assertEquals(adb.dbname, "turtle_sim")
# make sure the DB is clean
for acode in self.acodes:
self.assertEquals(False, adb.astridCodeExists(acode))
# here's what we're actually testing
adb.addProjects(self.pcodes)
# make sure they got in, then clean up
for acode in self.acodes:
self.assertEquals(True, adb.astridCodeExists(acode))
adb.removeAstridCode(acode)
self.assertEquals(False, adb.astridCodeExists(acode))
def setUp(self):
# pcodes should get converted to acodes
self.pcodes = ["GBTMy-Project"
, "TGBT-MyLittlePony"
, "GBT12A-001"
, "BB240"
, "VLBA12A-222"
]
self.acodes = ["AGBTMy_Project"
, "TGBT_MyLittlePony"
, "AGBT12A_001"
, "ABB240"
, "AVLB12A_222"
]
# make sure DB is clean
adb = AstridDB(dbname = "turtle_sim", test = True)
for acode in self.acodes:
adb.removeAstridCode(acode)
