diff --git a/ecs/crm-datafetch/tests/test_backup_crm_csv_data_process.py b/ecs/crm-datafetch/tests/test_backup_crm_csv_data_process.py new file mode 100644 index 00000000..267e87da --- /dev/null +++ b/ecs/crm-datafetch/tests/test_backup_crm_csv_data_process.py @@ -0,0 +1,157 @@ +import textwrap +from unittest.mock import patch + +import pytest +from src.backup_crm_csv_data_process import backup_crm_csv_data_process +from src.config.objects import TargetObject +from src.error.exceptions import FileUploadException +from src.system_var.constants import CSVBK_JP_NAME +from src.util.execute_datetime import ExecuteDateTime + +from .test_utils.log_message import generate_log_message_tuple + + +class TestBackupCrmCsvDataProcess: + + @pytest.fixture + def bucket_name(self): + return 'test-backup-bucket' + + @pytest.fixture + def prepare_bucket(self, s3_client, bucket_name): + s3_client.create_bucket(Bucket=bucket_name) + yield + + def test_run_process_success(self, s3_client, prepare_bucket, bucket_name, monkeypatch, caplog): + """ + Cases: + CSVバックアップ処理が正常終了し、期待通りの結果となること + Arranges: + - prepare_bucketフィクスチャで、CRM電文データをバックアップするためのモックバケットを作る + - 作成したモックバケットを指すように環境変数を設定する + Expects: + - CSVバックアップファイルがバケットに配置される + - CSVバックアップ処理の仕様に沿った正常系ログが出力されること(デバッグログは除く) + """ + # Arrange + csv_string = textwrap.dedent("""\ + "Id","AccountNumber","LastModifiedDate","LastModifiedById","SystemModstamp","IsDeleted" + "TEST001","test001","2022-06-01 09:00:00","1234567","2022-06-01 09:00:00","1" + "TEST002","test002","2022-06-01 09:00:00","1234567","2022-06-01 09:00:00","0" + "TEST003","test003","2022-06-01 09:00:00","1234567","2022-06-01 09:00:00","0" + """) + target_object_dict = { + 'object_name': 'Account', + 'columns': [ + 'Id', + 'AccountNumber', + 'LastModifiedDate', + 'LastModifiedById', + 'SystemModstamp', + 'IsDeleted' + ] + } + + execute_datetime = ExecuteDateTime() + target_object = TargetObject(target_object_dict, execute_datetime) + + # 環境変数を編集 + monkeypatch.setattr('src.aws.s3.CRM_BACKUP_BUCKET', bucket_name) + monkeypatch.setattr('src.aws.s3.CRM_IMPORT_DATA_BACKUP_FOLDER', 'data_import') + + # Act + backup_crm_csv_data_process(target_object, execute_datetime, csv_string) + + # Assert + + # ファイル確認 + actual = s3_client.get_object( + Bucket=bucket_name, + Key=f'data_import/{execute_datetime.to_path()}/CRM_Account_{execute_datetime.format_date()}.csv') + + assert actual['Body'].read().decode('utf-8') == csv_string + + # ログの確認 + assert generate_log_message_tuple( + log_message=f'I-CSVBK-01 [Account] のCSVデータのバックアップ処理を開始します ファイル名:[CRM_Account_{execute_datetime.format_date()}.csv]' + ) in caplog.record_tuples + assert generate_log_message_tuple( + log_message='I-CSVBK-03 [Account] のCSVデータのバックアップ処理を終了します') in caplog.record_tuples + + def test_call_depended_modules(self): + """ + Cases: + CSVバックアップ処理内で依存しているモジュールが正しく呼ばれていること + Arranges: + - CSVバックアップ処理の依存モジュールをモック化する + Expects: + - 依存しているモジュールが正しく呼ばれている + """ + + # Arrange + target_object_dict = { + 'object_name': 'Account', + 'columns': [ + 'Id', + 'AccountNumber', + 'LastModifiedDate', + 'LastModifiedById', + 'SystemModstamp', + 'IsDeleted' + ] + } + + execute_datetime = ExecuteDateTime() + target_object = TargetObject(target_object_dict, execute_datetime) + + with patch('src.backup_crm_csv_data_process.BackupBucket') as mock_backup_bucket: + mock_backup_bucket_inst = mock_backup_bucket.return_value + mock_backup_bucket_inst.put_csv.return_value = '' + # Act + backup_crm_csv_data_process(target_object, execute_datetime, '') + + # Assert + + assert mock_backup_bucket_inst.put_csv.called is True + + def test_raise_put_csv(self, bucket_name, monkeypatch, caplog): + """ + Cases: + CSVデータをバックアップできない場合、エラーが発生すること + Arranges: + - CSVデータをバックアップ処理で例外が発生するようにする + Expects: + - 例外が発生する + - ファイル登録できないエラーが返却される + """ + + # Arrange + target_object_dict = { + 'object_name': 'Account', + 'columns': [ + 'Id', + 'AccountNumber', + 'LastModifiedDate', + 'LastModifiedById', + 'SystemModstamp', + 'IsDeleted' + ] + } + + execute_datetime = ExecuteDateTime() + target_object = TargetObject(target_object_dict, execute_datetime) + + with patch('src.backup_crm_csv_data_process.BackupBucket') as mock_backup_bucket: + mock_backup_bucket_inst = mock_backup_bucket.return_value + mock_backup_bucket_inst.put_csv.side_effect = Exception('登録エラー') + # Act + with pytest.raises(FileUploadException) as e: + backup_crm_csv_data_process(target_object, execute_datetime, '') + + # Assert + + assert mock_backup_bucket_inst.put_csv.called is True + assert e.value.error_id == 'E-CSVBK-01' + assert e.value.func_name == CSVBK_JP_NAME + assert e.value.args[0] == \ + f'[Account] CSVデータのバックアップに失敗しました ファイル名:[CRM_Account_{execute_datetime.format_date()}.csv] エラー内容:[登録エラー]'