From fd34f9e732ba30d74c39a6c6de0bc4b50e6a2857 Mon Sep 17 00:00:00 2001 From: IlyaFaer Date: Mon, 29 Nov 2021 11:46:04 +0300 Subject: [PATCH] test: add base class for the Spanner specific tests --- test/test_suite.py | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/test/test_suite.py b/test/test_suite.py index 0bc4030c..e969e688 100644 --- a/test/test_suite.py +++ b/test/test_suite.py @@ -1510,16 +1510,20 @@ class Address(Base): assert str(query.statement.compile(session.bind)) == EXPECTED_QUERY -class InterleavedTablesTest(fixtures.TestBase): - """ - Check that CREATE TABLE statements for interleaved tables are correctly - generated. - """ +class SpannerSpecificTestBase(fixtures.TestBase): + """Base class for the Cloud Spanner related tests.""" def setUp(self): self._engine = create_engine(get_db_url()) self._metadata = MetaData(bind=self._engine) + +class InterleavedTablesTest(SpannerSpecificTestBase): + """ + Check that CREATE TABLE statements for interleaved tables are correctly + generated. + """ + def test_interleave(self): EXP_QUERY = ( "\nCREATE TABLE client (\n\tteam_id INT64 NOT NULL, " @@ -1562,13 +1566,9 @@ def test_interleave_on_delete_cascade(self): execute.assert_called_once_with(EXP_QUERY, []) -class UserAgentTest(fixtures.TestBase): +class UserAgentTest(SpannerSpecificTestBase): """Check that SQLAlchemy dialect uses correct user agent.""" - def setUp(self): - self._engine = create_engine(get_db_url()) - self._metadata = MetaData(bind=self._engine) - def test_user_agent(self): dist = pkg_resources.get_distribution("sqlalchemy-spanner")