mirror of
https://github.com/valitydev/yandex-tank.git
synced 2024-11-07 02:45:21 +00:00
40 lines
1.0 KiB
Python
40 lines
1.0 KiB
Python
from Tank.Core import TankCore
|
|
from Tank.Plugins.JMeter import JMeterPlugin
|
|
from Tests.TankTests import TankTestCase
|
|
import os
|
|
import tempfile
|
|
import time
|
|
import unittest
|
|
|
|
class JMeterPluginTestCase(TankTestCase):
|
|
def setUp(self):
|
|
core = TankCore()
|
|
(handler, name) = tempfile.mkstemp()
|
|
core.config.set_out_file(name)
|
|
core.load_configs(['config/jmeter.conf'])
|
|
self.foo = JMeterPlugin(core)
|
|
|
|
def tearDown(self):
|
|
del self.foo
|
|
self.foo = None
|
|
os.remove("jmeter.log")
|
|
|
|
def test_run(self):
|
|
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)
|
|
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)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|