1212# See the License for the specific language governing permissions and
1313# limitations under the License.
1414
15+ import http .client
16+
1517import flask
1618from google .auth import exceptions
1719import pytest
1820from pytest_localserver .http import WSGIServer
19- from six .moves import http_client
2021
2122# .invalid will never resolve, see https://tools.ietf.org/html/rfc2606
2223NXDOMAIN = "test.invalid"
@@ -40,11 +41,11 @@ def server(self):
4041 def index ():
4142 header_value = flask .request .headers .get ("x-test-header" , "value" )
4243 headers = {"X-Test-Header" : header_value }
43- return "Basic Content" , http_client .OK , headers
44+ return "Basic Content" , http . client .OK , headers
4445
4546 @app .route ("/server_error" )
4647 def server_error ():
47- return "Error" , http_client .INTERNAL_SERVER_ERROR
48+ return "Error" , http . client .INTERNAL_SERVER_ERROR
4849
4950 # pylint: enable=unused-variable
5051
@@ -57,15 +58,15 @@ def test_request_basic(self, server):
5758 request = self .make_request ()
5859 response = request (url = server .url + "/basic" , method = "GET" )
5960
60- assert response .status == http_client .OK
61+ assert response .status == http . client .OK
6162 assert response .headers ["x-test-header" ] == "value"
6263 assert response .data == b"Basic Content"
6364
6465 def test_request_timeout (self , server ):
6566 request = self .make_request ()
6667 response = request (url = server .url + "/basic" , method = "GET" , timeout = 2 )
6768
68- assert response .status == http_client .OK
69+ assert response .status == http . client .OK
6970 assert response .headers ["x-test-header" ] == "value"
7071 assert response .data == b"Basic Content"
7172
@@ -77,15 +78,15 @@ def test_request_headers(self, server):
7778 headers = {"x-test-header" : "hello world" },
7879 )
7980
80- assert response .status == http_client .OK
81+ assert response .status == http . client .OK
8182 assert response .headers ["x-test-header" ] == "hello world"
8283 assert response .data == b"Basic Content"
8384
8485 def test_request_error (self , server ):
8586 request = self .make_request ()
8687 response = request (url = server .url + "/server_error" , method = "GET" )
8788
88- assert response .status == http_client .INTERNAL_SERVER_ERROR
89+ assert response .status == http . client .INTERNAL_SERVER_ERROR
8990 assert response .data == b"Error"
9091
9192 def test_connection_error (self ):
0 commit comments