remote/sqlite-utils
1
0
Fork 0
mirror of https://github.com/simonw/sqlite-utils.git synced 2026-09-24 19:04:12 +02:00
Code Issues Projects Releases Packages Wiki Activity

More .transform() tests

Browse source
This commit is contained in:
Simon Willison 2020-09-21 16:58:02 -07:00
commit 59692895fd
1 changed files with 23 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

23
tests/test_transform.py
View file

@ -4,6 +4,7 @@ import pytest
@pytest.mark.parametrize( @pytest.mark.parametrize(
"params,expected_sql", "params,expected_sql",
[ [
# Identity transform - nothing changes
( (
{}, {},
[ [
@ -13,6 +14,17 @@ import pytest
"ALTER TABLE [dogs_new_suffix] RENAME TO [dogs]", "ALTER TABLE [dogs_new_suffix] RENAME TO [dogs]",
], ],
), ),
# Change column type
(
{"columns": {"age": int}},
[
"CREATE TABLE [dogs_new_suffix] (\n [id] INTEGER PRIMARY KEY,\n [name] TEXT,\n [age] INTEGER\n);",
"INSERT INTO [dogs_new_suffix] ([id], [name], [age]) SELECT [id], [name], [age] FROM [dogs]",
"DROP TABLE [dogs]",
"ALTER TABLE [dogs_new_suffix] RENAME TO [dogs]",
],
),
# Rename a column
( (
{"rename": {"age": "dog_age"}}, {"rename": {"age": "dog_age"}},
[ [
@ -22,6 +34,7 @@ import pytest
"ALTER TABLE [dogs_new_suffix] RENAME TO [dogs]", "ALTER TABLE [dogs_new_suffix] RENAME TO [dogs]",
], ],
), ),
# Drop a column
( (
{"drop": ["age"]}, {"drop": ["age"]},
[ [
@ -31,6 +44,16 @@ import pytest
"ALTER TABLE [dogs_new_suffix] RENAME TO [dogs]", "ALTER TABLE [dogs_new_suffix] RENAME TO [dogs]",
], ],
), ),
# Convert type AND rename column
(
{"columns": {"age": int}, "rename": {"age": "dog_age"}},
[
"CREATE TABLE [dogs_new_suffix] (\n [id] INTEGER PRIMARY KEY,\n [name] TEXT,\n [dog_age] INTEGER\n);",
"INSERT INTO [dogs_new_suffix] ([id], [name], [dog_age]) SELECT [id], [name], [age] FROM [dogs]",
"DROP TABLE [dogs]",
"ALTER TABLE [dogs_new_suffix] RENAME TO [dogs]",
],
),
], ],
) )
def test_transform_sql(fresh_db, params, expected_sql): def test_transform_sql(fresh_db, params, expected_sql):