def test_comprehensive_list_of_messages():
"""Ensure that we have a comprehensive set of tests for known messages, by
forcing a manual update to this list when a message is added or removed, and
reminding the developer to add or remove the associate test."""
valid_message_list = list(valid_usage_statistics_messages.keys())
# NOTE: If you are changing the expected valid message list below, you need
# to also update one or more tests below!
assert valid_message_list == [
"data_context.__init__",
"data_asset.validate",
"data_context.add_datasource",
"data_context.build_data_docs",
"data_context.open_data_docs",
"data_context.save_expectation_suite",
"datasource.sqlalchemy.connect",
"cli.init.create",
"cli.checkpoint.list",
"cli.datasource.list",
"cli.datasource.new",
"cli.datasource.profile",
"cli.docs.build",
"cli.docs.clean",
"cli.docs.list",
"cli.new_ds_choice",
"cli.project.check_config",
"cli.store.list",
"cli.suite.demo",
"cli.suite.edit",
"cli.suite.list",
"cli.suite.new",
"cli.suite.scaffold",
"cli.validation_operator.list",
"cli.validation_operator.run",
]
def test_comprehensive_list_of_messages():
"""Ensure that we have a comprehensive set of tests for known messages, by
forcing a manual update to this list when a message is added or removed, and
reminding the developer to add or remove the associate test."""
valid_message_list = list(valid_usage_statistics_messages.keys())
# NOTE: If you are changing the expected valid message list below, you need
# to also update one or more tests below!
assert set(valid_message_list) == {
"cli.checkpoint.delete",
"cli.checkpoint.list",
"cli.checkpoint.new",
"cli.checkpoint.run",
"cli.checkpoint.script",
"cli.datasource.delete",
"cli.datasource.list",
"cli.datasource.new",
"cli.datasource.profile",
"cli.docs.build",
"cli.docs.clean",
"cli.docs.list",
"cli.init.create",
"cli.new_ds_choice",
"cli.project.check_config",
"cli.project.upgrade",
"cli.store.list",
"cli.suite.delete",
"cli.suite.demo",
"cli.suite.edit",
"cli.suite.list",
"cli.suite.new",
"cli.suite.scaffold",
"cli.validation_operator.list",
"cli.validation_operator.run",
"data_asset.validate",
"data_context.__init__",
"data_context.add_datasource",
"data_context.get_batch_list",
"data_context.build_data_docs",
"data_context.open_data_docs",
"data_context.run_checkpoint",
"data_context.save_expectation_suite",
"data_context.test_yaml_config",
"data_context.run_validation_operator",
"datasource.sqlalchemy.connect",
"execution_engine.sqlalchemy.connect",
"checkpoint.run",
"expectation_suite.add_expectation",
"legacy_profiler.build_suite",
"profiler.run",
"data_context.run_profiler_on_data",
"data_context.run_profiler_with_dynamic_arguments",
"profiler.result.get_expectation_suite",
"data_assistant.result.get_expectation_suite",
}
# Note: "cli.project.upgrade" has no base event, only .begin and .end events
assert set(valid_message_list) == set(
UsageStatsEvents.get_all_event_names_no_begin_end_events() +
["cli.project.upgrade"])