def test_run_with_config(self):
config = {
'concurrency': 100,
'method': 'get',
'options': {
'cookies': {
'test_cookie': 'cookie_value'
},
'headers': {
'name': 'value'
}
},
'request_nums': 1,
'url': 'http://localhost:5000'
}
result = benchmark(**config)
for response in result.responses:
for cookie_pair in config['options']['cookies']:
assert _str(cookie_pair) in response.text
for header_pair in config['options']['headers'].items():
assert 'HTTP_{}={}'.format(header_pair[0],
header_pair[1]) in response.text
assert self.get_count() == config['concurrency']
def test_request_method(self):
for method in HTTP_VERBS:
result = benchmark(urlparse.urlparse(self.url, 'method'),
method=method)
responses = result.responses
for response in responses:
assert method in response.text
def test_with_cookie(self):
options = {'cookies': {'test': 'test'}}
result = benchmark(urlparse.urljoin(self.url, 'cookie'),
options=options)
responses = result.responses
for response in responses:
assert _str(options['cookies']) in response.text
def test_with_header(self):
options = {'headers': {'self_defined_header': 'blablabla'}}
result = benchmark(urlparse.urljoin(self.url, 'header'),
options=options)
responses = result.responses
for response in responses:
assert 'HTTP_{}={}'.format('self_defined_header'.upper(),
'blablabla') in response.text
def test_connect_error(self):
result = benchmark('http://127.0.0.1:23232',
request_nums=5,
concurrency=1)
assert len(result.errors) == 5
def test_request_num(self):
benchmark(self.url, concurrency=1, request_nums=233)
result = self.get_count()
assert result == 233