add retries to bulk upsert in scenario tests (#19355)

Author: swalrus1

ydb/tests/olap/scenario/helpers/scenario_tests_helper.py

@@ -363,13 +363,18 @@ def _bulk_upsert_impl(
         self, tablename: str, data_generator: ScenarioTestHelper.IDataGenerator, expected_status: ydb.StatusCode | Set[ydb.StatusCode]
     ):
         fullpath = self.get_full_path(tablename)
+        expect_success = (expected_status == ydb.StatusCode.SUCCESS)
+
+        def _call_upsert(data):
+            YdbCluster.get_ydb_driver().table_client.bulk_upsert(fullpath, data, data_generator.get_bulk_upsert_columns())
 
         def _upsert():
             data = data_generator.generate_data_portion(1000)
             allure.attach(repr(data), 'data', allure.attachment_type.TEXT)
-            YdbCluster.get_ydb_driver().table_client.bulk_upsert(
-                fullpath, data, data_generator.get_bulk_upsert_columns()
-            )
+            if expect_success:
+                ydb.retry_operation_sync(lambda: _call_upsert(data))
+            else:
+                _call_upsert(data)
 
         while not data_generator.EOF():
             self._run_with_expected_status(