103 lines
3.4 KiB
Python
103 lines
3.4 KiB
Python
"""テスト用共通処理関数"""
|
|
|
|
import io
|
|
from datetime import datetime
|
|
from types import ModuleType
|
|
|
|
from src.batch.ultmarc.datfile import DatFile
|
|
|
|
|
|
def create_ultmarc_test_csv(*csv_rows: str) -> DatFile:
|
|
"""アルトマーク取込テストのCSVを作成
|
|
Args:
|
|
csv_rows (tuple[str]): CSV文字列のリスト
|
|
Returns:
|
|
list[list[str]]: CSVデータ
|
|
"""
|
|
string_io = io.StringIO()
|
|
string_io.writelines(csv_rows)
|
|
string_io.seek(0)
|
|
dat_file = DatFile(string_io)
|
|
|
|
return dat_file
|
|
|
|
|
|
def create_ultmarc_common_column_names() -> list[str]:
|
|
"""アルトマークテーブル共通のカラム名を作成
|
|
Returns:
|
|
list[str]: 共通カラム名
|
|
"""
|
|
|
|
return [
|
|
'regist_ymd',
|
|
'update_ymd',
|
|
'delete_ymd',
|
|
'regist_date',
|
|
'create_user',
|
|
'update_date',
|
|
'update_user',
|
|
'sys_regist_date',
|
|
'regist_prgm_id',
|
|
'sys_update_date',
|
|
'update_prgm_id'
|
|
]
|
|
|
|
|
|
def create_ultmarc_common_column_values(expect_datetime: datetime, expect_date_str: str, module_name: str) -> list:
|
|
"""アルトマークテーブル共通のカラムを作成
|
|
|
|
Args:
|
|
expect_datetime (datetime): テスト実行年月日時分秒
|
|
expect_date_str (str): テスト実行年月日の文字列
|
|
module_name (str): モジュール名
|
|
|
|
Returns:
|
|
list: 共通カラム値
|
|
"""
|
|
|
|
return [
|
|
expect_date_str, # 登録年月日(regist_ymd)
|
|
None, # 更新年月日(update_ymd)
|
|
None, # 削除年月日(delete_ymd)
|
|
None, # 登録日時(regist_date)
|
|
None, # 登録者(create_user)
|
|
None, # 更新日時(update_date)
|
|
None, # 更新者(update_user)
|
|
expect_datetime, # システム登録日時(sys_regist_date)
|
|
module_name, # 登録プログラムid(regist_prgm_id)
|
|
expect_datetime, # システム更新日時(sys_update_date)
|
|
module_name # 更新プログラムid(update_prgm_id)
|
|
]
|
|
|
|
|
|
def get_module_name(module: ModuleType) -> str:
|
|
"""登録プログラムID、更新プログラムIDに登録するモジュール名を作成
|
|
|
|
Args:
|
|
module (ModuleType): pythonモジュール
|
|
|
|
Returns:
|
|
str: モジュール名
|
|
"""
|
|
return module.__name__.split('.')[-1]
|
|
|
|
|
|
def assert_table_results(actual_rows: list[dict], expect_rows: list[dict]) -> None:
|
|
"""テーブル同士の取得結果突き合わせ
|
|
|
|
Args:
|
|
actual_rows (list[dict]): テスト結果の辞書リスト
|
|
expect_rows (list[dict]): 期待値の辞書リスト
|
|
"""
|
|
# 取得件数が一致すること
|
|
assert len(actual_rows) == len(expect_rows)
|
|
|
|
# 1カラムずつ調査
|
|
for actual_row, expect_row in zip(actual_rows, expect_rows):
|
|
for actual_col_name, expect_col_name in zip(actual_row, expect_row):
|
|
# システム日付が設定されるカラムは、期待値以前になっていること
|
|
if actual_col_name in ['sys_regist_date', 'sys_update_date']:
|
|
assert actual_row[actual_col_name] < expect_row[expect_col_name], f'{actual_col_name}が、期待値の日時以前であること'
|
|
else:
|
|
assert actual_row[actual_col_name] == expect_row[expect_col_name], f'{actual_col_name}が、期待値と一致すること'
|