def test_pick_default_log_mode():
with mock.patch('paasta_tools.cli.cmds.logs.LogReader.tail_logs',
autospec=True) as tail_logs:
args, _ = parse_args(["logs"])
fake_reader = logs.LogReader()
fake_reader.SUPPORTS_TAILING = True
logs.pick_default_log_mode(args,
fake_reader,
service=None,
levels=None,
components=None,
clusters=None,
instances=None)
# Only supports tailing so that's the one that should be used
assert tail_logs.call_count == 1
with mock.patch('paasta_tools.cli.cmds.logs.LogReader.print_logs_by_time',
autospec=True) as logs_by_time:
args, _ = parse_args(["logs"])
fake_reader = logs.LogReader()
fake_reader.SUPPORTS_TAILING = True
fake_reader.SUPPORTS_TIME = True
logs.pick_default_log_mode(args,
fake_reader,
service=None,
levels=None,
components=None,
clusters=None,
instances=None)
# Supports tailing and time, but time should be prioritized
assert logs_by_time.call_count == 1
with mock.patch('paasta_tools.cli.cmds.logs.LogReader.print_last_n_logs',
autospec=True) as logs_by_lines:
args, _ = parse_args(["logs"])
fake_reader = logs.LogReader()
fake_reader.SUPPORTS_TAILING = True
fake_reader.SUPPORTS_TIME = True
fake_reader.SUPPORTS_LINE_COUNT = True
logs.pick_default_log_mode(args,
fake_reader,
service=None,
levels=None,
components=None,
clusters=None,
instances=None)
# Supports tailing , time and line counts. Line counts should be prioritized
assert logs_by_lines.call_count == 1
def test_pick_default_log_mode():
with mock.patch('paasta_tools.cli.cmds.logs.LogReader.tail_logs', autospec=True) as tail_logs:
args, _ = parse_args(["logs"])
fake_reader = logs.LogReader()
fake_reader.SUPPORTS_TAILING = True
logs.pick_default_log_mode(
args, fake_reader, service=None, levels=None, components=None, clusters=None, instances=None)
# Only supports tailing so that's the one that should be used
assert tail_logs.call_count == 1
with mock.patch('paasta_tools.cli.cmds.logs.LogReader.print_logs_by_time', autospec=True) as logs_by_time:
args, _ = parse_args(["logs"])
fake_reader = logs.LogReader()
fake_reader.SUPPORTS_TAILING = True
fake_reader.SUPPORTS_TIME = True
logs.pick_default_log_mode(
args, fake_reader, service=None, levels=None, components=None, clusters=None, instances=None)
# Supports tailing and time, but time should be prioritized
assert logs_by_time.call_count == 1
with mock.patch('paasta_tools.cli.cmds.logs.LogReader.print_last_n_logs', autospec=True) as logs_by_lines:
args, _ = parse_args(["logs"])
fake_reader = logs.LogReader()
fake_reader.SUPPORTS_TAILING = True
fake_reader.SUPPORTS_TIME = True
fake_reader.SUPPORTS_LINE_COUNT = True
logs.pick_default_log_mode(
args, fake_reader, service=None, levels=None, components=None, clusters=None, instances=None)
# Supports tailing , time and line counts. Line counts should be prioritized
assert logs_by_lines.call_count == 1