def test_add_family_priority(cli_runner: CliRunner, base_context: CGConfig, helpers: StoreHelpers):
"""Test that the added case get the priority we send in"""
# GIVEN a database with a customer and an panel
disk_store: Store = base_context.status_db
# WHEN adding a case
customer: models.Customer = helpers.ensure_customer(store=disk_store)
customer_id = customer.internal_id
panel: models.Panel = helpers.ensure_panel(store=disk_store)
panel_id = panel.name
name = "case_name"
priority = "priority"
result = cli_runner.invoke(
add,
[
"family",
"--panel",
panel_id,
"--priority",
priority,
"--analysis",
CLI_OPTION_ANALYSIS,
"--data-delivery",
CLI_OPTION_DELIVERY,
customer_id,
name,
],
obj=base_context,
)
# THEN it should be added
assert result.exit_code == 0
assert disk_store.Family.query.count() == 1
assert disk_store.Family.query.first().priority_human == priority
def test_add_family_bad_data_delivery(
cli_runner: CliRunner, base_context: CGConfig, helpers: StoreHelpers
):
"""Test to add a case using only the required arguments"""
# GIVEN a database with a customer and an panel
# WHEN adding a case without data delivery
disk_store: Store = base_context.status_db
customer: models.Customer = helpers.ensure_customer(store=disk_store)
customer_id = customer.internal_id
panel: models.Panel = helpers.ensure_panel(store=disk_store)
panel_id = panel.name
name = "case_name"
non_existing_data_delivery = "aws"
result = cli_runner.invoke(
add,
[
"case",
"--panel",
panel_id,
"--analysis",
CLI_OPTION_ANALYSIS,
"--data-delivery",
non_existing_data_delivery,
customer_id,
name,
],
obj=base_context,
)
# THEN it should not be added
assert result.exit_code != 0
assert disk_store.Family.query.count() == 0
def test_add_family_required(cli_runner: CliRunner, base_context: CGConfig, helpers: StoreHelpers):
"""Test to add a case using only the required arguments"""
# GIVEN a database with a customer and an panel
disk_store: Store = base_context.status_db
customer: models.Customer = helpers.ensure_customer(store=disk_store)
customer_id = customer.internal_id
panel: models.Panel = helpers.ensure_panel(store=disk_store)
panel_id = panel.name
name = "case_name"
# WHEN adding a panel
result = cli_runner.invoke(
add,
[
"family",
"--panel",
panel_id,
"--analysis",
CLI_OPTION_ANALYSIS,
"--data-delivery",
CLI_OPTION_DELIVERY,
customer_id,
name,
],
obj=base_context,
)
# THEN it should be added
assert result.exit_code == 0
assert disk_store.Family.query.count() == 1
assert disk_store.Family.query.first().name == name
assert disk_store.Family.query.first().panels == [panel_id]