1
1
import os
2
- from urllib .request import urlopen
3
2
4
3
import pytest
5
4
from flask import url_for
@@ -18,20 +17,21 @@ def test_server_is_alive(self, live_server):
18
17
assert live_server ._process
19
18
assert live_server ._process .is_alive ()
20
19
21
- def test_server_listening (self , live_server ):
22
- res = urlopen (url_for ("ping" , _external = True ))
23
- assert res .code == 200
24
- assert b"pong" in res .read ()
20
+ def test_server_listening (self , client , live_server ):
21
+ res = client . get (url_for ("ping" , _external = True ))
22
+ assert res .status_code == 200
23
+ assert b"pong" in res .data
25
24
26
25
def test_url_for (self , live_server ):
27
26
assert (
28
27
url_for ("ping" , _external = True )
29
- == "http://localhost:%s/ping" % live_server .port
28
+ == "http://localhost.localdomain :%s/ping" % live_server .port
30
29
)
31
30
32
31
def test_set_application_server_name (self , live_server ):
33
32
assert (
34
- live_server .app .config ["SERVER_NAME" ] == "localhost:%d" % live_server .port
33
+ live_server .app .config ["SERVER_NAME" ]
34
+ == "localhost.localdomain:%d" % live_server .port
35
35
)
36
36
37
37
def test_rewrite_application_server_name (self , appdir ):
@@ -106,20 +106,19 @@ def mocked_stop_cleanly(*args, **kwargs):
106
106
appdir .create_test_module (
107
107
"""
108
108
import pytest
109
- from urllib.request import urlopen
110
109
111
110
from flask import url_for
112
111
113
- def test_a(live_server):
112
+ def test_a(live_server, client ):
114
113
@live_server.app.route('/')
115
114
def index():
116
115
return 'got it', 200
117
116
118
117
live_server.start()
119
118
120
- res = urlopen (url_for('index', _external=True))
121
- assert res.code == 200
122
- assert b'got it' in res.read()
119
+ res = client.get (url_for('index', _external=True))
120
+ assert res.status_code == 200
121
+ assert b'got it' in res.data
123
122
"""
124
123
)
125
124
args = [] if clean_stop else ["--no-live-server-clean-stop" ]
@@ -134,20 +133,19 @@ def test_add_endpoint_to_live_server(self, appdir):
134
133
appdir .create_test_module (
135
134
"""
136
135
import pytest
137
- from urllib.request import urlopen
138
136
139
137
from flask import url_for
140
138
141
- def test_a(live_server):
139
+ def test_a(live_server, client ):
142
140
@live_server.app.route('/new-endpoint')
143
141
def new_endpoint():
144
142
return 'got it', 200
145
143
146
144
live_server.start()
147
145
148
- res = urlopen (url_for('new_endpoint', _external=True))
149
- assert res.code == 200
150
- assert b'got it' in res.read()
146
+ res = client.get (url_for('new_endpoint', _external=True))
147
+ assert res.status_code == 200
148
+ assert b'got it' in res.data
151
149
"""
152
150
)
153
151
result = appdir .runpytest ("-v" , "--no-start-live-server" )
@@ -159,25 +157,24 @@ def test_concurrent_requests_to_live_server(self, appdir):
159
157
appdir .create_test_module (
160
158
"""
161
159
import pytest
162
- from urllib.request import urlopen
163
160
164
161
from flask import url_for
165
162
166
- def test_concurrent_requests(live_server):
163
+ def test_concurrent_requests(live_server, client ):
167
164
@live_server.app.route('/one')
168
165
def one():
169
- res = urlopen (url_for('two', _external=True))
170
- return res.read()
166
+ res = client.get (url_for('two', _external=True))
167
+ return res.data
171
168
172
169
@live_server.app.route('/two')
173
170
def two():
174
171
return '42'
175
172
176
173
live_server.start()
177
174
178
- res = urlopen (url_for('one', _external=True))
179
- assert res.code == 200
180
- assert b'42' in res.read()
175
+ res = client.get (url_for('one', _external=True))
176
+ assert res.status_code == 200
177
+ assert b'42' in res.data
181
178
"""
182
179
)
183
180
result = appdir .runpytest ("-v" , "--no-start-live-server" )
0 commit comments