fix parameter name in channel send func to filter

Author: lalepee

cosmotech/coal/store/output/aws_channel.py

@@ -25,7 +25,7 @@ def __init__(self, dct: Dotdict = None):
         self.configuration = Configuration(dct)
         self._s3 = S3(self.configuration)
 
-    def send(self, tables_filter: Optional[list[str]] = None) -> bool:
+    def send(self, filter: Optional[list[str]] = None) -> bool:
 
         _s = Store(store_location=self.configuration.cosmotech.parameters_absolute_path)
 
@@ -43,8 +43,8 @@ def send(self, tables_filter: Optional[list[str]] = None) -> bool:
             self._s3.upload_file(_file_path, _uploaded_file_name)
         else:
             tables = list(_s.list_tables())
-            if tables_filter:
-                tables = [t for t in tables if t in tables_filter]
+            if filter:
+                tables = [t for t in tables if t in filter]
 
             for table_name in tables:
                 _data_stream = BytesIO()

cosmotech/coal/store/output/az_storage_channel.py

@@ -25,7 +25,7 @@ class AzureStorageChannel(ChannelInterface):
     def __init__(self, dct: Dotdict = None):
         self.configuration = Configuration(dct)
 
-    def send(self, tables_filter: Optional[list[str]] = None) -> bool:
+    def send(self, filter: Optional[list[str]] = None) -> bool:
         dump_store_to_azure(
             store_folder=self.configuration.cosmotech.dataset_absolute_path,
             account_name=self.configuration.azure.account_name,
@@ -35,7 +35,7 @@ def send(self, tables_filter: Optional[list[str]] = None) -> bool:
             client_secret=self.configuration.azure.client_secret,
             output_type=self.configuration.azure.output_type,
             file_prefix=self.configuration.azure.file_prefix,
-            selected_tables=tables_filter,
+            selected_tables=filter,
         )
 
     def delete(self):

cosmotech/coal/store/output/postgres_channel.py

@@ -26,12 +26,12 @@ class PostgresChannel(ChannelInterface):
     def __init__(self, dct: Dotdict = None):
         self.configuration = Configuration(dct)
 
-    def send(self, tables_filter: Optional[list[str]] = None) -> bool:
+    def send(self, filter: Optional[list[str]] = None) -> bool:
         run_id = send_runner_metadata_to_postgresql(self.configuration)
         dump_store_to_postgresql_from_conf(
             self.configuration,
             store_folder=self.configuration.cosmotech.dataset_absolute_path,
-            selected_tables=tables_filter,
+            selected_tables=filter,
             fk_id=run_id,
         )
 

tests/unit/coal/test_store/test_output/test_aws_channel.py

@@ -170,7 +170,7 @@ def test_send_with_tables_filter(self, mock_s3_class, mock_store_class):
         channel = AwsChannel(config)
 
         # Act
-        channel.send(tables_filter=["table1", "table3"])
+        channel.send(filter=["table1", "table3"])
 
         # Assert
         # Should only process table1 and table3

tests/unit/coal/test_store/test_output/test_az_storage_channel.py

@@ -107,7 +107,7 @@ def test_send_with_filter(self, mock_dump):
         tables_filter = ["table1", "table2"]
 
         # Act
-        channel.send(tables_filter=tables_filter)
+        channel.send(filter=tables_filter)
 
         # Assert
         mock_dump.assert_called_once_with(

tests/unit/coal/test_store/test_output/test_postgres_channel.py

@@ -122,7 +122,7 @@ def test_send_with_filter(self, mock_send_metadata, mock_dump):
         tables_filter = ["table1", "table2", "table3"]
 
         # Act
-        channel.send(tables_filter=tables_filter)
+        channel.send(filter=tables_filter)
 
         # Assert
         mock_send_metadata.assert_called_once()