|
56 | 56 | ) |
57 | 57 | from azul.http import ( |
58 | 58 | HasCachedHttpClient, |
59 | | - HttpClient, |
60 | 59 | ) |
61 | 60 | from azul.indexer import ( |
62 | 61 | Prefix, |
@@ -94,7 +93,7 @@ def patch_config(name: str, value: str) -> Patch: |
94 | 93 | return patch.object(Config, name, new=PropertyMock(return_value=value)) |
95 | 94 |
|
96 | 95 |
|
97 | | -class AzulTestCase(TestCase): |
| 96 | +class AzulTestCase(TestCase, HasCachedHttpClient): |
98 | 97 | _catch_warnings: Optional[AbstractContextManager] |
99 | 98 | _caught_warnings: list[warnings.WarningMessage] |
100 | 99 |
|
@@ -235,8 +234,12 @@ def stacked_patches(self, patches: Iterable[Patch]): |
235 | 234 | context.enter_context(cm) |
236 | 235 | yield |
237 | 236 |
|
| 237 | + @classmethod |
| 238 | + def http_client(cls): |
| 239 | + return cls._http_client |
| 240 | + |
238 | 241 |
|
239 | | -class AzulUnitTestCase(AzulTestCase, HasCachedHttpClient): |
| 242 | +class AzulUnitTestCase(AzulTestCase): |
240 | 243 |
|
241 | 244 | @classmethod |
242 | 245 | def setUpClass(cls) -> None: |
@@ -349,9 +352,6 @@ def _patch_valid_schema_domains(cls): |
349 | 352 | attribute='valid_schema_domains', |
350 | 353 | new=cls.valid_schema_domains)) |
351 | 354 |
|
352 | | - def http(self) -> HttpClient: |
353 | | - return self._http_client |
354 | | - |
355 | 355 |
|
356 | 356 | class CatalogTestCase(AzulUnitTestCase, metaclass=ABCMeta): |
357 | 357 | catalog: CatalogName = 'test' |
|
0 commit comments