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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
import osmapi
import pytest
from unittest import mock
import responses
import os
import re
__location__ = os.path.realpath(os.path.join(os.getcwd(), os.path.dirname(__file__)))
@pytest.fixture
def file_content():
def _file_content(filename):
path = os.path.join(__location__, "fixtures", filename)
if not os.path.exists(path):
return ""
with open(path) as f:
return f.read()
return _file_content
@pytest.fixture
def api():
api_base = "http://api06.dev.openstreetmap.org"
api = osmapi.OsmApi(api=api_base)
api._session._sleep = mock.Mock()
yield api
api.close()
@pytest.fixture
def auth_api():
api_base = "http://api06.dev.openstreetmap.org"
api = osmapi.OsmApi(api=api_base, username="testuser", password="testpassword")
api._session._sleep = mock.Mock()
yield api
api.close()
@pytest.fixture
def prod_api():
api_base = "https://www.openstreetmap.org"
api = osmapi.OsmApi(api=api_base, username="testuser", password="testpassword")
api._session._sleep = mock.Mock()
yield api
api.close()
@pytest.fixture
def mocked_responses():
with responses.RequestsMock() as rsps:
yield rsps
@pytest.fixture
def add_response(mocked_responses, file_content, request):
def _add_response(method, path=None, filename=None, body=None, status=200):
if not filename:
# use testname by default
filename = f"{request.node.originalname}.xml"
if path:
url = f"http://api06.dev.openstreetmap.org/api/0.6{path}"
else:
url = re.compile(r"http:\/\/api06\.dev\.openstreetmap\.org.*")
if not body:
body = file_content(filename)
mocked_responses.add(method, url=url, body=body, status=status)
return mocked_responses
return _add_response
|
