1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
|
import requests
from pytest_httpserver import HTTPServer
def test_oneshot(httpserver: HTTPServer):
httpserver.expect_oneshot_request("/foobar").respond_with_data("OK foobar")
httpserver.expect_oneshot_request("/foobaz").respond_with_data("OK foobaz")
assert len(httpserver.oneshot_handlers) == 2
# first requests should pass
response = requests.get(httpserver.url_for("/foobaz"))
httpserver.check_assertions()
assert response.status_code == 200
assert response.text == "OK foobaz"
response = requests.get(httpserver.url_for("/foobar"))
httpserver.check_assertions()
assert response.status_code == 200
assert response.text == "OK foobar"
assert len(httpserver.oneshot_handlers) == 0
# second requests should fail due to 'oneshot' type
assert requests.get(httpserver.url_for("/foobar")).status_code == 500
assert requests.get(httpserver.url_for("/foobaz")).status_code == 500
def test_oneshot_any_method(httpserver: HTTPServer):
for _ in range(5):
httpserver.expect_oneshot_request("/foobar").respond_with_data("OK")
response = requests.post(httpserver.url_for("/foobar"))
assert response.text == "OK"
assert response.status_code == 200
response = requests.get(httpserver.url_for("/foobar"))
assert response.text == "OK"
assert response.status_code == 200
response = requests.delete(httpserver.url_for("/foobar"))
assert response.text == "OK"
assert response.status_code == 200
response = requests.put(httpserver.url_for("/foobar"))
assert response.text == "OK"
assert response.status_code == 200
response = requests.patch(httpserver.url_for("/foobar"))
assert response.text == "OK"
assert response.status_code == 200
assert len(httpserver.oneshot_handlers) == 0
|
