def test_set_depot_uuid(self): service = TcdsService(Utils(Resolv()), self.var) uuidstr = self.depot_uuid depot = Depot(service, uuidstr) self.var.add_depot(depot, uuidstr, 30, depot.CONSTANTS['STATE_OFFLINE']) self.assertEquals(self.var.get_depot_state(depot), depot.CONSTANTS['STATE_OFFLINE']) uuidstr2 = self.depot_uuid2 self.var.set_depot_uuid(depot, uuidstr2) self.assertRaises(KeyError, self.var.get_depot_state, depot) depot.uuid = uuidstr2 self.assertEquals(self.var.get_depot_state(depot), depot.CONSTANTS['STATE_OFFLINE']) self.var.del_depot(depot)