def __init__(self, input_handler: Callable, bindings: KeyBindings, completer: Completer): self.search_field = create_search_field() self.input_field = create_input_field(completer=completer) self.output_field = create_output_field() self.log_field = create_log_field(self.search_field) self.timer = create_timer() self.process_usage = create_process_monitor() self.trade_monitor = create_trade_monitor() self.layout = generate_layout(self.input_field, self.output_field, self.log_field, self.search_field, self.timer, self.process_usage, self.trade_monitor) # add self.to_stop_config to know if cancel is triggered self.to_stop_config: bool = False self.live_updates = False self.bindings = bindings self.input_handler = input_handler self.input_field.accept_handler = self.accept self.app: Optional[Application] = None # settings self.prompt_text = ">>> " self.pending_input = None self.input_event = None self.hide_input = False # start ui tasks loop = asyncio.get_event_loop() loop.create_task(start_timer(self.timer)) loop.create_task(start_process_monitor(self.process_usage)) loop.create_task(start_trade_monitor(self.trade_monitor))
def test_start_process_monitor(self, mock_process, mock_sleep): mock_process.return_value.num_threads.return_value = 2 mock_process.return_value.cpu_percent.return_value = 30 mock_process.return_value.memory_info.return_value = [0, 1024.] mock_monitor = MagicMock() mock_sleep.side_effect = Exception("returns") with self.assertRaises(Exception) as context: asyncio.get_event_loop().run_until_complete( start_process_monitor(mock_monitor)) self.assertEqual('returns', str(context.exception)) self.assertEqual("CPU: 30%, Mem: 512.00 B, Threads: 2, ", mock_monitor.log.call_args_list[0].args[0])
def test_start_process_monitor(self, mock_process, mock_sleep): mock_process.return_value.num_threads.return_value = 2 mock_process.return_value.cpu_percent.return_value = 30 memory_info = MagicMock() type(memory_info).vms = PropertyMock(return_value=1024.0) type(memory_info).rss = PropertyMock(return_value=1024.0) mock_process.return_value.memory_info.return_value = memory_info mock_monitor = MagicMock() mock_sleep.side_effect = asyncio.CancelledError with self.assertRaises(asyncio.CancelledError): self.async_run_with_timeout(start_process_monitor(mock_monitor)) self.assertEqual( "CPU: 30%, Mem: 512.00 B (1.00 KB), Threads: 2, ", mock_monitor.log.call_args_list[0].args[0])
def __init__(self, client_config_map: ClientConfigAdapter, input_handler: Callable, bindings: KeyBindings, completer: Completer, command_tabs: Dict[str, CommandTab]): super().__init__() self.client_config_map: Union[ClientConfigAdapter, ClientConfigMap] = client_config_map self.command_tabs = command_tabs self.search_field = create_search_field() self.input_field = create_input_field(completer=completer) self.output_field = create_output_field(client_config_map) self.log_field = create_log_field(self.search_field) self.right_pane_toggle = create_log_toggle(self.toggle_right_pane) self.live_field = create_live_field() self.log_field_button = create_tab_button("Log-pane", self.log_button_clicked) self.timer = create_timer() self.process_usage = create_process_monitor() self.trade_monitor = create_trade_monitor() self.layout, self.layout_components = generate_layout( self.input_field, self.output_field, self.log_field, self.right_pane_toggle, self.log_field_button, self.search_field, self.timer, self.process_usage, self.trade_monitor, self.command_tabs) # add self.to_stop_config to know if cancel is triggered self.to_stop_config: bool = False self.live_updates = False self.bindings = bindings self.input_handler = input_handler self.input_field.accept_handler = self.accept self.app: Optional[Application] = None # settings self.prompt_text = ">>> " self.pending_input = None self.input_event = None self.hide_input = False # stdout redirection stack self._stdout_redirect_context: ExitStack = ExitStack() # start ui tasks loop = asyncio.get_event_loop() loop.create_task(start_timer(self.timer)) loop.create_task(start_process_monitor(self.process_usage)) loop.create_task(start_trade_monitor(self.trade_monitor))
def __init__(self, input_handler: Callable, bindings: KeyBindings, completer: Completer): self.search_field = create_search_field() self.input_field = create_input_field(completer=completer) self.output_field = create_output_field() self.log_field = create_log_field(self.search_field) self.timer = create_timer() self.process_usage = create_process_monitor() self.trade_monitor = create_trade_monitor() self.layout = generate_layout(self.input_field, self.output_field, self.log_field, self.search_field, self.timer, self.process_usage, self.trade_monitor) # add self.to_stop_config to know if cancel is triggered self.to_stop_config: bool = False self.live_updates = False self.bindings = bindings self.input_handler = input_handler self.input_field.accept_handler = self.accept self.app = Application(layout=self.layout, full_screen=True, key_bindings=self.bindings, style=load_style(), mouse_support=True, clipboard=PyperclipClipboard()) # settings self.prompt_text = ">>> " self.pending_input = None self.input_event = None self.hide_input = False # start ui tasks loop = asyncio.get_event_loop() loop.create_task(start_timer(self.timer)) loop.create_task(start_process_monitor(self.process_usage)) loop.create_task(start_trade_monitor(self.trade_monitor))