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()