mirror of
https://github.com/valitydev/yandex-tank.git
synced 2024-11-06 18:35:18 +00:00
122 lines
4.1 KiB
Python
122 lines
4.1 KiB
Python
from Tank.Plugins.Aggregator import AggregatorPlugin
|
|
from Tank.Plugins.Phantom import PhantomPlugin, PhantomReader, StepperWrapper
|
|
from Tests.TankTests import TankTestCase
|
|
import os
|
|
import tempfile
|
|
import time
|
|
import unittest
|
|
|
|
class PhantomPluginTestCase(TankTestCase):
|
|
def setUp(self):
|
|
core = self.get_core()
|
|
core.load_configs(['config/phantom.conf'])
|
|
core.load_plugins()
|
|
core.plugins_configure()
|
|
core.plugins_prepare_test()
|
|
self.foo = PhantomPlugin(core)
|
|
|
|
def tearDown(self):
|
|
del self.foo
|
|
self.foo = None
|
|
if os.path.exists("ready_conf_phout.txt"):
|
|
os.remove("ready_conf_phout.txt")
|
|
|
|
def test_run(self):
|
|
self.foo.core.set_option(PhantomPlugin.SECTION, "config", '')
|
|
self.foo.configure()
|
|
self.foo.prepare_test()
|
|
reader = PhantomReader(AggregatorPlugin(self.foo.core), self.foo)
|
|
self.foo.start_test()
|
|
|
|
while self.foo.is_test_finished() < 0:
|
|
self.foo.log.debug("Not finished")
|
|
reader.check_open_files()
|
|
reader.get_next_sample(False)
|
|
time.sleep(1)
|
|
if self.foo.is_test_finished() != 0:
|
|
raise RuntimeError("RC: %s" % self.foo.is_test_finished())
|
|
self.foo.end_test(0)
|
|
reader.get_next_sample(True)
|
|
|
|
def test_run_ready_conf(self):
|
|
self.foo.core.add_artifact_file("ready_conf_phout.txt")
|
|
self.foo.configure()
|
|
self.foo.prepare_test()
|
|
self.foo.start_test()
|
|
while self.foo.is_test_finished() < 0:
|
|
self.foo.log.debug("Not finished")
|
|
time.sleep(1)
|
|
if self.foo.is_test_finished() != 0:
|
|
raise RuntimeError("RC: %s" % self.foo.is_test_finished())
|
|
self.foo.end_test(0)
|
|
|
|
def test_run_uri_style(self):
|
|
self.foo.set_option("ammofile", "")
|
|
self.foo.set_option("uris", "/")
|
|
self.foo.configure()
|
|
self.foo.prepare_test()
|
|
self.foo.start_test()
|
|
while self.foo.is_test_finished() < 0:
|
|
self.foo.log.debug("Not finished")
|
|
time.sleep(1)
|
|
if self.foo.is_test_finished() != 0:
|
|
raise RuntimeError("RC: %s" % self.foo.is_test_finished())
|
|
self.foo.end_test(0)
|
|
|
|
def test_run_interrupt(self):
|
|
self.foo.configure()
|
|
self.foo.prepare_test()
|
|
self.foo.start_test()
|
|
time.sleep(2)
|
|
self.foo.end_test(0)
|
|
|
|
def test_run_stepper_cache(self):
|
|
self.foo.configure()
|
|
self.foo.prepare_test()
|
|
self.foo.prepare_test()
|
|
|
|
def test_domain_name(self):
|
|
self.foo.core.set_option('phantom', 'address', 'yandex.ru')
|
|
self.foo.configure()
|
|
|
|
|
|
def test_domain_name_fail(self):
|
|
self.foo.core.set_option('phantom', 'address', 'ya.ru')
|
|
try:
|
|
self.foo.configure()
|
|
raise RuntimeError()
|
|
except:
|
|
pass
|
|
|
|
|
|
def test_multiload_parsing(self):
|
|
self.foo.core.set_option('phantom', 'rps_schedule', 'const(1,1) line(1,100,60)\nstep(1,10,1,10)')
|
|
self.foo.configure()
|
|
self.assertEquals(['const(1,1)', 'line(1,100,60)', 'step(1,10,1,10)'], self.foo.stepper.rps_schedule)
|
|
|
|
|
|
def test_reader(self):
|
|
self.foo.phout_file = 'data/phout_timeout_mix.txt'
|
|
self.foo.phantom_start_time = time.time()
|
|
reader = PhantomReader(AggregatorPlugin(self.foo.core), self.foo)
|
|
reader.check_open_files()
|
|
|
|
data = reader.get_next_sample(False)
|
|
while data:
|
|
times_sum = 0
|
|
for timing in data.overall.times_dist:
|
|
times_sum += timing['count']
|
|
self.assertEquals(sum(data.overall.net_codes.values()), times_sum)
|
|
data = reader.get_next_sample(False)
|
|
|
|
def test_stepper_no_steps(self):
|
|
self.foo.core.set_option('phantom', 'rps_schedule', '')
|
|
self.foo.core.set_option('phantom', 'instances_schedule', '')
|
|
wrapper=StepperWrapper(self.foo)
|
|
wrapper.ammo_file='data/dummy.ammo'
|
|
wrapper.prepare_stepper()
|
|
wrapper.prepare_stepper()
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|