fix: test fix (googleapis#310)

asthamohta · surbhigarg92 · web-flow · commit c376d422ab45 · 2023-03-20T11:10:25.000+05:30
* chore: auto-release

* changes

* changes

* changes

* changes

* changes

* Without changes

* Update noxfile.py

* changes

* changes

---------

Co-authored-by: surbhigarg92 &lt;surbhigarg.92@gmail.com&gt;
diff --git a/.github/workflows/test_suite.yml b/.github/workflows/test_suite.yml
@@ -85,6 +85,7 @@ jobs:
         env:
           SPANNER_EMULATOR_HOST: localhost:9010
           GOOGLE_CLOUD_PROJECT: appdev-soda-spanner-staging
+          SQLALCHEMY_SILENCE_UBER_WARNING: 1
 
   migration_tests:
     runs-on: ubuntu-latest
diff --git a/noxfile.py b/noxfile.py
@@ -190,10 +190,9 @@ def compliance_test_14(session):
 
     session.install("mock")
     session.install("-e", ".[tracing]")
+    session.run("pip", "install", "sqlalchemy>=1.4,<2.0", "--force-reinstall")
     session.run("python", "create_test_database.py")
 
-    session.install("sqlalchemy>=1.4")
-
     session.run(
         "py.test",
         "--cov=google.cloud.sqlalchemy_spanner",
@@ -224,7 +223,7 @@ def unit(session):
 @nox.session(python=DEFAULT_PYTHON_VERSION)
 def migration_test(session):
     """Test migrations with SQLAlchemy v1.3.11+ and Alembic"""
-    session.run("pip", "install", "sqlalchemy>=1.3.11", "--force-reinstall")
+    session.run("pip", "install", "sqlalchemy>=1.3.11,<2.0", "--force-reinstall")
     _migration_test(session)
 
 
@@ -242,6 +241,11 @@ def _migration_test(session):
     import os
     import shutil
 
+    try:
+        import sqlalchemy
+    except:
+        session.run("pip", "install", "sqlalchemy>=1.3.11,<2.0", "--force-reinstall")
+
     session.install("pytest")
     session.install("-e", ".")
     session.install("alembic")
@@ -307,7 +311,7 @@ def snippets(session):
         session.skip("Credentials must be set via environment variable.")
 
     session.install("pytest")
-    session.install("sqlalchemy")
+    session.install("sqlalchemy>=1.4,<2.0")
     session.install(
         "git+https://github.com/googleapis/python-spanner.git#egg=google-cloud-spanner"
     )
