11import unittest
2- from unittest .mock import MagicMock , patch
2+ from unittest .mock import patch
33
44from azure .functions .extension .base import (ModuleTrackerMeta , RequestTrackerMeta ,
55 ResponseTrackerMeta , WebApp , WebServer ,
6- http_v2_enabled , ResponseLabels )
6+ HttpV2FeatureChecker , ResponseLabels )
77
88
99class TestModuleTrackerMeta (unittest .TestCase ):
1010 def setUp (self ):
1111 # Reset the _module attribute after each test
1212 ModuleTrackerMeta ._module = None
13- self .assertFalse (http_v2_enabled ())
13+ self .assertFalse (HttpV2FeatureChecker . http_v2_enabled ())
1414
1515 def test_classes_imported_from_same_module (self ):
1616 class TestClass1 (metaclass = ModuleTrackerMeta ):
@@ -20,14 +20,14 @@ class TestClass2(metaclass=ModuleTrackerMeta):
2020
2121 self .assertEqual (ModuleTrackerMeta .get_module (), __name__ )
2222 self .assertTrue (ModuleTrackerMeta .module_imported ())
23- self .assertTrue (http_v2_enabled ())
23+ self .assertTrue (HttpV2FeatureChecker . http_v2_enabled ())
2424
2525 def test_class_imported_from_a_module (self ):
2626 class TestClass1 (metaclass = ModuleTrackerMeta ):
2727 pass
2828 self .assertEqual (ModuleTrackerMeta .get_module (), __name__ )
2929 self .assertTrue (ModuleTrackerMeta .module_imported ())
30- self .assertTrue (http_v2_enabled ())
30+ self .assertTrue (HttpV2FeatureChecker . http_v2_enabled ())
3131
3232 def test_classes_imported_from_different_modules (self ):
3333 class TestClass1 (metaclass = ModuleTrackerMeta ):
@@ -210,7 +210,7 @@ class TestHttpV2Enabled(unittest.TestCase):
210210 @patch ('azure.functions.extension.base.ModuleTrackerMeta.module_imported' )
211211 def test_http_v2_enabled (self , mock_module_imported ):
212212 mock_module_imported .return_value = True
213- self .assertTrue (http_v2_enabled ())
213+ self .assertTrue (HttpV2FeatureChecker . http_v2_enabled ())
214214
215215 mock_module_imported .return_value = False
216- self .assertFalse (http_v2_enabled ())
216+ self .assertFalse (HttpV2FeatureChecker . http_v2_enabled ())
0 commit comments