valitydev/yandex-tank
14
0
Fork 0
mirror of https://github.com/valitydev/yandex-tank.git synced 2024-11-06 18:35:18 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

stepper tests

Browse Source
This commit is contained in:
Alexey Lavrenuke (load testing) 2013-09-06 17:50:04 +04:00
parent 055991a742
commit 66581df9a9
2 changed files with 75 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

102
Tests/StepperTest.py
View File

@ -7,13 +7,13 @@ import os
class StepperTestCase(TankTestCase):
data = None
def test_regular(self, src_f_path='data/dummy.ammo'):
def test_regular(self):
temp_stpd = tempfile.mkstemp()[1]
with open(temp_stpd, 'w') as stpd_file:
Stepper(
rps_schedule=["const(1,10)"],
http_ver='1.1',
ammo_file=src_f_path,
ammo_file='data/dummy.ammo',
instances_schedule=[],
loop_limit=-1,
ammo_limit=-1,
@ -23,41 +23,89 @@ class StepperTestCase(TankTestCase):
).write(stpd_file)
res = open(temp_stpd, 'r').read()
self.assertNotEquals("", res)
# ensure that we got a valid stpd file here
self.assertEquals(317, os.path.getsize(temp_stpd))
self.assertEquals(277, os.path.getsize(temp_stpd))
def test_regular_gziped(self, src_f_path='data/dummy-ammo-compressed.gz'):
def test_regular_gziped(self):
""" Call stepper on dummy HTTP ammo file with 1 req.
Source ammo file compressed with gzip 1.4 lib.
"""
return self.test_regular(src_f_path=src_f_path)
temp_stpd = tempfile.mkstemp()[1]
with open(temp_stpd, 'w') as stpd_file:
Stepper(
rps_schedule=["const(1,10)"],
http_ver='1.1',
ammo_file='data/dummy-ammo-compressed.gz',
instances_schedule=[],
loop_limit=-1,
ammo_limit=-1,
uris=[],
headers=[],
autocases=0,
).write(stpd_file)
res = open(temp_stpd, 'r').read()
self.assertNotEquals("", res)
self.assertEquals(347, os.path.getsize(temp_stpd))
def test_uri(self):
stepper = Stepper(tempfile.mkstemp()[1])
stepper.rps_schedule = ["const(1,10)"]
stepper.uris = ["/", "/test"]
stepper.header_http = "1.1"
stepper.headers = ["[Host: ya.ru]", "[Connection: close]"]
stepper.generate_stpd()
res = open(stepper.stpd_file, 'r').read()
temp_stpd = tempfile.mkstemp()[1]
with open(temp_stpd, 'w') as stpd_file:
Stepper(
rps_schedule=["const(1,10)"],
http_ver='1.1',
instances_schedule=[],
loop_limit=-1,
ammo_limit=-1,
uris=["/", "/test"],
headers=["[Host: ya.ru]", "[Connection: close]"],
).write(stpd_file)
res = open(temp_stpd, 'r').read()
self.assertNotEquals("", res)
self.assertEquals(619, os.path.getsize(stepper.stpd_file))
self.assertEquals(657, os.path.getsize(temp_stpd))
def test_uri_style(self):
temp_stpd = tempfile.mkstemp()[1]
with open(temp_stpd, 'w') as stpd_file:
Stepper(
rps_schedule=["const(1,10)"],
http_ver='1.1',
ammo_file="data/uri.ammo",
instances_schedule=[],
loop_limit=-1,
ammo_limit=-1,
headers=["[Host: ya.ru]", "[Connection: close]"],
).write(stpd_file)
res = open(temp_stpd, 'r').read()
self.assertNotEquals("", res)
self.assertEquals(1567, os.path.getsize(temp_stpd))
def test_free_inst_sched(self):
stepper = Stepper(tempfile.mkstemp()[1])
stepper.ammofile = "data/dummy.ammo"
stepper.instances_schedule = "line(1,5,15)"
stepper.loop_limit = 15
stepper.generate_stpd()
res = open(stepper.stpd_file, 'r').read()
temp_stpd = tempfile.mkstemp()[1]
with open(temp_stpd, 'w') as stpd_file:
Stepper(
rps_schedule=[],
http_ver='1.1',
instances_schedule=["line(1,5,15)"],
loop_limit=15,
ammo_limit=-1,
uris=["/", "/test"],
headers=["[Host: ya.ru]", "[Connection: close]"],
).write(stpd_file)
res = open(temp_stpd, 'r').read()
self.assertNotEquals("", res)
self.assertEquals(406, os.path.getsize(stepper.stpd_file))
self.assertEquals(1900, os.path.getsize(temp_stpd))
def test_free_inst(self):
stepper = Stepper(tempfile.mkstemp()[1])
stepper.ammofile = "data/dummy.ammo"
stepper.loop_limit = 2
stepper.generate_stpd()
res = open(stepper.stpd_file, 'r').read()
temp_stpd = tempfile.mkstemp()[1]
with open(temp_stpd, 'w') as stpd_file:
Stepper(
rps_schedule=[],
http_ver='1.1',
instances_schedule=["line(1,5,15)"],
loop_limit=2,
ammo_limit=-1,
uris=["/", "/test"],
headers=["[Host: ya.ru]", "[Connection: close]"],
).write(stpd_file)
res = open(temp_stpd, 'r').read()
self.assertNotEquals("", res)
self.assertEquals(56, os.path.getsize(stepper.stpd_file))
self.assertEquals(258, os.path.getsize(temp_stpd))

BIN
Tests/data/dummy-ammo-compressed.gz Normal file
View File

Binary file not shown.