12
12
import pytest
13
13
import sqlalchemy as sqla
14
14
from sqlalchemy .sql import and_ , not_ , or_
15
- from sqlalchemy_utils import create_view
16
15
17
16
from tests .unit .conftest import sqlalchemy_version
18
17
from trino .sqlalchemy .datatype import JSON
@@ -382,24 +381,17 @@ def test_get_table_names(trino_connection, schema):
382
381
engine .execute (sqla .schema .CreateSchema (name ))
383
382
384
383
try :
385
- create_view (
386
- 'my_view' ,
387
- sqla .select (
388
- [
389
- sqla .Table (
390
- 'my_table' ,
391
- metadata ,
392
- sqla .Column ('id' , sqla .Integer ),
393
- ),
394
- ],
395
- ),
384
+ sqla .Table (
385
+ 'test_get_table_names' ,
396
386
metadata ,
397
- cascade_on_drop = False ,
387
+ sqla . Column ( 'id' , sqla . Integer ) ,
398
388
)
399
-
400
389
metadata .create_all (engine )
401
- assert sqla .inspect (engine ).get_table_names (schema ) == ['my_table' ]
390
+ view_name = name + ".test_view"
391
+ conn .execute (f"CREATE VIEW { view_name } AS SELECT * FROM test_get_table_names" )
392
+ assert sqla .inspect (engine ).get_table_names (name ) == ['test_get_table_names' ]
402
393
finally :
394
+ conn .execute (f"DROP VIEW IF EXISTS { view_name } " )
403
395
metadata .drop_all (engine )
404
396
405
397
@@ -422,24 +414,17 @@ def test_get_view_names(trino_connection, schema):
422
414
engine .execute (sqla .schema .CreateSchema (name ))
423
415
424
416
try :
425
- create_view (
426
- 'test_get_view_names' ,
427
- sqla .select (
428
- [
429
- sqla .Table (
430
- 'my_table' ,
431
- metadata ,
432
- sqla .Column ('id' , sqla .Integer ),
433
- ),
434
- ],
435
- ),
417
+ sqla .Table (
418
+ 'test_table' ,
436
419
metadata ,
437
- cascade_on_drop = False ,
420
+ sqla . Column ( 'id' , sqla . Integer ) ,
438
421
)
439
-
440
422
metadata .create_all (engine )
441
- assert sqla .inspect (engine ).get_view_names (schema ) == ['test_get_view_names' ]
423
+ view_name = name + ".test_get_view_names"
424
+ conn .execute (f"CREATE VIEW { view_name } AS SELECT * FROM test_table" )
425
+ assert sqla .inspect (engine ).get_view_names (name ) == ['test_get_view_names' ]
442
426
finally :
427
+ conn .execute (f"DROP VIEW IF EXISTS { view_name } " )
443
428
metadata .drop_all (engine )
444
429
445
430
0 commit comments