|
8 | 8 |
|
9 | 9 | from newrelic.common import certs |
10 | 10 | from newrelic.common.agent_http import ( |
| 11 | + ApplicationModeClient, |
11 | 12 | DeveloperModeClient, |
12 | 13 | HttpClient, |
13 | 14 | InsecureHttpClient, |
@@ -190,7 +191,7 @@ def test_http_no_payload(server, method): |
190 | 191 | def test_non_ok_response(server): |
191 | 192 | internal_metrics = CustomMetrics() |
192 | 193 |
|
193 | | - with HttpClient( |
| 194 | + with ApplicationModeClient( |
194 | 195 | "localhost", server.port, disable_certificate_validation=True |
195 | 196 | ) as client: |
196 | 197 | with InternalTraceContext(internal_metrics): |
@@ -232,7 +233,7 @@ def test_http_payload_compression(server, method, threshold): |
232 | 233 |
|
233 | 234 | internal_metrics = CustomMetrics() |
234 | 235 |
|
235 | | - with HttpClient( |
| 236 | + with ApplicationModeClient( |
236 | 237 | "localhost", |
237 | 238 | server.port, |
238 | 239 | disable_certificate_validation=True, |
@@ -543,8 +544,8 @@ def test_serverless_mode_client(): |
543 | 544 | ( |
544 | 545 | (HttpClient, "localhost", False), |
545 | 546 | (HttpClient, None, False), |
546 | | - (HttpClient, None, True), |
547 | | - (HttpClient, "localhost", True), |
| 547 | + (ApplicationModeClient, None, True), |
| 548 | + (ApplicationModeClient, "localhost", True), |
548 | 549 | (DeveloperModeClient, None, False), |
549 | 550 | (ServerlessModeClient, None, False), |
550 | 551 | (InsecureHttpClient, None, False), |
|
0 commit comments