コード例 #1
0
ファイル: test_custom.py プロジェクト: serbrech/azure-cli
    def test_update_addons(self):
        # http_application_routing enabled
        instance = mock.Mock()
        instance.addon_profiles = None
        instance = _update_addons(instance, '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 enabled
        instance = _update_addons(instance, 'http_application_routing', enable=False)
        addon_profile = instance.addon_profiles['httpApplicationRouting']
        self.assertFalse(addon_profile.enabled)

        # monitoring added
        instance = _update_addons(instance, '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(instance, 'monitoring', enable=False)
        instance = _update_addons(instance, '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'])
コード例 #2
0
ファイル: test_custom.py プロジェクト: sptramer/azure-cli
    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 enabled
        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'])

        # 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)
コード例 #3
0
ファイル: test_custom.py プロジェクト: woakesd/azure-cli
    def test_update_addons(self, rg_def, cf_resource_groups, cf_resources):
        # http_application_routing enabled
        instance = mock.MagicMock()
        instance.addon_profiles = None

        instance = _update_addons(MockCmd(self.cli),
                                  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(MockCmd(self.cli),
                                  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(MockCmd(self.cli),
                                  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(MockCmd(self.cli),
                                  instance,
                                  '00000000-0000-0000-0000-000000000000',
                                  'clitest000001',
                                  'clitest000001',
                                  'monitoring',
                                  enable=False)
        instance = _update_addons(MockCmd(self.cli),
                                  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(MockCmd(self.cli),
                                  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(MockCmd(self.cli),
                                  instance,
                                  '00000000-0000-0000-0000-000000000000',
                                  'clitest000001',
                                  'clitest000001',
                                  'azure-policy',
                                  enable=False)
        instance = _update_addons(MockCmd(self.cli),
                                  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(MockCmd(self.cli),
                                  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(MockCmd(self.cli),
                                  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(MockCmd(self.cli),
                                  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(MockCmd(self.cli),
                                  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(MockCmd(self.cli),
                                  instance,
                                  '00000000-0000-0000-0000-000000000000',
                                  'clitest000001',
                                  'clitest000001',
                                  'monitoring',
                                  enable=True)
        with self.assertRaises(CLIError):
            instance = _update_addons(MockCmd(self.cli),
                                      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(MockCmd(self.cli),
                                  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(MockCmd(self.cli),
                                  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(MockCmd(self.cli),
                                  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(MockCmd(self.cli),
                                  instance,
                                  '00000000-0000-0000-0000-000000000000',
                                  'clitest000001',
                                  'clitest000001',
                                  'ingress-appgw',
                                  enable=False)
        addon_profile = instance.addon_profiles['ingressApplicationGateway']
        self.assertFalse(addon_profile.enabled)
コード例 #4
0
ファイル: test_custom.py プロジェクト: limingu/azure-cli
    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 enabled
        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'])

        # 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)
コード例 #5
0
ファイル: test_custom.py プロジェクト: wiktorn/azure-cli
    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)