def testGetLabIdByMachNone(self): dptDB = DOMProdTestDB(self.conn) hostName = 'foo.bar.com' expId = None cursor = MockCursor("LabMachNone") tmpName = hostName while True: qry = 'select lab_id from LabMachine where machine="' + tmpName + \ '"' cursor.addExpectedExecute(qry, None) dot = tmpName.find('.') if dot < 0: break tmpName = tmpName[dot + 1:] id = dptDB.getLabIdByMachine(cursor, hostName) self.assertEqual( id, expId, "Expected getLabIdByName to return " + str(expId) + ", not " + str(id)) cursor.verify()
def testGetLabIdByName(self): dptDB = DOMProdTestDB(self.conn) hostName = 'foo.bar.com' expId = 12345 cursor = MockCursor("LabName") qry = 'select lab_id from Laboratory where name="' + hostName + '"' cursor.addExpectedExecute(qry, (expId, )) id = dptDB.getLabIdByName(cursor, hostName) self.assertEqual( id, expId, "Expected getLabIdByName to return " + str(expId) + ", not " + str(id)) cursor.verify()
def testGetLabIdByMach(self): dptDB = DOMProdTestDB(self.conn) pieces = ['foo', 'bar', 'com'] hostName = '.'.join(pieces) expId = 12345 for i in range(len(pieces)): cursor = MockCursor("LabMach#" + str(i)) n = 0 tmpName = hostName while True: qry = 'select lab_id from LabMachine where machine="' + \ tmpName + '"' if n < i: cursor.addExpectedExecute(qry, None) else: cursor.addExpectedExecute(qry, (expId, )) break n = n + 1 dot = tmpName.find('.') if dot < 0: break tmpName = tmpName[dot + 1:] id = dptDB.getLabIdByMachine(cursor, hostName) self.assertEqual( id, expId, "Expected getLabIdByName to return " + str(expId) + ", not " + str(id)) cursor.verify()