mirror of
https://github.com/open-metadata/OpenMetadata.git
synced 2025-07-10 02:28:22 +00:00
29cd58b628
* adventure works mssql test case * adventure works mssql test case * fixed tests * fixed tests * fixed tests * fixed tests
49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
import os
|
|
import shutil
|
|
|
|
import pytest
|
|
from testcontainers.mssql import SqlServerContainer
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def sql_server_container(tmp_path_factory):
|
|
msqql_container = SqlServerContainer("mcr.microsoft.com/mssql/server:2017-latest")
|
|
data_dir = tmp_path_factory.mktemp("data")
|
|
shutil.copy(
|
|
os.path.join(os.path.dirname(__file__), "data", "AdventureWorks2017.bak"),
|
|
str(data_dir),
|
|
)
|
|
with open(data_dir / "install.sql", "w") as f:
|
|
f.write(
|
|
"""
|
|
USE [master]
|
|
RESTORE DATABASE [AdventureWorks]
|
|
FROM DISK = '/data/AdventureWorks2017.bak'
|
|
WITH MOVE 'AdventureWorks2017' TO '/var/opt/mssql/data/AdventureWorks.mdf',
|
|
MOVE 'AdventureWorks2017_log' TO '/var/opt/mssql/data/AdventureWorks_log.ldf'
|
|
GO
|
|
"""
|
|
)
|
|
|
|
msqql_container.volumes = {str(data_dir): {"bind": "/data"}}
|
|
with msqql_container as container:
|
|
docker_container = container.get_wrapped_container()
|
|
res = docker_container.exec_run(
|
|
[
|
|
"/opt/mssql-tools/bin/sqlcmd",
|
|
"-S",
|
|
"localhost",
|
|
"-U",
|
|
container.username,
|
|
"-P",
|
|
container.password,
|
|
"-d",
|
|
"master",
|
|
"-i",
|
|
"/data/install.sql",
|
|
]
|
|
)
|
|
if res[0] != 0:
|
|
raise Exception("Failed to create mssql database:" + res[1].decode("utf-8"))
|
|
yield container
|