Add __init__.py for sqlalchemy integration tests

Author: jonathanl-bq

tests/integration/container/sqlalchemy/__init__.py

@@ -0,0 +1,13 @@
+#  Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+#  Licensed under the Apache License, Version 2.0 (the "License").
+#  You may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#
+#  http://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License.

tests/integration/container/sqlalchemy/test_sqlalchemy_basic.py

@@ -21,8 +21,8 @@
 from typing import Any
 
 import pytest
-from sqlalchemy.orm import declarative_base, Mapped, mapped_column
-from sqlalchemy import Column, Integer, String
+from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column
+from sqlalchemy import Column, Integer, String, Boolean, DateTime
 
 from tests.integration.container.utils.rds_test_utility import RdsTestUtility
 from ..utils.conditions import (disable_on_features, enable_on_deployments,
@@ -52,18 +52,20 @@ def rds_utils(self):
     @pytest.fixture(scope='class')
     def sqlalchemy_models(self, sqlalchemy_setup):
         """Create SQLAlchemy models after SQLAlchemy is set up"""
+	    #Base = declarative_base()
 
-	Base = declarative_base()
+    class Base(DeclarativeBase):
+        pass
 
-        class TestModel(Base):
-            """Basic test model for SQLAlchemy ORM functionality"""
-            __tablename__ = 'sqlalchemy_test_model'
+    class TestModel(Base):
+        """Basic test model for SQLAlchemy ORM functionality"""
+        __tablename__ = 'sqlalchemy_test_model'
 
-            name: Mapped[str] = mapped_column(String(100))
-            email: Mapped[str] = mapped_column(String, primary_key=True)
-            age: Mapped[int] = mapped_column(Integer)
-            is_active: Mapped[bool] = mapped_column(Bool, server_default=True)
-            created_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now(), onupdate=func.now())
+        name: Mapped[str] = mapped_column(String(100))
+        email: Mapped[str] = mapped_column(String, primary_key=True)
+        age: Mapped[int] = mapped_column(Integer)
+        is_active: Mapped[bool] = mapped_column(Boolean)
+        created_at: Mapped[datetime] = mapped_column(DateTime)
 
 '''
         class DataTypeModel(models.Model):