|
| 1 | +// SPDX-License-Identifier: BUSL-1.1 |
| 2 | + |
| 3 | +//! Name-based tenant reference resolution on the DROP / ALTER / PURGE |
| 4 | +//! TENANT paths. |
| 5 | +//! |
| 6 | +//! Verifies that the legacy `DROP TENANT <id>` form keeps working and that |
| 7 | +//! a tenant name (bare or single-quoted) resolves to the same id via the |
| 8 | +//! shared `resolve_tenant_ref` helper, mirroring the already-shipped |
| 9 | +//! `CREATE TENANT <name>` / `SHOW TENANT <name>` paths. |
| 10 | +
|
| 11 | +use crate::common::pgwire_auth_helpers::{ddl_err, ddl_ok, make_state_with_catalog, superuser}; |
| 12 | + |
| 13 | +// ─── DROP TENANT by name ───────────────────────────────────────────────────── |
| 14 | + |
| 15 | +/// `DROP TENANT <id>` (numeric) — regression that the legacy form still |
| 16 | +/// works after the resolver refactor. |
| 17 | +#[tokio::test] |
| 18 | +async fn drop_tenant_by_numeric_id_still_works() { |
| 19 | + let state = make_state_with_catalog(); |
| 20 | + let su = superuser(); |
| 21 | + |
| 22 | + ddl_ok(&state, &su, "CREATE TENANT acme_drop_num ID 7142").await; |
| 23 | + ddl_ok(&state, &su, "DROP TENANT 7142").await; |
| 24 | +} |
| 25 | + |
| 26 | +/// `DROP TENANT <name>` — the new path; name resolves to the catalog id. |
| 27 | +#[tokio::test] |
| 28 | +async fn drop_tenant_by_bare_name() { |
| 29 | + let state = make_state_with_catalog(); |
| 30 | + let su = superuser(); |
| 31 | + |
| 32 | + ddl_ok(&state, &su, "CREATE TENANT acme_drop_name ID 7143").await; |
| 33 | + ddl_ok(&state, &su, "DROP TENANT acme_drop_name").await; |
| 34 | +} |
| 35 | + |
| 36 | +/// `DROP TENANT '<name>'` — single-quoted name, matches the AST |
| 37 | +/// `TenantSelector` behavior on CREATE/SHOW. |
| 38 | +#[tokio::test] |
| 39 | +async fn drop_tenant_by_quoted_name() { |
| 40 | + let state = make_state_with_catalog(); |
| 41 | + let su = superuser(); |
| 42 | + |
| 43 | + ddl_ok(&state, &su, "CREATE TENANT acme_drop_quoted ID 7144").await; |
| 44 | + ddl_ok(&state, &su, "DROP TENANT 'acme_drop_quoted'").await; |
| 45 | +} |
| 46 | + |
| 47 | +/// `DROP TENANT <unknown_name>` without `IF EXISTS` errors with `42704`. |
| 48 | +#[tokio::test] |
| 49 | +async fn drop_tenant_unknown_name_without_if_exists_errors() { |
| 50 | + let state = make_state_with_catalog(); |
| 51 | + let su = superuser(); |
| 52 | + |
| 53 | + let err = ddl_err(&state, &su, "DROP TENANT no_such_tenant").await; |
| 54 | + assert!( |
| 55 | + err.contains("does not exist") && err.contains("42704"), |
| 56 | + "expected 42704/does not exist, got: {err}" |
| 57 | + ); |
| 58 | +} |
| 59 | + |
| 60 | +/// `DROP TENANT IF EXISTS <unknown_name>` is a no-op success — parallels the |
| 61 | +/// `IF EXISTS <unknown_id>` semantics. |
| 62 | +#[tokio::test] |
| 63 | +async fn drop_tenant_if_exists_unknown_name_is_noop() { |
| 64 | + let state = make_state_with_catalog(); |
| 65 | + let su = superuser(); |
| 66 | + |
| 67 | + ddl_ok(&state, &su, "DROP TENANT IF EXISTS no_such_tenant").await; |
| 68 | +} |
| 69 | + |
| 70 | +/// `DROP TENANT ''` (empty quoted name) → `42601` syntax error. |
| 71 | +#[tokio::test] |
| 72 | +async fn drop_tenant_empty_name_errors() { |
| 73 | + let state = make_state_with_catalog(); |
| 74 | + let su = superuser(); |
| 75 | + |
| 76 | + let err = ddl_err(&state, &su, "DROP TENANT ''").await; |
| 77 | + assert!( |
| 78 | + err.contains("42601") && err.contains("numeric id or a tenant name"), |
| 79 | + "expected 42601 empty-name error, got: {err}" |
| 80 | + ); |
| 81 | +} |
| 82 | + |
| 83 | +// ─── ALTER TENANT by name ──────────────────────────────────────────────────── |
| 84 | + |
| 85 | +/// `ALTER TENANT <id> SET QUOTA ...` — regression: numeric form still works. |
| 86 | +#[tokio::test] |
| 87 | +async fn alter_tenant_by_numeric_id_still_works() { |
| 88 | + let state = make_state_with_catalog(); |
| 89 | + let su = superuser(); |
| 90 | + |
| 91 | + ddl_ok(&state, &su, "CREATE TENANT acme_alter_num ID 7145").await; |
| 92 | + ddl_ok( |
| 93 | + &state, |
| 94 | + &su, |
| 95 | + "ALTER TENANT 7145 SET QUOTA max_qps = 250", |
| 96 | + ) |
| 97 | + .await; |
| 98 | +} |
| 99 | + |
| 100 | +/// `ALTER TENANT <name> SET QUOTA ...` — name resolves to id. |
| 101 | +#[tokio::test] |
| 102 | +async fn alter_tenant_by_name() { |
| 103 | + let state = make_state_with_catalog(); |
| 104 | + let su = superuser(); |
| 105 | + |
| 106 | + ddl_ok(&state, &su, "CREATE TENANT acme_alter_name ID 7146").await; |
| 107 | + ddl_ok( |
| 108 | + &state, |
| 109 | + &su, |
| 110 | + "ALTER TENANT acme_alter_name SET QUOTA max_qps = 250", |
| 111 | + ) |
| 112 | + .await; |
| 113 | +} |
| 114 | + |
| 115 | +/// `ALTER TENANT <unknown_name> SET QUOTA ...` errors with `42704`. |
| 116 | +#[tokio::test] |
| 117 | +async fn alter_tenant_unknown_name_errors() { |
| 118 | + let state = make_state_with_catalog(); |
| 119 | + let su = superuser(); |
| 120 | + |
| 121 | + let err = ddl_err( |
| 122 | + &state, |
| 123 | + &su, |
| 124 | + "ALTER TENANT no_such_tenant SET QUOTA max_qps = 250", |
| 125 | + ) |
| 126 | + .await; |
| 127 | + assert!( |
| 128 | + err.contains("does not exist") && err.contains("42704"), |
| 129 | + "expected 42704/does not exist, got: {err}" |
| 130 | + ); |
| 131 | +} |
0 commit comments