From d1cf6134617cc374fa3aee2b374dc1387c9bc868 Mon Sep 17 00:00:00 2001
From: Sandro Tosi <sandrotosi@users.noreply.github.com>
Date: Mon, 23 Oct 2023 07:22:10 -0400
Subject: [PATCH] =?UTF-8?q?=E2=AC=86=EF=B8=8F=20Upgrade=20support=20for=20?=
 =?UTF-8?q?SQLAlchemy=201.4.49,=20update=20tests=20(#519)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Co-authored-by: Sebastián Ramírez <tiangolo@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
---
 pyproject.toml                                     |  2 +-
 .../test_multiple_models/test_tutorial001.py       | 14 ++++++++++++--
 .../test_multiple_models/test_tutorial002.py       | 14 ++++++++++++--
 .../test_tutorial/test_indexes/test_tutorial001.py | 14 ++++++++++++--
 .../test_tutorial/test_indexes/test_tutorial006.py | 14 ++++++++++++--
 5 files changed, 49 insertions(+), 9 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index cd59bd4cf4..94fe55d2e0 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -31,7 +31,7 @@ classifiers = [
 
 [tool.poetry.dependencies]
 python = "^3.7"
-SQLAlchemy = ">=1.4.29,<=1.4.41"
+SQLAlchemy = ">=1.4.29,<2.0.0"
 pydantic = "^1.8.2"
 sqlalchemy2-stubs = {version = "*", allow-prereleases = true}
 
diff --git a/tests/test_tutorial/test_fastapi/test_multiple_models/test_tutorial001.py b/tests/test_tutorial/test_fastapi/test_multiple_models/test_tutorial001.py
index 8ad038f98a..8d99cf9f5b 100644
--- a/tests/test_tutorial/test_fastapi/test_multiple_models/test_tutorial001.py
+++ b/tests/test_tutorial/test_fastapi/test_multiple_models/test_tutorial001.py
@@ -172,8 +172,18 @@ def test_tutorial(clear_sqlmodel):
     insp: Inspector = inspect(mod.engine)
     indexes = insp.get_indexes(str(mod.Hero.__tablename__))
     expected_indexes = [
-        {"name": "ix_hero_name", "column_names": ["name"], "unique": 0},
-        {"name": "ix_hero_age", "column_names": ["age"], "unique": 0},
+        {
+            "name": "ix_hero_name",
+            "dialect_options": {},
+            "column_names": ["name"],
+            "unique": 0,
+        },
+        {
+            "name": "ix_hero_age",
+            "dialect_options": {},
+            "column_names": ["age"],
+            "unique": 0,
+        },
     ]
     for index in expected_indexes:
         assert index in indexes, "This expected index should be in the indexes in DB"
diff --git a/tests/test_tutorial/test_fastapi/test_multiple_models/test_tutorial002.py b/tests/test_tutorial/test_fastapi/test_multiple_models/test_tutorial002.py
index 9fd328238a..94a41b3076 100644
--- a/tests/test_tutorial/test_fastapi/test_multiple_models/test_tutorial002.py
+++ b/tests/test_tutorial/test_fastapi/test_multiple_models/test_tutorial002.py
@@ -172,8 +172,18 @@ def test_tutorial(clear_sqlmodel):
     insp: Inspector = inspect(mod.engine)
     indexes = insp.get_indexes(str(mod.Hero.__tablename__))
     expected_indexes = [
-        {"name": "ix_hero_age", "column_names": ["age"], "unique": 0},
-        {"name": "ix_hero_name", "column_names": ["name"], "unique": 0},
+        {
+            "name": "ix_hero_age",
+            "dialect_options": {},
+            "column_names": ["age"],
+            "unique": 0,
+        },
+        {
+            "name": "ix_hero_name",
+            "dialect_options": {},
+            "column_names": ["name"],
+            "unique": 0,
+        },
     ]
     for index in expected_indexes:
         assert index in indexes, "This expected index should be in the indexes in DB"
diff --git a/tests/test_tutorial/test_indexes/test_tutorial001.py b/tests/test_tutorial/test_indexes/test_tutorial001.py
index 596207737d..f33db5bcc7 100644
--- a/tests/test_tutorial/test_indexes/test_tutorial001.py
+++ b/tests/test_tutorial/test_indexes/test_tutorial001.py
@@ -25,8 +25,18 @@ def test_tutorial(clear_sqlmodel):
     insp: Inspector = inspect(mod.engine)
     indexes = insp.get_indexes(str(mod.Hero.__tablename__))
     expected_indexes = [
-        {"name": "ix_hero_name", "column_names": ["name"], "unique": 0},
-        {"name": "ix_hero_age", "column_names": ["age"], "unique": 0},
+        {
+            "name": "ix_hero_name",
+            "dialect_options": {},
+            "column_names": ["name"],
+            "unique": 0,
+        },
+        {
+            "name": "ix_hero_age",
+            "dialect_options": {},
+            "column_names": ["age"],
+            "unique": 0,
+        },
     ]
     for index in expected_indexes:
         assert index in indexes, "This expected index should be in the indexes in DB"
diff --git a/tests/test_tutorial/test_indexes/test_tutorial006.py b/tests/test_tutorial/test_indexes/test_tutorial006.py
index e26f8b2ed8..893043dad1 100644
--- a/tests/test_tutorial/test_indexes/test_tutorial006.py
+++ b/tests/test_tutorial/test_indexes/test_tutorial006.py
@@ -26,8 +26,18 @@ def test_tutorial(clear_sqlmodel):
     insp: Inspector = inspect(mod.engine)
     indexes = insp.get_indexes(str(mod.Hero.__tablename__))
     expected_indexes = [
-        {"name": "ix_hero_name", "column_names": ["name"], "unique": 0},
-        {"name": "ix_hero_age", "column_names": ["age"], "unique": 0},
+        {
+            "name": "ix_hero_name",
+            "dialect_options": {},
+            "column_names": ["name"],
+            "unique": 0,
+        },
+        {
+            "name": "ix_hero_age",
+            "dialect_options": {},
+            "column_names": ["age"],
+            "unique": 0,
+        },
     ]
     for index in expected_indexes:
         assert index in indexes, "This expected index should be in the indexes in DB"