register command group for security-policy waf

src/alb/HISTORY.rst

@@ -3,6 +3,10 @@
 Release History
 ===============
 
+2.0.1
+++++++
+* Register command group for security-policy waf and add description.
+
 2.0.0
 ++++++
 * Upgrade API version to 2025-01-01, introducing Security Policy WAF.

src/alb/azext_alb/_help.py

@@ -9,3 +9,8 @@
 # pylint: disable=too-many-lines
 
 from knack.help_files import helps  # pylint: disable=unused-import
+
+helps['network alb security-policy waf'] = """
+type: group
+short-summary: Manage and Configure WAF Security Policies for Application Gateway for Containers resource
+"""

src/alb/azext_alb/commands.py

@@ -15,3 +15,6 @@ def load_command_table(self, _):  # pylint: disable=unused-argument
     self.command_table["network alb security-policy waf create"] = WafSecurityPolicyCreate(loader=self)
     self.command_table["network alb security-policy waf delete"] = WafSecurityPolicyDelete(loader=self)
     self.command_table["network alb security-policy waf update"] = WafSecurityPolicyUpdate(loader=self)
+
+    with self.command_group("network alb security-policy waf") as _:
+        pass

src/alb/azext_alb/tests/latest/test_alb_commands.py

@@ -21,9 +21,10 @@
 TEST_DIR = os.path.abspath(os.path.join(os.path.abspath(__file__), '..'))
 
 class AlbScenario(ScenarioTest):
+    @live_only()
+    @ResourceGroupPreparer(name_prefix='cli_record_test_tc_000999990', location="eastus2euap")
     def test_alb_resources(self):
         self.kwargs.update({
-            'rg': 'cli_record_test_tc_000999990',
             'tc': 'tc1',
             'fe': 'fe1',
             'aso': 'a1',
@@ -32,10 +33,6 @@ def test_alb_resources(self):
             'waf_policy': 'waf_policy1',
             'sp_waf': 'sp_waf1',
         })
-
-        #Works with subscription 668f1741-ef48-4132-bf98-28b3a3d3f257
-        # Create Resource group.
-        self.cmd('group create -n {rg} --location eastus2euap')
 
         # Create Traffic Controller
         tcCount1 = len(self.cmd('network alb list -g {rg}').get_output_in_json())

src/alb/setup.py

@@ -10,7 +10,7 @@
 
 
 # HISTORY.rst entry.
-VERSION = '2.0.0'
+VERSION = '2.0.1'
 
 # The full list of classifiers is available at
 # https://pypi.python.org/pypi?%3Aaction=list_classifiers