def test_load_balancer_update_domains(self):
self.truncate_all_tables()
new_domains = [
{'name': 'www.hao.com'},
{'name': 'www.xyz.com'},
]
db.load_balancer_create(self.context, self.configs)
actual = db.load_balancer_get(self.context, self.load_balancer_id)
db.load_balancer_update_domains(self.context,
self.load_balancer_id, new_domains)
actual = db.load_balancer_get(self.context, self.load_balancer_id)
actual_domains = map(lambda x: x['name'], actual.domains)
expect_domains = map(lambda x: x['name'], new_domains)
self.assertEqual(expect_domains.sort(), actual_domains.sort())
def test_load_balancer_update_state(self):
self.truncate_all_tables()
db.load_balancer_create(self.context, self.configs)
db.load_balancer_update_state(self.context,
self.load_balancer_id, 'active')
actual = db.load_balancer_get(self.context, self.load_balancer_id)
self.assertEqual(actual.state, 'active')
def test_load_balancer_update_instances(self):
self.truncate_all_tables()
new_instances = [
{
'instance_uuid': 'inst-1',
'instance_ip': '192.168.1.1',
},
{
'instance_uuid': 'inst-2',
'instance_ip': '192.168.1.2',
},
]
db.load_balancer_create(self.context, self.configs)
actual = db.load_balancer_get(self.context, self.load_balancer_id)
db.load_balancer_update_instances(self.context,
self.load_balancer_id, new_instances)
actual = db.load_balancer_get(self.context, self.load_balancer_id)
actual_instances = map(lambda x: x['instance_uuid'], actual.instances)
expect_instances = map(lambda x: x['instance_uuid'], new_instances)
self.assertEqual(expect_instances.sort(), actual_instances.sort())
def test_load_balancer_update_config(self):
self.truncate_all_tables()
db.load_balancer_create(self.context, self.configs)
self.config['balancing_method'] = 'test_method'
self.config['health_check_target_path'] = '/index.html'
db.load_balancer_update_config(self.context,
self.load_balancer_id, self.config)
actual = db.load_balancer_get(self.context, self.load_balancer_id)
config_ref = models.LoadBalancerConfig()
config_ref.update(self.config)
self.compare_records(actual.config, config_ref, skiped=['id'])
def test_load_balancer_create(self):
self.truncate_all_tables()
expect = db.load_balancer_create(self.context, self.configs)
actual = db.load_balancer_get(self.context, expect.id)
self.compare_records(expect, actual, skiped=['id'])
