feat: 件数取得時の接続タイムアウトのテスト追加

2022-08-17 13:05:29 +09:00 · 2022-08-17 13:05:29 +09:00 · 9504b8ce50
commit 9504b8ce50
parent ff6b41b18d
1 changed files with 102 additions and 0 deletions
--- a/ecs/crm-datafetch/tests/test_fetch_crm_data_process.py
+++ b/ecs/crm-datafetch/tests/test_fetch_crm_data_process.py
@ -306,3 +306,105 @@ class TestFetchCrmDataProcess:
        assert e.value.func_name == FETCH_JP_NAME
        # リトライ例外のオブジェクトIDが違うため、in句で比較
        assert f'[Account] の件数取得に失敗しました エラー内容：[RetryError' in e.value.args[0]
+
+    def test_raise_fetch_sf_count_read_timeout_with_retry_success(self, monkeypatch, caplog):
+        """
+        Cases:
+            データ件数取得処理で読み取り例外が発生した場合、リトライした結果復旧し、正常終了すること
+        Arranges:
+            - データ件数取得処理の最大リトライ試行回数を3に設定する
+            - データ件数取得処理の初回に読み取り例外が発生するようにする
+        Expects:
+            - 正常終了する
+            - データ件数取得に失敗した胸のエラーが出力されない
+        """
+
+        monkeypatch.setattr('src.fetch_crm_data_process.CRM_GET_RECORD_COUNT_MAX_RETRY_ATTEMPT', 3)
+        monkeypatch.setattr('src.fetch_crm_data_process.CRM_GET_RECORD_COUNT_RETRY_MAX_INTERVAL', 1)
+        monkeypatch.setattr('src.fetch_crm_data_process.CRM_GET_RECORD_COUNT_RETRY_MIN_INTERVAL', 1)
+        monkeypatch.setattr('src.fetch_crm_data_process.CRM_GET_RECORD_COUNT_RETRY_INTERVAL', 1)
+        monkeypatch.setattr('src.fetch_crm_data_process.CRM_GET_RECORD_COUNT_TIMEOUT', 1)
+        # Arrange
+        with patch('src.fetch_crm_data_process.CounterObject', ) as mock_counter, \
+                patch('src.fetch_crm_data_process.SOQLBuilder') as mock_soql_builder, \
+                patch('src.fetch_crm_data_process.SalesforceApiClient') as mock_api_client:
+            # モック化
+            mock_counter_inst = mock_counter.return_value
+            mock_counter_inst.describe.side_effect = [1, 2, 3]
+            mock_counter_inst.increment.side_effect = [2, 3, 4]
+            mock_builder_inst = mock_soql_builder.return_value
+            mock_builder_inst.create_count_soql.return_value = ''
+            mock_builder_inst.create_fetch_soql.return_value = ''
+            mock_client_inst = mock_api_client.return_value
+            mock_client_inst.fetch_sf_count.side_effect = [ReadTimeout('読み取りタイムアウト'), len(common_expect)]
+            mock_client_inst.fetch_sf_data.return_value = common_expect
+            # Act
+            fetch_crm_data_process(common_target_object, common_last_fetch_datetime)
+
+        # Assert
+
+        assert mock_counter_inst.describe.call_count == 1
+        assert mock_counter_inst.increment.call_count == 1
+
+        retry_log_message = 'W-FETCH-02 [Account] の件数取得処理がタイムアウトしたため、リトライします：[1] エラー内容：[読み取りタイムアウト]'
+        assert generate_log_message_tuple(
+            log_level=logging.WARNING,
+            log_message=retry_log_message) in caplog.record_tuples
+        called_log_counts = len([log for log in caplog.messages if log == retry_log_message])
+        assert called_log_counts == 1
+        assert generate_log_message_tuple(log_message='I-FETCH-06 [Account] のCRMからのデータ取得処理を終了します') in caplog.record_tuples
+
+    def test_raise_fetch_sf_count_read_timeout_with_retry_fail(self, monkeypatch, caplog):
+        """
+        Cases:
+            データ件数取得処理で読み取り例外が発生した場合、リトライした結果復旧せず、異常終了すること
+        Arranges:
+            - データ件数取得処理の最大リトライ試行回数を3に設定する
+            - データ件数取得処理の1回目、2回目、3回目で読み取り例外が発生するようにする
+        Expects:
+            - 正常終了する
+            - データ件数取得に失敗した胸のエラーが出力されない
+        """
+
+        monkeypatch.setattr('src.fetch_crm_data_process.CRM_GET_RECORD_COUNT_MAX_RETRY_ATTEMPT', 3)
+        monkeypatch.setattr('src.fetch_crm_data_process.CRM_GET_RECORD_COUNT_RETRY_MAX_INTERVAL', 1)
+        monkeypatch.setattr('src.fetch_crm_data_process.CRM_GET_RECORD_COUNT_RETRY_MIN_INTERVAL', 1)
+        monkeypatch.setattr('src.fetch_crm_data_process.CRM_GET_RECORD_COUNT_RETRY_INTERVAL', 1)
+        monkeypatch.setattr('src.fetch_crm_data_process.CRM_GET_RECORD_COUNT_TIMEOUT', 1)
+        # Arrange
+        with patch('src.fetch_crm_data_process.CounterObject', ) as mock_counter, \
+                patch('src.fetch_crm_data_process.SOQLBuilder') as mock_soql_builder, \
+                patch('src.fetch_crm_data_process.SalesforceApiClient') as mock_api_client:
+            # モック化
+            mock_counter_inst = mock_counter.return_value
+            mock_counter_inst.describe.side_effect = [1, 2, 3]
+            mock_counter_inst.increment.side_effect = [2, 3, 4]
+            mock_builder_inst = mock_soql_builder.return_value
+            mock_builder_inst.create_count_soql.return_value = ''
+            mock_builder_inst.create_fetch_soql.return_value = ''
+            mock_client_inst = mock_api_client.return_value
+            mock_client_inst.fetch_sf_count.side_effect = [ReadTimeout('読み取りタイムアウト'), ReadTimeout('読み取りタイムアウト'), ReadTimeout('読み取りタイムアウト')]
+            mock_client_inst.fetch_sf_data.return_value = common_expect
+            # Act
+            with pytest.raises(SalesforceAPIException) as e:
+                fetch_crm_data_process(common_target_object, common_last_fetch_datetime)
+
+        # Assert
+
+        # 取得は3回行われる
+        assert mock_counter_inst.describe.call_count == 3
+        # 足し込みは2回のみ
+        assert mock_counter_inst.increment.call_count == 2
+
+        retry_log_message = 'W-FETCH-02 [Account] の件数取得処理がタイムアウトしたため、リトライします：[1] エラー内容：[読み取りタイムアウト]'
+        assert generate_log_message_tuple(
+            log_level=logging.WARNING,
+            log_message=retry_log_message) in caplog.record_tuples
+        called_log_counts = len([log for log in caplog.messages if log == retry_log_message])
+        assert called_log_counts == 2
+        assert generate_log_message_tuple(log_message='I-FETCH-06 [Account] のCRMからのデータ取得処理を終了します') not in caplog.record_tuples
+
+        assert e.value.error_id == 'E-FETCH-01'
+        assert e.value.func_name == FETCH_JP_NAME
+        # リトライ例外のオブジェクトIDが違うため、in句で比較
+        assert f'[Account] の件数取得に失敗しました エラー内容：[RetryError' in e.value.args[0]