Coverage for tests / test_advanced / test_datetime / test_tutorial004.py: 100%

1import importlib 9 ctx1abcdefghi

2from io import StringIO 9 ctx1abcdefghi

3from types import ModuleType 9 ctx1abcdefghi

4 

5import pytest 9 ctx1abcdefghi

6from alembic.migration import MigrationContext 9 ctx1abcdefghi

7from alembic.operations import Operations 9 ctx1abcdefghi

8 

9 

10@pytest.fixture( 9 ctx1abcdefghi

11 name="mod", 

12 params=[ 

13 pytest.param("tutorial004_py310"), 

14 ], 

15) 

16def get_module(request: pytest.FixtureRequest) -> ModuleType: 9 ctx1abcdefghi

17 return importlib.import_module(f"docs_src.advanced.datetime.{request.param}") 9 ctx1abcdefghi

18 

19 

20def test_tutorial(mod: ModuleType) -> None: 9 ctx1abcdefghi

21 output = StringIO() 9 ctx1jklmnopqr

22 context = MigrationContext.configure( 9 ctx1jklmnopqr

23 dialect_name="postgresql", 

24 opts={"as_sql": True, "output_buffer": output}, 

25 ) 

26 with Operations.context(context): 9 ctx1jklmnopqr

27 mod.upgrade() 9 ctx1jklmnopqr

28 mod.downgrade() 9 ctx1jklmnopqr

29 

30 assert output.getvalue() == ( 9 ctx1jklmnopqr

31 "ALTER TABLE event ALTER COLUMN created_at TYPE TIMESTAMP WITH TIME ZONE " 

32 "USING created_at AT TIME ZONE 'UTC';\n\n" 

33 "ALTER TABLE event ALTER COLUMN created_at TYPE TIMESTAMP WITHOUT TIME ZONE " 

34 "USING created_at AT TIME ZONE 'UTC';\n\n" 

35 )