File tree 2 files changed +19
-1
lines changed
2 files changed +19
-1
lines changed Original file line number Diff line number Diff line change @@ -96,4 +96,5 @@ def get_hooks() -> typing.List[Hook]:
96
96
97
97
98
98
def shutdown () -> None :
99
- _provider .shutdown ()
99
+ for provider in {_provider , * _providers .values ()}:
100
+ provider .shutdown ()
Original file line number Diff line number Diff line change @@ -193,3 +193,20 @@ def test_should_not_shutdown_provider_bound_to_another_domain():
193
193
set_provider (other_provider , "foo" )
194
194
195
195
provider .shutdown .assert_not_called ()
196
+
197
+
198
+ def test_shutdown_should_shutdown_every_registered_provider_once ():
199
+ # Given
200
+ provider_1 = MagicMock (spec = FeatureProvider )
201
+ provider_2 = MagicMock (spec = FeatureProvider )
202
+ set_provider (provider_1 )
203
+ set_provider (provider_1 , "foo" )
204
+ set_provider (provider_2 , "bar" )
205
+ set_provider (provider_2 , "baz" )
206
+
207
+ # When
208
+ shutdown ()
209
+
210
+ # Then
211
+ provider_1 .shutdown .assert_called_once ()
212
+ provider_2 .shutdown .assert_called_once ()
You can’t perform that action at this time.
0 commit comments