wehub-resource-sync
bf2343b7e4
Integration Tests - MySQL + Elasticsearch / Detect Changes (push) Has been cancelled
Integration Tests - MySQL + Elasticsearch / integration-tests-mysql-elasticsearch (push) Has been cancelled
Integration Tests - PostgreSQL + Elasticsearch + Redis / Detect Changes (push) Has been cancelled
Integration Tests - PostgreSQL + Elasticsearch + Redis / integration-tests-postgres-elasticsearch-redis (push) Has been cancelled
Integration Tests - PostgreSQL + OpenSearch / Detect Changes (push) Has been cancelled
Integration Tests - PostgreSQL + OpenSearch / integration-tests-postgres-opensearch (push) Has been cancelled
Java Checkstyle / java-checkstyle (push) Has been cancelled
Maven Collate Tests / maven-collate-ci (push) Has been cancelled
OpenMetadata Service Unit Tests / openmetadata-service-unit-tests-status (push) Has been cancelled
Publish Package to Maven Central Repository / publish-maven-packages (push) Has been cancelled
OpenMetadata Service Unit Tests / Detect Changes (push) Has been cancelled
OpenMetadata Service Unit Tests / openmetadata-service-unit-tests (push) Has been cancelled
OpenMetadata Service Unit Tests / k8s_operator-unit-tests (push) Has been cancelled
61 lines
1.8 KiB
Python
61 lines
1.8 KiB
Python
"""SQLAlchemy utilities for testing purposes."""
|
|
|
|
from typing import Sequence # noqa: UP035
|
|
|
|
from sqlalchemy import Column, Integer, String, create_engine
|
|
from sqlalchemy.orm import DeclarativeBase, Session
|
|
|
|
|
|
class Base(DeclarativeBase):
|
|
pass
|
|
|
|
|
|
class User(Base):
|
|
__tablename__ = "users"
|
|
id = Column(Integer, primary_key=True)
|
|
name = Column(String(256))
|
|
fullname = Column(String(256))
|
|
nickname = Column(String(256))
|
|
age = Column(Integer)
|
|
|
|
|
|
class UserWithLongName(Base):
|
|
__tablename__ = "u" * 63 # Keep a max length name of 63 chars (max for Postgres)
|
|
id = Column(Integer, primary_key=True)
|
|
name = Column(String(256))
|
|
fullname = Column(String(256))
|
|
nickname = Column(String(256))
|
|
age = Column(Integer)
|
|
|
|
|
|
class SQATestUtils:
|
|
def __init__(self, connection_url: str):
|
|
self.connection_url = connection_url
|
|
self.engine = create_engine(connection_url)
|
|
self.session = Session(self.engine)
|
|
|
|
def load_data(self, data: Sequence[DeclarativeBase]):
|
|
"""load data into the database using sqlalchemy ORM
|
|
|
|
Args:
|
|
data List[DeclarativeBase]: list of ORM objects to load
|
|
"""
|
|
self.session.add_all(data)
|
|
self.session.commit()
|
|
|
|
def load_user_data(self):
|
|
for clz in (User, UserWithLongName):
|
|
data = [
|
|
clz(name="John", fullname="John Doe", nickname="johnny b goode", age=30), # type: ignore
|
|
clz(name="Jane", fullname="Jone Doe", nickname=None, age=31), # type: ignore
|
|
] * 20
|
|
self.load_data(data)
|
|
|
|
def create_user_table(self):
|
|
User.__table__.create(bind=self.session.get_bind())
|
|
UserWithLongName.__table__.create(bind=self.session.get_bind())
|
|
|
|
def close(self):
|
|
self.session.close()
|
|
self.engine.dispose()
|