def test_update_addons(self, rg_def, cf_resource_groups, cf_resources):
# http_application_routing enabled
instance = mock.MagicMock()
instance.addon_profiles = None
cmd = mock.MagicMock()
instance = _update_addons(cmd, instance, '00000000-0000-0000-0000-000000000000',
'clitest000001', 'http_application_routing', enable=True)
self.assertIn('httpApplicationRouting', instance.addon_profiles)
addon_profile = instance.addon_profiles['httpApplicationRouting']
self.assertTrue(addon_profile.enabled)
# http_application_routing disabled
instance = _update_addons(cmd, instance, '00000000-0000-0000-0000-000000000000',
'clitest000001', 'http_application_routing', enable=False)
addon_profile = instance.addon_profiles['httpApplicationRouting']
self.assertFalse(addon_profile.enabled)
# monitoring added
instance = _update_addons(cmd, instance, '00000000-0000-0000-0000-000000000000',
'clitest000001', 'monitoring', enable=True)
monitoring_addon_profile = instance.addon_profiles['omsagent']
self.assertTrue(monitoring_addon_profile.enabled)
routing_addon_profile = instance.addon_profiles['httpApplicationRouting']
self.assertFalse(routing_addon_profile.enabled)
# monitoring disabled, routing enabled
instance = _update_addons(cmd, instance, '00000000-0000-0000-0000-000000000000',
'clitest000001', 'monitoring', enable=False)
instance = _update_addons(cmd, instance, '00000000-0000-0000-0000-000000000000', 'clitest000001',
'http_application_routing', enable=True)
monitoring_addon_profile = instance.addon_profiles['omsagent']
self.assertFalse(monitoring_addon_profile.enabled)
routing_addon_profile = instance.addon_profiles['httpApplicationRouting']
self.assertTrue(routing_addon_profile.enabled)
self.assertEqual(sorted(list(instance.addon_profiles)), ['httpApplicationRouting', 'omsagent'])
# azurepolicy added
instance = _update_addons(cmd, instance, '00000000-0000-0000-0000-000000000000',
'clitest000001', 'azure-policy', enable=True)
azurepolicy_addon_profile = instance.addon_profiles['azurepolicy']
self.assertTrue(azurepolicy_addon_profile.enabled)
routing_addon_profile = instance.addon_profiles['httpApplicationRouting']
self.assertTrue(routing_addon_profile.enabled)
monitoring_addon_profile = instance.addon_profiles['omsagent']
self.assertFalse(monitoring_addon_profile.enabled)
# azurepolicy disabled, routing enabled
instance = _update_addons(cmd, instance, '00000000-0000-0000-0000-000000000000',
'clitest000001', 'azure-policy', enable=False)
instance = _update_addons(cmd, instance, '00000000-0000-0000-0000-000000000000', 'clitest000001',
'http_application_routing', enable=True)
azurepolicy_addon_profile = instance.addon_profiles['azurepolicy']
self.assertFalse(azurepolicy_addon_profile.enabled)
monitoring_addon_profile = instance.addon_profiles['omsagent']
self.assertFalse(monitoring_addon_profile.enabled)
routing_addon_profile = instance.addon_profiles['httpApplicationRouting']
self.assertTrue(routing_addon_profile.enabled)
self.assertEqual(sorted(list(instance.addon_profiles)), ['azurepolicy', 'httpApplicationRouting', 'omsagent'])
# kube-dashboard disabled, no existing dashboard addon profile
instance = _update_addons(cmd, instance, '00000000-0000-0000-0000-000000000000',
'clitest000001', 'kube-dashboard', enable=False)
dashboard_addon_profile = instance.addon_profiles['kubeDashboard']
self.assertFalse(dashboard_addon_profile.enabled)
# kube-dashboard enabled, no existing dashboard addon profile
instance.addon_profiles.pop('kubeDashboard', None)
instance = _update_addons(cmd, instance, '00000000-0000-0000-0000-000000000000',
'clitest000001', 'kube-dashboard', enable=True)
dashboard_addon_profile = instance.addon_profiles['kubeDashboard']
self.assertTrue(dashboard_addon_profile.enabled)
# kube-dashboard disabled, there's existing dashboard addon profile
instance.addon_profiles.pop('kubeDashboard', None)
# test lower cased key name
instance.addon_profiles['kubedashboard'] = ManagedClusterAddonProfile(enabled=True)
instance = _update_addons(cmd, instance, '00000000-0000-0000-0000-000000000000',
'clitest000001', 'kube-dashboard', enable=False)
dashboard_addon_profile = instance.addon_profiles['kubedashboard']
self.assertFalse(dashboard_addon_profile.enabled)
# kube-dashboard enabled, there's existing dashboard addon profile
instance.addon_profiles.pop('kubedashboard', None)
# test lower cased key name
instance.addon_profiles['kubedashboard'] = ManagedClusterAddonProfile(enabled=False)
instance = _update_addons(cmd, instance, '00000000-0000-0000-0000-000000000000',
'clitest000001', 'kube-dashboard', enable=True)
dashboard_addon_profile = instance.addon_profiles['kubedashboard']
self.assertTrue(dashboard_addon_profile.enabled)
# monitoring enabled and then enabled again should error
instance = mock.Mock()
instance.addon_profiles = None
instance = _update_addons(cmd, instance, '00000000-0000-0000-0000-000000000000',
'clitest000001', 'monitoring', enable=True)
with self.assertRaises(CLIError):
instance = _update_addons(cmd, instance, '00000000-0000-0000-0000-000000000000',
'clitest000001', 'monitoring', enable=True)
def test_update_addons(self, rg_def, cf_resource_groups, cf_resources):
# http_application_routing enabled
instance = mock.MagicMock()
instance.addon_profiles = None
cmd = mock.MagicMock()
instance = _update_addons(cmd, instance, '00000000-0000-0000-0000-000000000000',
'clitest000001', 'clitest000001', 'http_application_routing', enable=True)
self.assertIn(CONST_HTTP_APPLICATION_ROUTING_ADDON_NAME, instance.addon_profiles)
addon_profile = instance.addon_profiles[CONST_HTTP_APPLICATION_ROUTING_ADDON_NAME]
self.assertTrue(addon_profile.enabled)
# http_application_routing disabled
instance = _update_addons(cmd, instance, '00000000-0000-0000-0000-000000000000',
'clitest000001', 'clitest000001', 'http_application_routing', enable=False)
addon_profile = instance.addon_profiles[CONST_HTTP_APPLICATION_ROUTING_ADDON_NAME]
self.assertFalse(addon_profile.enabled)
# monitoring added
instance = _update_addons(cmd, instance, '00000000-0000-0000-0000-000000000000',
'clitest000001', 'clitest000001', 'monitoring', enable=True)
monitoring_addon_profile = instance.addon_profiles[CONST_MONITORING_ADDON_NAME]
self.assertTrue(monitoring_addon_profile.enabled)
routing_addon_profile = instance.addon_profiles[CONST_HTTP_APPLICATION_ROUTING_ADDON_NAME]
self.assertFalse(routing_addon_profile.enabled)
# monitoring disabled, routing enabled
instance = _update_addons(cmd, instance, '00000000-0000-0000-0000-000000000000',
'clitest000001', 'clitest000001', 'monitoring', enable=False)
instance = _update_addons(cmd, instance, '00000000-0000-0000-0000-000000000000',
'clitest000001', 'clitest000001', 'http_application_routing', enable=True)
monitoring_addon_profile = instance.addon_profiles[CONST_MONITORING_ADDON_NAME]
self.assertFalse(monitoring_addon_profile.enabled)
routing_addon_profile = instance.addon_profiles[CONST_HTTP_APPLICATION_ROUTING_ADDON_NAME]
self.assertTrue(routing_addon_profile.enabled)
self.assertEqual(sorted(list(instance.addon_profiles)), [CONST_HTTP_APPLICATION_ROUTING_ADDON_NAME, CONST_MONITORING_ADDON_NAME])
# azurepolicy added
instance = _update_addons(cmd, instance, '00000000-0000-0000-0000-000000000000',
'clitest000001', 'clitest000001', 'azure-policy', enable=True)
azurepolicy_addon_profile = instance.addon_profiles[CONST_AZURE_POLICY_ADDON_NAME]
self.assertTrue(azurepolicy_addon_profile.enabled)
routing_addon_profile = instance.addon_profiles[CONST_HTTP_APPLICATION_ROUTING_ADDON_NAME]
self.assertTrue(routing_addon_profile.enabled)
monitoring_addon_profile = instance.addon_profiles[CONST_MONITORING_ADDON_NAME]
self.assertFalse(monitoring_addon_profile.enabled)
# azurepolicy disabled, routing enabled
instance = _update_addons(cmd, instance, '00000000-0000-0000-0000-000000000000',
'clitest000001', 'clitest000001', 'azure-policy', enable=False)
instance = _update_addons(cmd, instance, '00000000-0000-0000-0000-000000000000',
'clitest000001', 'clitest000001', 'http_application_routing', enable=True)
azurepolicy_addon_profile = instance.addon_profiles[CONST_AZURE_POLICY_ADDON_NAME]
self.assertFalse(azurepolicy_addon_profile.enabled)
monitoring_addon_profile = instance.addon_profiles[CONST_MONITORING_ADDON_NAME]
self.assertFalse(monitoring_addon_profile.enabled)
routing_addon_profile = instance.addon_profiles[CONST_HTTP_APPLICATION_ROUTING_ADDON_NAME]
self.assertTrue(routing_addon_profile.enabled)
self.assertEqual(sorted(list(instance.addon_profiles)), [CONST_AZURE_POLICY_ADDON_NAME, CONST_HTTP_APPLICATION_ROUTING_ADDON_NAME, CONST_MONITORING_ADDON_NAME])
# kube-dashboard disabled, no existing dashboard addon profile
instance = _update_addons(cmd, instance, '00000000-0000-0000-0000-000000000000',
'clitest000001', 'clitest000001', 'kube-dashboard', enable=False)
dashboard_addon_profile = instance.addon_profiles[CONST_KUBE_DASHBOARD_ADDON_NAME]
self.assertFalse(dashboard_addon_profile.enabled)
# kube-dashboard enabled, no existing dashboard addon profile
instance.addon_profiles.pop(CONST_KUBE_DASHBOARD_ADDON_NAME, None)
instance = _update_addons(cmd, instance, '00000000-0000-0000-0000-000000000000',
'clitest000001', 'clitest000001', 'kube-dashboard', enable=True)
dashboard_addon_profile = instance.addon_profiles[CONST_KUBE_DASHBOARD_ADDON_NAME]
self.assertTrue(dashboard_addon_profile.enabled)
# kube-dashboard disabled, there's existing dashboard addon profile
instance.addon_profiles.pop(CONST_KUBE_DASHBOARD_ADDON_NAME, None)
# test lower cased key name
instance.addon_profiles['kubedashboard'] = ManagedClusterAddonProfile(enabled=True)
instance = _update_addons(cmd, instance, '00000000-0000-0000-0000-000000000000',
'clitest000001', 'clitest000001', 'kube-dashboard', enable=False)
dashboard_addon_profile = instance.addon_profiles[CONST_KUBE_DASHBOARD_ADDON_NAME]
self.assertFalse(dashboard_addon_profile.enabled)
# kube-dashboard enabled, there's existing dashboard addon profile
instance.addon_profiles.pop(CONST_KUBE_DASHBOARD_ADDON_NAME, None)
# test lower cased key name
instance.addon_profiles['kubedashboard'] = ManagedClusterAddonProfile(enabled=False)
instance = _update_addons(cmd, instance, '00000000-0000-0000-0000-000000000000',
'clitest000001', 'clitest000001', 'kube-dashboard', enable=True)
dashboard_addon_profile = instance.addon_profiles[CONST_KUBE_DASHBOARD_ADDON_NAME]
self.assertTrue(dashboard_addon_profile.enabled)
# monitoring enabled and then enabled again should error
instance = mock.Mock()
instance.addon_profiles = None
instance = _update_addons(cmd, instance, '00000000-0000-0000-0000-000000000000',
'clitest000001', 'clitest000001', 'monitoring', enable=True)
with self.assertRaises(CLIError):
instance = _update_addons(cmd, instance, '00000000-0000-0000-0000-000000000000',
'clitest000001', 'clitest000001', 'monitoring', enable=True)
# virtual-node enabled
instance = mock.MagicMock()
instance.addon_profiles = None
cmd = mock.MagicMock()
instance = _update_addons(cmd, instance, '00000000-0000-0000-0000-000000000000',
'clitest000001', 'clitest000001', 'virtual-node', enable=True, subnet_name='foo')
self.assertIn('aciConnectorLinux', instance.addon_profiles)
addon_profile = instance.addon_profiles['aciConnectorLinux']
self.assertTrue(addon_profile.enabled)
# virtual-node disabled
instance = _update_addons(cmd, instance, '00000000-0000-0000-0000-000000000000',
'clitest000001', 'clitest000001', 'virtual-node', enable=False)
addon_profile = instance.addon_profiles['aciConnectorLinux']
self.assertFalse(addon_profile.enabled)
# ingress-appgw enabled
instance = mock.MagicMock()
instance.addon_profiles = None
cmd = mock.MagicMock()
instance = _update_addons(cmd, instance, '00000000-0000-0000-0000-000000000000',
'clitest000001', 'clitest000001', 'ingress-appgw', enable=True, appgw_subnet_cidr='10.2.0.0/16')
self.assertIn('ingressApplicationGateway', instance.addon_profiles)
addon_profile = instance.addon_profiles['ingressApplicationGateway']
self.assertTrue(addon_profile.enabled)
# ingress-appgw disabled
instance = _update_addons(cmd, instance, '00000000-0000-0000-0000-000000000000',
'clitest000001', 'clitest000001', 'ingress-appgw', enable=False)
addon_profile = instance.addon_profiles['ingressApplicationGateway']
self.assertFalse(addon_profile.enabled)