Add New Basic Tests

tests/test_csv_export.py

@@ -0,0 +1,131 @@
+# flake8: noqa
+# mypy: ignore-errors
+"""Test the CSV Export System."""
+
+import pytest
+import csv
+import commodity
+
+
+@pytest.fixture
+def sample_capi_data():
+    """Provides a standard CAPI data structure for testing."""
+    return {
+        "timestamp": "2024-01-01T12:00:00Z",
+        "lastSystem": {"name": "Sol"},
+        "lastStarport": {
+            "name": "Daedalus",
+            "commodities": [
+                {
+                    "id": 123,
+                    "name": "Gold",
+                    "sellPrice": 1000,
+                    "buyPrice": 1100,
+                    "demand": 500,
+                    "demandBracket": 3,
+                    "stock": 10,
+                    "stockBracket": 1,
+                    "meanPrice": 1050,
+                }
+            ],
+        },
+    }
+
+
+class TestCSVExport:
+
+    def test_delimiter_mapping(self, tmp_path, sample_capi_data):
+        """Verify that 'kind' correctly changes the file delimiter."""
+        kinds = [
+            (commodity.COMMODITY_CSV, ","),
+            (commodity.COMMODITY_TAB, "\t"),
+            (commodity.COMMODITY_PIPE, "|"),
+            (commodity.COMMODITY_SEMICOLON, ";"),
+        ]
+
+        for kind, expected_delim in kinds:
+            fname = tmp_path / f"test_{kind}.txt"
+            commodity.export(sample_capi_data, kind=kind, filename=fname)
+
+            # Read first line to check delimiter
+            content = fname.read_text(encoding="utf-8")
+            assert expected_delim in content
+
+    def test_legacy_csv_format_row_length(self, tmp_path, sample_capi_data):
+        """COMMODITY_CSV (Legacy) should have 10 columns."""
+        fname = tmp_path / "legacy.csv"
+        commodity.export(sample_capi_data, kind=commodity.COMMODITY_CSV, filename=fname)
+
+        with open(fname, newline="", encoding="utf-8") as f:
+            reader = csv.reader(f, delimiter=",")
+            header = next(reader)
+            first_row = next(reader)
+
+            assert len(header) == 10
+            assert "Average" not in header
+            assert len(first_row) == 10
+
+    def test_new_format_row_length(self, tmp_path, sample_capi_data):
+        """Other formats (TAB, PIPE, etc) should have 12 columns."""
+        fname = tmp_path / "new_format.txt"
+        commodity.export(sample_capi_data, kind=commodity.COMMODITY_TAB, filename=fname)
+
+        with open(fname, newline="", encoding="utf-8") as f:
+            reader = csv.reader(f, delimiter="\t")
+            header = next(reader)
+            first_row = next(reader)
+
+            assert len(header) == 12
+            assert "Average" in header
+            assert "FDevID" in header
+            assert len(first_row) == 12
+
+    def test_data_value_mapping(self, tmp_path, sample_capi_data):
+        """Check if specific fields like demandBracket are mapped via bracketmap."""
+        fname = tmp_path / "data_check.csv"
+        commodity.export(
+            sample_capi_data, kind=commodity.COMMODITY_CSV_NEW, filename=fname
+        )
+
+        with open(fname, newline="", encoding="utf-8") as f:
+            reader = csv.DictReader(f, delimiter=",")
+            row = next(reader)
+
+            assert row["System"] == "Sol"
+            assert row["Commodity"] == "Gold"
+            assert row["demandBracket"] == "High"  # Mapped from 3
+            assert row["stockBracket"] == "Low"  # Mapped from 1
+            assert row["FDevID"] == "123"
+
+    def test_missing_data_handling(self, tmp_path):
+        """Ensure the exporter handles None values gracefully by writing empty strings."""
+        minimal_data = {
+            "timestamp": "now",
+            "lastSystem": {"name": "Sothis"},
+            "lastStarport": {
+                "name": "MiningHub",
+                "commodities": [
+                    {
+                        "name": "Iron",
+                        "sellPrice": None,
+                        "buyPrice": None,
+                        "demandBracket": None,
+                        "stockBracket": None,
+                    }
+                ],
+            },
+        }
+        fname = tmp_path / "empty.csv"
+        commodity.export(minimal_data, kind=commodity.COMMODITY_CSV, filename=fname)
+
+        with open(fname, newline="", encoding="utf-8") as f:
+            reader = csv.reader(f)
+            next(reader)  # skip header
+            row = next(reader)
+
+            # Indices for Sell, Buy, Demand, Supply
+            # ['Sothis', 'MiningHub', 'Iron', '', '', '', '', '', '', 'now']
+            assert row[3] == ""
+            assert row[4] == ""
+            assert row[5] == ""
+            assert row[7] == ""