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
|
from __future__ import unicode_literals
from osmapi import OsmApi
import mock
import os
import unittest
__location__ = os.path.realpath(
os.path.join(
os.getcwd(),
os.path.dirname(__file__)
)
)
class TestOsmApi(unittest.TestCase):
def setUp(self):
self.api_base = "http://api06.dev.openstreetmap.org"
self.api = OsmApi(
api=self.api_base
)
self.maxDiff = None
print(self._testMethodName)
print(self.api)
def _session_mock(self, auth=False, filenames=None, status=200):
response_mock = mock.Mock()
response_mock.status_code = status
return_values = self._return_values(filenames)
print(filenames)
print(return_values)
assert len(return_values) < 2
if return_values:
response_mock.content = return_values[0]
self.session_mock = mock.Mock()
self.session_mock.request = mock.Mock(return_value=response_mock)
if auth:
self.api = OsmApi(
api=self.api_base,
username='testuser',
password='testpassword',
session=self.session_mock
)
else:
self.api = OsmApi(
api=self.api_base,
session=self.session_mock
)
self.api._get_http_session = mock.Mock(return_value=self.session_mock)
self.api._session._sleep = mock.Mock()
def _return_values(self, filenames):
if filenames is None:
filenames = [self._testMethodName + ".xml"]
return_values = []
for filename in filenames:
path = os.path.join(
__location__,
'fixtures',
filename
)
try:
with open(path) as file:
return_values.append(file.read())
except Exception:
pass
return return_values
def teardown(self):
pass
def test_constructor(self):
self.assertTrue(isinstance(self.api, OsmApi))
|