def test_cookies_with_jar(self):
jar = requests_mock.CookieJar()
jar.set('fig', 'newton', path='/foo', domain='.example.com')
jar.set('sugar', 'apple', path='/bar', domain='.example.com')
self.adapter.register_uri('GET', self.url, text='text', cookies=jar)
resp = self.session.get(self.url)
self.assertEqual('newton', resp.cookies['fig'])
self.assertEqual('apple', resp.cookies['sugar'])
self.assertEqual(set(['/foo', '/bar']), set(resp.cookies.list_paths()))
self.assertEqual(['.example.com'], resp.cookies.list_domains())
def test_get_event_data(self, mock_request):
jar = requests_mock.CookieJar()
jar.set('crumb', 'bar', domain='cityblossoms.org')
mock_request.register_uri('GET',
url='http://cityblossoms.org/calendar',
status_code=200,
cookies=jar)
month = datetime.now().strftime("%m-%Y")
url = f'http://cityblossoms.org/api/open/GetItemsByMonth?month={month}&collectionId=55a52dfce4b09a8bb0485083&crumb=bar'
mock_request.register_uri('GET',
url=url,
status_code=200,
json=self.events_api_json)
result = city_blossoms.get_event_data()
expected = self.events_api_json
self.assertEqual(result, expected)
def test_handle_archive_redirect(self,
mock_get_notice_url_from_archive_list,
mock_request):
mock_get_notice_url_from_archive_list.return_value = 'https://www.fbo.gov/index?s=opportunity&mode=form&id=6577a9fff18f484f728af34bb6179872&tab=core&_cview=0'
redirect_url = 'https://www.fbo.gov/index?s=opportunity&mode=list&tab=archives'
jar = requests_mock.CookieJar()
jar.set('foo', 'bar', domain='fbo.gov')
mock_request.register_uri('GET',
url=redirect_url,
content=handle_archive_redirect_table.body,
status_code=200,
cookies=jar,
complete_qs=True)
result = handle_archive_redirect('www.test.gov', redirect_url,
{'foo': 'bar'}, 'notice_date',
'notice_type')
expected = mock_get_notice_url_from_archive_list.return_value
self.assertEqual(result, expected)
def test_handle_dla_url_list(self, mock_handle_archive_redirect,
mock_get_redirect_url, mock_request):
mock_handle_archive_redirect.return_value = 'https://www.foo.gov'
mock_get_redirect_url.return_value = 'https://www.fbo.gov/index?s=opportunity&mode=list&tab=list'
url = 'https://www.test.gov/spg/dla'
notice_date = '011715'
notice_type = 'PRESOL'
jar = requests_mock.CookieJar()
jar.set('foo', 'bar', domain='fbo.gov')
mock_request.register_uri(
'HEAD',
url=url,
text='test',
headers={'Location': mock_get_redirect_url.return_value},
status_code=302,
cookies=jar)
result = handle_dla_url(url, notice_date, notice_type)
expected = mock_handle_archive_redirect.return_value
self.assertEqual(result, expected)
def testing_login(self):
with open('./tests/unit/fixtures/step_1.txt', 'r') as f:
step_1 = f.read()
with open('./tests/unit/fixtures/step_4.txt', 'r') as f:
step_4 = f.read()
with open('./tests/unit/fixtures/step_7.txt', 'r') as f:
step_7 = f.read()
with open('./tests/unit/fixtures/cookies.json', 'r') as f:
c = json.load(f)
jar = requests_mock.CookieJar()
for x in ('gl5PlayerId', 'gl5SessionKey'):
jar.set(x, c.get(x, None), domain='lobby.kingdoms.com', path='/')
with open('./tests/unit/fixtures/step_8.json', 'r') as f:
step_8 = json.load(f)
with open('./tests/unit/fixtures/step_10.txt', 'r') as f:
step_10 = f.read()
with open('./tests/unit/fixtures/step_13.txt', 'r') as f:
step_13 = f.read()
with open('./tests/unit/fixtures/step_15.json', 'r') as f:
step_15 = json.load(f)
jar2 = requests_mock.CookieJar()
for x in ('t5SessionKey', 't5mu'):
jar2.set(x,
c.get(x, None),
domain=f'{c.get("gameworld")}.kingdoms.com',
path='/')
with requests_mock.mock() as m:
m.register_uri(
'GET',
'https://mellon-t5.traviangames.com/authentication/login',
text=step_1)
m.register_uri(
'POST',
'https://mellon-t5.traviangames.com/authentication/login/ajax/form-validate?msid=hef39l99m6b9ae5gj42mf7udf4&msname=msid',
text=step_4)
m.register_uri(
'GET',
'https://lobby.kingdoms.com/api/login.php?token=5b717b2096697410478b2c0d77cfa8ad&msid=hef39l99m6b9ae5gj42mf7udf4&msname=msid',
text=step_7,
cookies=jar)
m.register_uri('POST',
'https://lobby.kingdoms.com/api/index.php',
json=step_8)
m.register_uri(
'GET',
'https://mellon-t5.traviangames.com/game-world/join/gameWorldId/111?msid=hef39l99m6b9ae5gj42mf7udf4&msname=msid',
text=step_10)
m.register_uri(
'GET',
'https://com93.kingdoms.com/api/login.php?token=db3dad78c114051bfa01da3a452cea23&msid=hef39l99m6b9ae5gj42mf7udf4&msname=msid',
text=step_13,
cookies=jar2)
m.register_uri('POST',
'https://com93.kingdoms.com/api/',
json=step_15)
g = login('*****@*****.**', 'dummypassword', 'com93', None)
self.assertEqual(g.session, '6a07238e83ccc69977ea')
self.assertEqual(g.player_id, 123)
self.assertEqual(g.gameworld, 'com93')
self.assertEqual(g.api_root, 'https://com93.kingdoms.com/api/?')
def __init__(self, m: requests_mock.Mocker):
self._m = m
self._headers: Dict[str, str] = {}
self._cookiejar = requests_mock.CookieJar()
