fix issue #11658 for is_valid_resource_id (#11709)

* fix issue #11658 * use regex * add tests * revert print
Azure · Sep 1, 2020 · f683b29 · f683b29
1 parent 43d7ebf
commit f683b29
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/sdk/core/azure-mgmt-core/azure/mgmt/core/tools.py b/sdk/core/azure-mgmt-core/azure/mgmt/core/tools.py
@@ -30,13 +30,13 @@
 
 _LOGGER = logging.getLogger(__name__)
 _ARMID_RE = re.compile(
-    "(?i)/subscriptions/(?P<subscription>[^/]*)(/resourceGroups/(?P<resource_group>[^/]*))?"
-    "(/providers/(?P<namespace>[^/]*)/(?P<type>[^/]*)/(?P<name>[^/]*)(?P<children>.*))?"
+    "(?i)/subscriptions/(?P<subscription>[^/]+)(/resourceGroups/(?P<resource_group>[^/]+))?"
+    "(/providers/(?P<namespace>[^/]+)/(?P<type>[^/]*)/(?P<name>[^/]+)(?P<children>.*))?"
 )
 
 _CHILDREN_RE = re.compile(
-    "(?i)(/providers/(?P<child_namespace>[^/]*))?/"
-    "(?P<child_type>[^/]*)/(?P<child_name>[^/]*)"
+    "(?i)(/providers/(?P<child_namespace>[^/]+))?/"
+    "(?P<child_type>[^/]*)/(?P<child_name>[^/]+)"
 )
 
 _ARMNAME_RE = re.compile("^[^<>%&:\\?/]{1,260}$")

diff --git a/sdk/core/azure-mgmt-core/tests/test_tools.py b/sdk/core/azure-mgmt-core/tests/test_tools.py
@@ -239,7 +239,9 @@ def test_resource_parse(self):
         invalid_ids = [
             '/subscriptions/fakesub/resourceGroups/myRg/type1/name1',
             '/subscriptions/fakesub/resourceGroups/myRg/providers/Microsoft.Provider/foo',
-            '/subscriptions/fakesub/resourceGroups/myRg/providers/namespace/type/name/type1'
+            '/subscriptions/fakesub/resourceGroups/myRg/providers/namespace/type/name/type1',
+            '/subscriptions/fakesub/resourceGroups/',
+            '/subscriptions//resourceGroups/'
         ]
         for invalid_id in invalid_ids:
             self.assertFalse(is_valid_resource_id(invalid_id))