def test_vmss_default_std_lb(self):
cmd = mock.MagicMock()
lb_sku_mock = mock.MagicMock()
lb_sku_mock.standard.value = 'Standard'
lb_sku_mock.basic.value = 'Basic'
cmd.get_models.return_value = lb_sku_mock
# default to standard when single-placement-group is off
ns = argparse.Namespace()
ns.load_balancer, ns.application_gateway = None, None
ns.app_gateway_subnet_address_prefix, ns.application_gateway = None, None
ns.app_gateway_sku, ns.app_gateway_capacity = None, None
ns.load_balancer_sku = None
ns.single_placement_group = False
_validate_vmss_create_load_balancer_or_app_gateway(cmd, ns)
self.assertEqual(ns.load_balancer_sku, 'Standard')
# error on conflicts
ns = argparse.Namespace()
ns.load_balancer, ns.application_gateway = None, None
ns.app_gateway_subnet_address_prefix, ns.application_gateway = None, None
ns.app_gateway_sku, ns.app_gateway_capacity = None, None
ns.load_balancer_sku = 'Basic'
ns.single_placement_group = False
ns.zones = '1'
self.assertRaises(CLIError,
_validate_vmss_create_load_balancer_or_app_gateway,
cmd, ns)
def test_vmss_default_std_lb(self):
cmd = mock.MagicMock()
lb_sku_mock = mock.MagicMock()
lb_sku_mock.standard.value = 'Standard'
lb_sku_mock.basic.value = 'Basic'
cmd.get_models.return_value = lb_sku_mock
# default to standard when single-placement-group is off
ns = argparse.Namespace()
ns.load_balancer, ns.application_gateway = None, None
ns.app_gateway_subnet_address_prefix, ns.application_gateway = None, None
ns.app_gateway_sku, ns.app_gateway_capacity = None, None
ns.load_balancer_sku = None
ns.single_placement_group = False
_validate_vmss_create_load_balancer_or_app_gateway(cmd, ns)
self.assertEqual(ns.load_balancer_sku, 'Standard')
# error on conflicts
ns = argparse.Namespace()
ns.load_balancer, ns.application_gateway = None, None
ns.app_gateway_subnet_address_prefix, ns.application_gateway = None, None
ns.app_gateway_sku, ns.app_gateway_capacity = None, None
ns.load_balancer_sku = 'Basic'
ns.single_placement_group = False
ns.zones = '1'
self.assertRaises(CLIError, _validate_vmss_create_load_balancer_or_app_gateway, cmd, ns)
