mirror of
https://github.com/valitydev/yandex-tank.git
synced 2024-11-08 11:13:51 +00:00
57 lines
2.0 KiB
Python
57 lines
2.0 KiB
Python
from Tank.stepper import Stepper
|
|
from Tests.TankTests import TankTestCase
|
|
import tempfile
|
|
import os
|
|
|
|
class StepperTestCase(TankTestCase):
|
|
data = None
|
|
|
|
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="data/dummy.ammo",
|
|
instances_schedule=[],
|
|
loop_limit=-1,
|
|
ammo_limit=-1,
|
|
uris=[],
|
|
headers=[],
|
|
autocases=0,
|
|
).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))
|
|
|
|
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()
|
|
self.assertNotEquals("", res)
|
|
self.assertEquals(619, os.path.getsize(stepper.stpd_file))
|
|
|
|
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()
|
|
self.assertNotEquals("", res)
|
|
self.assertEquals(406, os.path.getsize(stepper.stpd_file))
|
|
|
|
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()
|
|
self.assertNotEquals("", res)
|
|
self.assertEquals(56, os.path.getsize(stepper.stpd_file))
|