def test_statusbar(monkeypatch): o = options.Options() m = master.ConsoleMaster(o) m.options.update( setheaders=[":~q:foo:bar"], replacements=[":~q:foo:bar"], ignore_hosts=["example.com", "example.org"], tcp_hosts=["example.tcp"], intercept="~q", view_filter="~dst example.com", stickycookie="~dst example.com", stickyauth="~dst example.com", console_default_contentview="javascript", anticache=True, anticomp=True, showhost=True, server_replay_refresh=False, server_replay_kill_extra=True, upstream_cert=False, stream_large_bodies="3m", mode="transparent", ) m.options.update(view_order='url', console_focus_follow=True) monkeypatch.setattr(m.addons.get("clientplayback"), "count", lambda: 42) monkeypatch.setattr(m.addons.get("serverplayback"), "count", lambda: 42) bar = statusbar.StatusBar(m) # this already causes a redraw assert bar.ib._w
def test_commands_exist(): km = keymap.Keymap(None) defaultkeys.map(km) assert km.bindings m = master.ConsoleMaster(None) m.load_flow(tflow()) for binding in km.bindings: cmd, *args = command.lexer(binding.command) assert cmd in m.commands.commands cmd_obj = m.commands.commands[cmd] try: cmd_obj.prepare_args(args) except Exception as e: raise ValueError("Invalid command: {}".format( binding.command)) from e
async def test_commands_exist(): command_manager = command.CommandManager(ctx) km = keymap.Keymap(None) defaultkeys.map(km) assert km.bindings m = master.ConsoleMaster(None) await m.load_flow(tflow()) for binding in km.bindings: parsed, _ = command_manager.parse_partial(binding.command.strip()) cmd = parsed[0].value args = [a.value for a in parsed[1:] if a.type != mitmproxy.types.Space] assert cmd in m.commands.commands cmd_obj = m.commands.commands[cmd] try: cmd_obj.prepare_args(args) except Exception as e: raise ValueError(f"Invalid command: {binding.command}") from e
def test_shorten_message_narrow(): o = options.Options() m = master.ConsoleMaster(o) ab = statusbar.ActionBar(m) shorten_msg = ab.shorten_message("error", max_width=4) assert shorten_msg == [(None, "\u2026"), ("warn", "(more in eventlog)")]
def test_shorten_message(message, ready_message): o = options.Options() m = master.ConsoleMaster(o) ab = statusbar.ActionBar(m) assert ab.shorten_message(message, max_width=30) == ready_message