icsDataValidation 1.0.439__py3-none-any.whl → 1.0.441__py3-none-any.whl

icsDataValidation/services/database_services/snowflake_service.py

@@ -524,7 +524,7 @@ class SnowflakeService:
 
             if 'Arithmetic overflow' in err_msg[2]:
                 # re-calculate queries with bool_cast_before_sum=True in case of error
-                query_checksums = self.create_checksum_statement(
+                query_checksums = self._get_checksum_statement(
                     object=object,
                     column_intersections=column_intersections,
                     where_clause=where_clause,

icsDataValidation/services/database_services/sqlserver_service.py

@@ -587,7 +587,7 @@ class SQLServerService:
 
             if 'Arithmetic overflow' in err_msg[2]:
                 # re-calculate queries with bool_cast_before_sum=True in case of error
-                query_checksums = self.create_checksum_statement(
+                query_checksums = self._get_checksum_statement(
                     object=object,
                     column_intersections=column_intersections,
                     where_clause=where_clause,

{icsdatavalidation-1.0.439.dist-info → icsdatavalidation-1.0.441.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: icsDataValidation
-Version: 1.0.439
+Version: 1.0.441
 Summary: Add your description here
 Author-email: initions <ICSMC_EXT_PYPIORG@accenture.com>
 License: MIT

{icsdatavalidation-1.0.439.dist-info → icsdatavalidation-1.0.441.dist-info}/RECORD

@@ -27,8 +27,8 @@ icsDataValidation/services/database_services/databricks_hive_metastore_service.p
 icsDataValidation/services/database_services/databricks_unity_catalog_service.py,sha256=8iV75kvtQsGPdC35m89jO5s0ZQDekPdRVPYGbdCAPVI,70835
 icsDataValidation/services/database_services/exasol_service.py,sha256=LdjU8mM77zTmNmhJPQrgQO-HwAZv0C0seYMDjuWU9BQ,11153
 icsDataValidation/services/database_services/oracle_service.py,sha256=Ejxi0HBRF_c0xWY4wEsw8L8Rb5FMRf9cjQbhz8kerIA,31805
-icsDataValidation/services/database_services/snowflake_service.py,sha256=hygxlqvLw-6PLJFoATsp5zHup4vcuGAOvEAzgkLBkXw,68657
-icsDataValidation/services/database_services/sqlserver_service.py,sha256=6FD6vp8K3bMkfQSaCLuPEtR3KMc0IsvVAybRDwsg3Po,43303
+icsDataValidation/services/database_services/snowflake_service.py,sha256=8uJm8psd3A_SHfu4Ci3GP0bpWj6ctKACoknJuqwiBaE,68655
+icsDataValidation/services/database_services/sqlserver_service.py,sha256=cF2z4QzUz20C_D2bpL3gOX5atgPWqGjMt8DVg6QVPog,43301
 icsDataValidation/services/database_services/teradata_service.py,sha256=2x7onntG5E1qqw65HXUmFwcrYmT5I8HSS3eWXIhTfiw,40252
 icsDataValidation/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 icsDataValidation/utils/file_util.py,sha256=ZTMB1sTnIIdffg9tEJRCFQQ5SG8Fksc5ie1PM4gHXG4,3432
@@ -36,21 +36,21 @@ icsDataValidation/utils/logger_util.py,sha256=xS48_FFMot_hyQgJY8DUeRTn5jpdvRt5QI
 icsDataValidation/utils/pandas_util.py,sha256=D_g7Xw7BIS2E-1ZhJIvp62K5xuKjIkj-7TxH4HN_8SI,6505
 icsDataValidation/utils/parallelization_util.py,sha256=6P0YcQLmunW_fHR4f5-kdncZbOlxxqKyk6ZAFQQEd2k,2088
 icsDataValidation/utils/sql_util.py,sha256=0c-BInElSsRmXUedfLP_h9Wsiscv9aic7IIc5f15Uzo,396
-tests/snowflake_service/test_create_checksums.py,sha256=ifXxuNk7RHg5haTznBDsnJM9lrxXyBssVICn_oWkgj8,5397
+tests/snowflake_service/test_create_checksums.py,sha256=XE3VU11LPiMS_NG-cxNSpWwR1Ett5i0GTlzehB3Md4o,5376
 tests/snowflake_service/test_create_pandas_df_from_group_by.py,sha256=7ZF-RbloV6kZTirWdBRNkpfhNeER7vOQE83BGILbXHk,17013
 tests/snowflake_service/test_create_pandas_df_from_sample.py,sha256=z3-G_cctDttL7SR8eUep421xGBo54t3duWjeHpzt7LI,16794
 tests/snowflake_service/test_get_checksum_statement.py,sha256=U9IZqjr5YzwGKcVU6Q4h1l1G-CbwX0STQeFdEvOqV0c,8313
 tests/snowflake_service/test_get_column_clause.py,sha256=5ZIMvElXWfnjBUrYaEMDFzIpEcMO4oXJEoO82QObTz0,11019
 tests/snowflake_service/test_get_countnulls_statement.py,sha256=l1hah4oVKp983IdWUfO08ojgmgy6hmfVopn6xW1cXu4,4254
 tests/snowflake_service/test_get_in_clause.py,sha256=z0IkJhh1XSQ6rIB0VL1RBc616i1BvXjzJqZ6_ijfToU,2442
-tests/sqlserver_service/test_create_checksums.py,sha256=aknl9JzfDCHomIYlNcnp-zNYoinJcZ-rgTGUJoVC4Zs,5557
+tests/sqlserver_service/test_create_checksums.py,sha256=7wRzuM6Gd8MWNjjjjVSYWlKcBuSVu7LLfBHHm4ZFyxY,5482
 tests/sqlserver_service/test_create_pandas_df_from_group_by.py,sha256=hFZCyHep3jiVdrFGQF3eWugXK2tCv75E-WkGoU_-JsY,15071
 tests/sqlserver_service/test_create_pandas_df_from_sample.py,sha256=d4XG4O6JcXIW1JlA2E9hJw_FXlH6zRkQP_K6lUWUOl0,10818
 tests/sqlserver_service/test_get_checksum_statement.py,sha256=IZq1lwS5IEoBqNcioLq1w8mjRuCTq6Uwc27xD9OD8PA,5422
 tests/sqlserver_service/test_get_column_clause.py,sha256=JKgglx_KJTbPeVfmSMdbl2wiSwu3_R6fZs7a36LlO0Q,6232
 tests/sqlserver_service/test_get_countnulls_statement.py,sha256=ZwowVsdAuYAMeGgTk4puJMDDXlETJngTygeesqsKv7w,4003
 tests/sqlserver_service/test_get_in_clause.py,sha256=Ee4kAZdbxMQ_evlJscV8DwKtjbuEeRYUt2PaxZHLoXA,3160
-icsdatavalidation-1.0.439.dist-info/METADATA,sha256=afSxru9XnlnPI6SzukmWri68IQ95ExYuN6_s9VzxT-Q,661
-icsdatavalidation-1.0.439.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-icsdatavalidation-1.0.439.dist-info/top_level.txt,sha256=y0PjCpmJ_Vhc0QB0SgXxxcRSR7__mQV5rmFyfQc60nA,24
-icsdatavalidation-1.0.439.dist-info/RECORD,,
+icsdatavalidation-1.0.441.dist-info/METADATA,sha256=DD89djkuvSY3-pCVqdgFPDrUgb80PeW4ShLoFZRKHwI,661
+icsdatavalidation-1.0.441.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+icsdatavalidation-1.0.441.dist-info/top_level.txt,sha256=y0PjCpmJ_Vhc0QB0SgXxxcRSR7__mQV5rmFyfQc60nA,24
+icsdatavalidation-1.0.441.dist-info/RECORD,,

tests/snowflake_service/test_create_checksums.py

@@ -29,7 +29,7 @@ class TestCreateChecksumsParametrized:
 
     @pytest.mark.parametrize(
         "column_intersections,where_clause,numeric_scale,execute_behavior," \
-        "expected_columns,expected_errors,expect_retry,expected_execute_calls",
+        "expected_columns,expected_errors,expect_retry",
         [
             (  # success path
                 ['amount', 'name'],
@@ -46,8 +46,7 @@ class TestCreateChecksumsParametrized:
                     'NAME': ['COUNTDISTINCT', 3, 0]
                 },
                 [],
-                False,
-                1
+                False
             ),
             (  # arithmetic overflow triggers retry
                 ['amount'],
@@ -63,8 +62,7 @@ class TestCreateChecksumsParametrized:
                     'AMOUNT': ['SUM', 5, 0]
                 },
                 [],
-                True,
-                2
+                True
             ),
             (  # non-overflow error surfaces in TESTATM_ERRORS
                 ['amount'],
@@ -75,8 +73,7 @@ class TestCreateChecksumsParametrized:
                 },
                 {},
                 [['ERROR', 'checksum_sql', 'Some other error']],
-                False,
-                1
+                False
             ),
         ],
     )
@@ -90,20 +87,20 @@ class TestCreateChecksumsParametrized:
         execute_behavior,
         expected_columns,
         expected_errors,
-        expect_retry,
-        expected_execute_calls
+        expect_retry
     ):
         """Test create_checksums behavior across success, retry, and error scenarios."""
-        snowflake_service.create_checksum_statement = MagicMock(return_value='checksum_retry_sql')
-
         with patch.object(snowflake_service, '_get_checksum_statement', return_value='checksum_sql') as mock_checksum_stmt, \
              patch.object(snowflake_service, '_get_countnulls_statement', return_value='countnulls_sql') as mock_countnulls_stmt, \
              patch.object(snowflake_service, 'execute_queries') as mock_execute:
 
-            if 'side_effect' in execute_behavior:
-                mock_execute.side_effect = execute_behavior['side_effect']
+            side_effect = execute_behavior.get('side_effect')
+            if side_effect is not None:
+                mock_execute.side_effect = side_effect
+                expected_call_count = len(side_effect) if isinstance(side_effect, list) else 1
             else:
                 mock_execute.return_value = execute_behavior['return_value']
+                expected_call_count = 1
 
             result = snowflake_service.create_checksums(
                 object=mock_database_object,
@@ -114,14 +111,22 @@ class TestCreateChecksumsParametrized:
                 enclose_column_by_double_quotes=False
             )
 
-            mock_checksum_stmt.assert_called_once_with(
-                object=mock_database_object,
-                column_intersections=column_intersections,
-                where_clause=where_clause,
-                exclude_columns=[],
-                numeric_scale=numeric_scale,
-                enclose_column_by_double_quotes=False
-            )
+            checksum_calls = mock_checksum_stmt.call_args_list
+            base_kwargs = {
+                'object': mock_database_object,
+                'column_intersections': column_intersections,
+                'where_clause': where_clause,
+                'exclude_columns': [],
+                'numeric_scale': numeric_scale,
+                'enclose_column_by_double_quotes': False
+            }
+            assert checksum_calls[0].kwargs == base_kwargs
+            if expect_retry:
+                assert len(checksum_calls) == 2
+                retry_kwargs = {**base_kwargs, 'bool_cast_before_sum': True}
+                assert checksum_calls[1].kwargs == retry_kwargs
+            else:
+                assert len(checksum_calls) == 1
             mock_countnulls_stmt.assert_called_once_with(
                 object=mock_database_object,
                 column_intersections=column_intersections,
@@ -129,14 +134,7 @@ class TestCreateChecksumsParametrized:
                 exclude_columns=[],
                 enclose_column_by_double_quotes=False
             )
-            assert mock_execute.call_count == expected_execute_calls
-
-            if expect_retry:
-                snowflake_service.create_checksum_statement.assert_called_once()
-                retry_kwargs = snowflake_service.create_checksum_statement.call_args.kwargs
-                assert retry_kwargs['bool_cast_before_sum'] is True
-            else:
-                snowflake_service.create_checksum_statement.assert_not_called()
+            assert mock_execute.call_count == expected_call_count
 
             for column, expected in expected_columns.items():
                 assert result[column] == expected

tests/sqlserver_service/test_create_checksums.py

@@ -103,7 +103,6 @@ class TestCreateChecksumsParametrized:
         expected_execute_calls
     ):
         """Test create_checksums behavior across success, retry, and error scenarios."""
-        sqlserver_service.create_checksum_statement = MagicMock(return_value='checksum_retry_sql')
 
         with patch.object(sqlserver_service, '_get_checksum_statement', return_value='checksum_sql') as mock_checksum_stmt, \
              patch.object(sqlserver_service, '_get_countnulls_statement', return_value='countnulls_sql') as mock_countnulls_stmt, \
@@ -123,13 +122,21 @@ class TestCreateChecksumsParametrized:
                 enclose_column_by_double_quotes=False
             )
 
-            mock_checksum_stmt.assert_called_once_with(
-                object=mock_database_object,
-                column_intersections=column_intersections,
-                where_clause=where_clause,
-                exclude_columns=[],
-                numeric_scale=numeric_scale
-            )
+            checksum_calls = mock_checksum_stmt.call_args_list
+            base_kwargs = {
+                'object': mock_database_object,
+                'column_intersections': column_intersections,
+                'where_clause': where_clause,
+                'exclude_columns': [],
+                'numeric_scale': numeric_scale
+            }
+            assert checksum_calls[0].kwargs == base_kwargs
+            if expect_retry:
+                assert len(checksum_calls) == 2
+                retry_kwargs = {**base_kwargs, 'bool_cast_before_sum': True}
+                assert checksum_calls[1].kwargs == retry_kwargs
+            else:
+                assert len(checksum_calls) == 1
             mock_countnulls_stmt.assert_called_once_with(
                 object=mock_database_object,
                 column_intersections=column_intersections,
@@ -138,13 +145,6 @@ class TestCreateChecksumsParametrized:
             )
             assert mock_execute.call_count == expected_execute_calls
 
-            if expect_retry:
-                sqlserver_service.create_checksum_statement.assert_called_once()
-                retry_kwargs = sqlserver_service.create_checksum_statement.call_args.kwargs
-                assert retry_kwargs['bool_cast_before_sum'] is True
-            else:
-                sqlserver_service.create_checksum_statement.assert_not_called()
-
             for column, expected in expected_columns.items():
                 assert result[column] == expected