def test_try_get_default_header_raises_cli_error_when_using_include_all_with_none_table_format(
):
with pytest.raises(errors.Code42CLIError) as err:
try_get_default_header(True, {}, OutputFormat.CSV)
assert str(
err.value) == "--include-all only allowed for Table output format."
def search(
state,
format,
begin,
end,
advanced_query,
use_checkpoint,
saved_search,
or_query,
include_all,
**kwargs,
):
"""Search for file events."""
output_header = ext.try_get_default_header(include_all,
SEARCH_DEFAULT_HEADER, format)
formatter = FileEventsOutputFormatter(format, output_header)
cursor = (_get_file_event_cursor_store(state.profile.name)
if use_checkpoint else None)
handlers = ext.create_handlers(
state.sdk,
FileEventExtractor,
cursor,
use_checkpoint,
formatter=formatter,
force_pager=include_all,
)
_call_extractor(state, handlers, begin, end, or_query, advanced_query,
saved_search, **kwargs)
handle_no_events(not handlers.TOTAL_EVENTS and not errors.ERRORED)
def search(
state,
format,
begin,
end,
advanced_query,
use_checkpoint,
saved_search,
or_query,
include_all,
**kwargs,
):
"""Search for file events."""
output_header = ext.try_get_default_header(include_all,
_create_search_header_map(),
format)
formatter = FileEventsOutputFormatter(format, output_header)
cursor = _get_cursor(state, use_checkpoint)
handlers = ext.create_handlers(
state.sdk,
FileEventExtractor,
cursor,
use_checkpoint,
formatter=formatter,
force_pager=include_all,
)
_extract(state, handlers, begin, end, or_query, advanced_query,
saved_search, **kwargs)
def search(
cli_state,
format,
begin,
end,
advanced_query,
use_checkpoint,
or_query,
include_all,
**kwargs,
):
"""Search for alerts."""
output_header = ext.try_get_default_header(include_all,
_get_default_output_header(),
format)
formatter = OutputFormatter(format, output_header)
cursor = _get_alert_cursor_store(
cli_state.profile.name) if use_checkpoint else None
handlers = ext.create_handlers(
cli_state.sdk,
AlertExtractor,
cursor,
use_checkpoint,
formatter=formatter,
force_pager=include_all,
)
_call_extractor(cli_state, handlers, begin, end, or_query, advanced_query,
**kwargs)
handle_no_events(not handlers.TOTAL_EVENTS and not errors.ERRORED)
def test_try_get_default_header_returns_none_when_is_table_and_told_to_include_all(
):
default_header = {"default": "header"}
actual = try_get_default_header(True, default_header, OutputFormat.TABLE)
assert actual is None
def test_try_get_default_header_uses_default_header_when_not_include_all():
default_header = {"default": "header"}
actual = try_get_default_header(False, default_header, OutputFormat.TABLE)
assert actual is default_header