refactor: Pass args instead of station_name

suisa_sendemeldung/suisa_sendemeldung.py

@@ -511,19 +511,20 @@ def get_isrc(music: Any) -> str:  # noqa: ANN401
     return isrc
 
 
-def get_csv(data: dict, station_name: str = "") -> str:
+def get_csv(data: dict, args: ArgparseNamespace) -> str:
     """Create SUISA compatible csv data.
 
     Arguments:
     ---------
         data: To data to create csv from
-        station_name: Default station name for output
+        args: Parsed arguments
 
     Returns:
     -------
         csv: The converted data
 
     """
+    station_name = args.station_name
     header = [
         "Titel",
         "Komponist",
@@ -642,20 +643,20 @@ def get_csv(data: dict, station_name: str = "") -> str:
     return csv.getvalue()
 
 
-def get_xlsx(data: Any, station_name: str = "") -> BytesIO:  # noqa: ANN401
+def get_xlsx(data: Any, args: ArgparseNamespace) -> BytesIO:  # noqa: ANN401
     """Create SUISA compatible xlsx data.
 
     Arguments:
     ---------
         data: The data to create xlsx from
-        station_name: Default station name for output
+        args: Parsed arguments
 
     Returns:
     -------
         xlsx: The converted data as BytesIO object
 
     """
-    csv = get_csv(data, station_name=station_name)
+    csv = get_csv(data, args=args)
     csv_reader = reader(StringIO(csv))
 
     xlsx = BytesIO()
@@ -862,9 +863,9 @@ def main() -> None:  # pragma: no cover
     )
     data = merge_duplicates(data)
     if args.filetype == "xlsx":
-        data = get_xlsx(data, station_name=args.station_name)
+        data = get_xlsx(data, args=args)
     elif args.filetype == "csv":
-        data = get_csv(data, station_name=args.station_name)
+        data = get_csv(data, args=args)
     if args.email:
         email_subject = Template(args.email_subject).substitute(
             {

tests/conftest.py

@@ -0,0 +1,19 @@
+import pytest
+from configargparse import ArgumentParser  # type: ignore[import-untyped]
+
+from suisa_sendemeldung.suisa_sendemeldung import get_arguments
+
+
+@pytest.fixture
+def args():
+    return get_arguments(
+        ArgumentParser(),
+        [
+            "--bearer-token=butaisheeph6sewoo8aiDa8ieyaethoo",
+            "--project-id=123456789",
+            "--stream-id=123456789",
+            "--timezone=UTC",
+            "--file",
+            "--station-name=Station Name",
+        ],
+    )

tests/test_suisa_sendemeldung.py

@@ -195,13 +195,13 @@ def test_funge_release_date(test_date, expected):
 
 
 @patch("cridlib.get")
-def test_get_csv(mock_cridlib_get, snapshot):
+def test_get_csv(mock_cridlib_get, snapshot, args):
     """Test get_csv."""
     mock_cridlib_get.return_value = "crid://rabe.ch/v1/test"
 
     # empty data
     data = []
-    csv = suisa_sendemeldung.get_csv(data)
+    csv = suisa_sendemeldung.get_csv(data, args=args)
     assert csv == snapshot
     mock_cridlib_get.assert_not_called()
 
@@ -322,7 +322,7 @@ def test_get_csv(mock_cridlib_get, snapshot):
             },
         },
     ]
-    csv = suisa_sendemeldung.get_csv(data, station_name="Station Name")
+    csv = suisa_sendemeldung.get_csv(data, args=args)
     assert csv == snapshot
     mock_cridlib_get.assert_has_calls(
         [
@@ -350,12 +350,12 @@ def test_get_csv(mock_cridlib_get, snapshot):
     )
 
 
-def test_get_xlsx(snapshot):
+def test_get_xlsx(snapshot, args):
     """Test get_xlsx."""
 
     # empty data
     data = []
-    xlsx = suisa_sendemeldung.get_xlsx(data)
+    xlsx = suisa_sendemeldung.get_xlsx(data, args=args)
     workbook = load_workbook(xlsx)
     worksheet = workbook.active
     assert list(worksheet.values) == snapshot