def test_user_agent(self):
browser = RoboBrowser(user_agent='freddie')
browser.open('http://robobrowser.com/links/')
assert_true('User-Agent' in browser.session.headers)
assert_equal(
browser.session.headers['User-Agent'], 'freddie'
)
def test_cache_off(self):
self.browser = RoboBrowser(cache=False)
self.browser.open('http://httpbin.org/')
resp1 = self.browser.state.response
self.browser.open('http://httpbin.org/')
resp2 = self.browser.state.response
assert_true(resp1 is not resp2)
def test_state_deque_max_length(self):
browser = RoboBrowser(history=5)
for _ in range(5):
browser.open('http://robobrowser.com/page1/')
assert_equal(len(browser._states), 5)
browser.open('http://robobrowser.com/page2/')
assert_equal(len(browser._states), 5)
def test_custom_headers(self):
session = requests.Session()
session.headers.update({
'Content-Encoding': 'gzip',
})
browser = RoboBrowser(session=session)
browser.open('http://robobrowser.com/links/')
assert_equal(browser.response.request.headers.get('Content-Encoding'),
'gzip')
def test_headers(self):
headers = {
'X-Song': 'Innuendo',
'X-Writer': 'Freddie',
}
browser = RoboBrowser(headers=headers)
browser.open('http://robobrowser.com/links/')
for key, value in headers.items():
assert_equal(browser.session.headers[key], value)
def setUp(self):
self.browser = RoboBrowser()
self.browser.open('http://robobrowser.com/links/')
def test_call_allow_redirects(self, mock_get):
browser = RoboBrowser(allow_redirects=True)
browser.open('http://robobrowser.com/', allow_redirects=False)
assert_true(mock_get.called)
kwargs = mock_get.mock_calls[0][2]
assert_true(kwargs.get('allow_redirects') is False)
def test_call_timeout(self, mock_get):
browser = RoboBrowser(timeout=5)
browser.open('http://robobrowser.com/', timeout=10)
assert_true(mock_get.called)
kwargs = mock_get.mock_calls[0][2]
assert_equal(kwargs.get('timeout'), 10)
def test_no_timeout(self, mock_get):
browser = RoboBrowser()
browser.open('http://robobrowser.com/')
assert_true(mock_get.called)
kwargs = mock_get.mock_calls[0][2]
assert_true(kwargs.get('timeout') is None)
def test_default_headers(self):
browser = RoboBrowser()
assert_equal(browser.session.headers, requests.Session().headers)
def setUp(self):
self.browser = RoboBrowser(history=True)
self.browser.open('http://robobrowser.com/page1/')
self.browser.open('http://robobrowser.com/page2/')
self.browser.open('http://robobrowser.com/page3/')
def test_call_verify(self, mock_get):
browser = RoboBrowser(verify=True)
browser.open('http://robobrowser.com/', verify=False)
mock_get.assert_called_once_with(
'http://robobrowser.com/', verify=False, timeout=None
)
def setUp(self):
self.browser = RoboBrowser(history=True)
def setUp(self):
self.browser = RoboBrowser()
self.browser.open('http://robobrowser.com/noname/')
def test_no_allow_redirects(self, mock_request):
browser = RoboBrowser()
browser.open('http://robobrowser.com/')
assert_true(mock_request.called)
kwargs = mock_request.mock_calls[0][2]
assert_true(kwargs.get('allow_redirects') is True)
def test_instance_timeout(self, mock_request):
browser = RoboBrowser(timeout=5)
browser.open('http://robobrowser.com/')
assert_true(mock_request.called)
kwargs = mock_request.mock_calls[0][2]
assert_equal(kwargs.get('timeout'), 5)
def setUp(self):
self.browser = RoboBrowser()
def test_call_timeout(self, mock_get):
browser = RoboBrowser(timeout=5)
browser.open('http://robobrowser.com/', timeout=10)
mock_get.assert_called_once_with(
'http://robobrowser.com/', timeout=10, verify=True
)
def test_state_deque_no_history(self):
browser = RoboBrowser(history=False)
for _ in range(5):
browser.open('http://robobrowser.com/page1/')
assert_equal(len(browser._states), 1)
assert_equal(browser._cursor, 0)