@@ -97,30 +97,26 @@ def test_parent():
97
97
98
98
def test_keyring_with_generator_but_no_children ():
99
99
generator_keyring = RawAESKeyring (
100
- key_namespace = _PROVIDER_ID ,
101
- key_name = _KEY_ID ,
102
- wrapping_key = _WRAPPING_KEY_AES ,
103
- wrapping_algorithm = WrappingAlgorithm .AES_256_GCM_IV12_TAG16_NO_PADDING ,
104
- )
105
- test_multi_keyring = MultiKeyring (
106
- generator = generator_keyring
100
+ key_namespace = _PROVIDER_ID ,
101
+ key_name = _KEY_ID ,
102
+ wrapping_key = _WRAPPING_KEY_AES ,
103
+ wrapping_algorithm = WrappingAlgorithm .AES_256_GCM_IV12_TAG16_NO_PADDING ,
107
104
)
105
+ test_multi_keyring = MultiKeyring (generator = generator_keyring )
108
106
assert test_multi_keyring .generator is generator_keyring
109
107
assert not test_multi_keyring .children
110
108
111
109
112
110
def test_keyring_with_children_but_no_generator ():
113
111
children_keyring = [
114
- RawAESKeyring (
115
- key_namespace = _PROVIDER_ID ,
116
- key_name = _KEY_ID ,
117
- wrapping_key = _WRAPPING_KEY_AES ,
118
- wrapping_algorithm = WrappingAlgorithm .AES_256_GCM_IV12_TAG16_NO_PADDING ,
119
- )
120
- ]
121
- test_multi_keyring = MultiKeyring (
122
- children = children_keyring
123
- )
112
+ RawAESKeyring (
113
+ key_namespace = _PROVIDER_ID ,
114
+ key_name = _KEY_ID ,
115
+ wrapping_key = _WRAPPING_KEY_AES ,
116
+ wrapping_algorithm = WrappingAlgorithm .AES_256_GCM_IV12_TAG16_NO_PADDING ,
117
+ )
118
+ ]
119
+ test_multi_keyring = MultiKeyring (children = children_keyring )
124
120
assert test_multi_keyring .children is children_keyring
125
121
assert test_multi_keyring .generator is None
126
122
@@ -151,7 +147,7 @@ def test_decryption_keyrings():
151
147
assert test_multi_keyring .generator in test_multi_keyring ._decryption_keyrings
152
148
for child_keyring in test_multi_keyring .children :
153
149
assert child_keyring in test_multi_keyring ._decryption_keyrings
154
- # assert len(list( test_multi_keyring._decryption_keyrings) ) == len(test_multi_keyring.children) + 1
150
+ assert len (test_multi_keyring ._decryption_keyrings ) == len (test_multi_keyring .children ) + 1
155
151
156
152
157
153
def test_on_encrypt_with_no_generator_no_data_encryption_key ():
0 commit comments