22# Copyright (c) Microsoft Corporation.
33# Licensed under the MIT License.
44# -------------------------------------
5- import functools
65from azure .keyvault .secrets import SecretClient , parse_key_vault_secret_id
76from devtools_testutils import ResourceGroupPreparer , KeyVaultPreparer
87
9- from _shared .preparer import KeyVaultClientPreparer as _KeyVaultClientPreparer
8+ from _shared .preparer import KeyVaultClientPreparer
109from _shared .test_case import KeyVaultTestCase
1110
12- # pre-apply the client_cls positional argument so it needn't be explicitly passed below
13- KeyVaultClientPreparer = functools .partial (_KeyVaultClientPreparer , SecretClient )
14-
1511
1612class TestParseId (KeyVaultTestCase ):
1713 @ResourceGroupPreparer (random_name_enabled = True )
1814 @KeyVaultPreparer ()
19- @KeyVaultClientPreparer ()
15+ @KeyVaultClientPreparer (SecretClient )
2016 def test_parse_secret_id_with_version (self , client ):
2117 secret_name = self .get_resource_name ("secret" )
2218 secret_value = "secret_value"
@@ -32,31 +28,27 @@ def test_parse_secret_id_with_version(self, client):
3228 print (parsed_secret_id .version )
3329 print (parsed_secret_id .source_id )
3430 # [END parse_key_vault_secret_id]
35- self .assertEqual (parsed_secret_id .name , secret_name )
36- self .assertEqual (parsed_secret_id .vault_url , client .vault_url )
37- self .assertEqual (parsed_secret_id .version , secret .properties .version )
38- self .assertEqual (parsed_secret_id .source_id , secret .id )
39-
40- def test_parse_secret_id_with_pending_version (self ):
41- source_id = "https://keyvault-name.vault.azure.net/secrets/secret-name/pending"
42- parsed_secret_id = parse_key_vault_secret_id (source_id )
43-
44- self .assertEqual (parsed_secret_id .name , "secret-name" )
45- self .assertEqual (parsed_secret_id .vault_url , "https://keyvault-name.vault.azure.net" )
46- self .assertEqual (parsed_secret_id .version , "pending" )
47- self .assertEqual (
48- parsed_secret_id .source_id ,
49- "https://keyvault-name.vault.azure.net/secrets/secret-name/pending" ,
50- )
51-
52- def test_parse_deleted_secret_id (self ):
53- source_id = "https://keyvault-name.vault.azure.net/deletedsecrets/deleted-secret"
54- parsed_secret_id = parse_key_vault_secret_id (source_id )
55-
56- self .assertEqual (parsed_secret_id .name , "deleted-secret" )
57- self .assertEqual (parsed_secret_id .vault_url , "https://keyvault-name.vault.azure.net" )
58- self .assertIsNone (parsed_secret_id .version )
59- self .assertEqual (
60- parsed_secret_id .source_id ,
61- "https://keyvault-name.vault.azure.net/deletedsecrets/deleted-secret" ,
62- )
31+ assert parsed_secret_id .name == secret_name
32+ assert parsed_secret_id .vault_url == client .vault_url
33+ assert parsed_secret_id .version == secret .properties .version
34+ assert parsed_secret_id .source_id == secret .id
35+
36+
37+ def test_parse_secret_id_with_pending_version ():
38+ source_id = "https://keyvault-name.vault.azure.net/secrets/secret-name/pending"
39+ parsed_secret_id = parse_key_vault_secret_id (source_id )
40+
41+ assert parsed_secret_id .name == "secret-name"
42+ assert parsed_secret_id .vault_url == "https://keyvault-name.vault.azure.net"
43+ assert parsed_secret_id .version == "pending"
44+ assert parsed_secret_id .source_id == "https://keyvault-name.vault.azure.net/secrets/secret-name/pending"
45+
46+
47+ def test_parse_deleted_secret_id ():
48+ source_id = "https://keyvault-name.vault.azure.net/deletedsecrets/deleted-secret"
49+ parsed_secret_id = parse_key_vault_secret_id (source_id )
50+
51+ assert parsed_secret_id .name == "deleted-secret"
52+ assert parsed_secret_id .vault_url == "https://keyvault-name.vault.azure.net"
53+ assert parsed_secret_id .version is None
54+ assert parsed_secret_id .source_id == "https://keyvault-name.vault.azure.net/deletedsecrets/deleted-secret"
0 commit comments