diff options
author | Mike Bayer <mike_mp@zzzcomputing.com> | 2021-02-08 11:58:15 -0500 |
---|---|---|
committer | Mike Bayer <mike_mp@zzzcomputing.com> | 2021-02-08 13:14:52 -0500 |
commit | c0521e6f0688b794048e44ff3df429249a093b72 (patch) | |
tree | d509123bb03c6e938952c4eb6444fd00d5edbc67 /lib/sqlalchemy/testing/provision.py | |
parent | 80010c63149be411e89c7434a9d52096f9de56b8 (diff) | |
download | sqlalchemy-c0521e6f0688b794048e44ff3df429249a093b72.tar.gz |
Add identifier_preparer per-execution context for schema translates
Fixed bug where the "schema_translate_map" feature failed to be taken into
account for the use case of direct execution of
:class:`_schema.DefaultGenerator` objects such as sequences, which included
the case where they were "pre-executed" in order to generate primary key
values when implicit_returning was disabled.
Fixes: #5929
Change-Id: I3fed1d0af28be5ce9c9bb572524dcc8411633f60
Diffstat (limited to 'lib/sqlalchemy/testing/provision.py')
-rw-r--r-- | lib/sqlalchemy/testing/provision.py | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/lib/sqlalchemy/testing/provision.py b/lib/sqlalchemy/testing/provision.py index 2fade1c32..a976abee0 100644 --- a/lib/sqlalchemy/testing/provision.py +++ b/lib/sqlalchemy/testing/provision.py @@ -262,7 +262,6 @@ def drop_all_schema_objects(cfg, eng): ) util.drop_all_tables(eng, inspector) - if config.requirements.schemas.enabled_for_config(cfg): util.drop_all_tables(eng, inspector, schema=cfg.test_schema) util.drop_all_tables(eng, inspector, schema=cfg.test_schema_2) @@ -273,6 +272,16 @@ def drop_all_schema_objects(cfg, eng): with eng.begin() as conn: for seq in inspector.get_sequence_names(): conn.execute(ddl.DropSequence(schema.Sequence(seq))) + if config.requirements.schemas.enabled_for_config(cfg): + for schema_name in [cfg.test_schema, cfg.test_schema_2]: + for seq in inspector.get_sequence_names( + schema=schema_name + ): + conn.execute( + ddl.DropSequence( + schema.Sequence(seq, schema=schema_name) + ) + ) @register.init |