mirror of
https://github.com/valitydev/yandex-tank.git
synced 2024-11-06 18:35:18 +00:00
71 lines
1.8 KiB
Python
71 lines
1.8 KiB
Python
from Tank.Core import TankCore
|
|
from Tank.Plugins.Aggregator import AggregatorPlugin, SecondAggregateData
|
|
from Tank.Plugins.ConsoleOnline import ConsoleOnlinePlugin, AbstractInfoWidget, \
|
|
RealConsoleMarkup
|
|
from Tests.TankTests import TankTestCase
|
|
import os
|
|
import tempfile
|
|
import unittest
|
|
from Tests import TankTests
|
|
|
|
|
|
class FakeConsoleMarkup(RealConsoleMarkup):
|
|
clear = "\n[clear]\n"
|
|
new_line = "\n"
|
|
|
|
YELLOW = '<y>'
|
|
RED = '<r>'
|
|
RED_DARK = '<rd>'
|
|
RESET = '<rst>'
|
|
CYAN = "<c>"
|
|
WHITE="<w>"
|
|
GREEN = "<g>"
|
|
MAGENTA = '<m>'
|
|
BG_MAGENTA = '<M>'
|
|
BG_GREEN = '<G>'
|
|
|
|
|
|
class ConsoleOnlinePluginTestCase(TankTestCase):
|
|
def setUp(self):
|
|
core = self.get_core()
|
|
name = tempfile.mkstemp()[1]
|
|
core.config.set_out_file(name)
|
|
core.load_configs(['config/console.conf'])
|
|
core.load_plugins()
|
|
self.foo = ConsoleOnlinePlugin(core)
|
|
self.foo.console_markup = FakeConsoleMarkup()
|
|
|
|
def tearDown(self):
|
|
del self.foo
|
|
self.foo = None
|
|
|
|
def test_run(self):
|
|
self.data=self.get_aggregate_data('data/preproc_single2.txt')
|
|
|
|
self.foo.configure()
|
|
self.foo.prepare_test()
|
|
self.foo.add_info_widget(TestWidget())
|
|
self.foo.add_info_widget(TestWidget2())
|
|
|
|
self.foo.start_test()
|
|
for i in range(1, 10):
|
|
self.foo.aggregate_second(self.data)
|
|
self.foo.end_test(0)
|
|
self.assertFalse(self.foo.render_exception)
|
|
|
|
class TestWidget(AbstractInfoWidget):
|
|
def render(self, screen):
|
|
return "Widget Data";
|
|
|
|
class TestWidget2(AbstractInfoWidget):
|
|
def get_index(self):
|
|
return 100;
|
|
|
|
def render(self, screen):
|
|
return "Widget Data 2";
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|
|
|