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)
