def test_predictor_get_all_by_sf_id(self): values = get_fake_predictor("1") predictor1 = db_api.predictor_create(self.conf, values) values = get_fake_predictor("2") predictor2 = db_api.predictor_create(self.conf, values) predictors1 = db_api.predictor_get_all_by_sf_id(self.conf, "1") predictors2 = db_api.predictor_get_all_by_sf_id(self.conf, "2") self.assertEqual([dict(predictor1.iteritems())], [dict(pr.iteritems()) for pr in predictors1]) self.assertEqual([dict(predictor2.iteritems())], [dict(pr.iteritems()) for pr in predictors2])
def test_predictor_get_all_by_sf_id(self): values = get_fake_predictor('1') predictor1 = db_api.predictor_create(self.conf, values) values = get_fake_predictor('2') predictor2 = db_api.predictor_create(self.conf, values) predictors1 = db_api.predictor_get_all_by_sf_id(self.conf, '1') predictors2 = db_api.predictor_get_all_by_sf_id(self.conf, '2') self.assertEqual(len(predictors1), 1) self.assertEqual(predictors1[0]['id'], predictor1['id']) self.assertEqual(len(predictors2), 1) self.assertEqual(predictors2[0]['id'], predictor2['id'])
def create_server_farm(ctx, sf): try: pr = db_api.predictor_get_all_by_sf_id(ctx.conf, sf['id']) ctx.device.create_server_farm(sf, pr) db_api.serverfarm_update(ctx.conf, sf['id'], {'deployed': True}) yield except Exception: delete_server_farm(ctx, sf) raise
def create_server_farm(ctx, sf): try: pr = db_api.predictor_get_all_by_sf_id(ctx.conf, sf["id"]) ctx.device.create_server_farm(sf, pr) sf["deployed"] = "True" db_api.serverfarm_update(ctx.conf, sf["id"], sf) yield except Exception: delete_server_farm(ctx, sf) raise
def loadFromDB(self, lb_id): self.lb = db_api.loadbalancer_get(self.conf, lb_id) self.sf = db_api.serverfarm_get_all_by_lb_id(self.conf, lb_id)[0] sf_id = self.sf['id'] self.vips = db_api.virtualserver_get_all_by_sf_id(self.conf, sf_id) predictor = db_api.predictor_get_all_by_sf_id(self.conf, sf_id)[0] self.sf._predictor = predictor self.rs = db_api.server_get_all_by_sf_id(self.conf, sf_id) self.sf._rservers = [] for rs in self.rs: self.sf._rservers.append(rs) self.probes = db_api.probe_get_all_by_sf_id(self.conf, sf_id) self.sf._probes = [] for prob in self.probes: self.sf._probes.append(prob) sticks = db_api.sticky_get_all_by_sf_id(self.conf, sf_id) self.sf._sticky = [] for st in sticks: self.sf._sticky.append(st)