134 lines
4.7 KiB
Python
134 lines
4.7 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Unit and integration tests for chart.py
|
|
"""
|
|
|
|
import json
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
# Add scripts directory to path
|
|
SCRIPTS_DIR = Path(__file__).resolve().parent
|
|
sys.path.insert(0, str(SCRIPTS_DIR))
|
|
|
|
from chart import parse_input_data
|
|
|
|
|
|
class TestChartParser(unittest.TestCase):
|
|
def test_list_of_dicts(self):
|
|
raw = json.dumps([
|
|
{"day": "Mo", "temp": 18},
|
|
{"day": "Di", "temp": 22},
|
|
{"day": "Mi", "temp": 19},
|
|
])
|
|
cols, rows, x, y = parse_input_data(raw)
|
|
self.assertEqual(cols, ["day", "temp"])
|
|
self.assertEqual(len(rows), 3)
|
|
self.assertEqual(x, "day")
|
|
self.assertEqual(y, "temp")
|
|
|
|
def test_dict_key_value(self):
|
|
raw = json.dumps({"Bier": 4.5, "Pizza": 12.0, "Kaffee": 3.2})
|
|
cols, rows, x, y = parse_input_data(raw)
|
|
self.assertEqual(cols, ["category", "value"])
|
|
self.assertEqual(len(rows), 3)
|
|
self.assertEqual(x, "category")
|
|
self.assertEqual(y, "value")
|
|
|
|
def test_explicit_columns_values(self):
|
|
raw = json.dumps({
|
|
"columns": ["Monat", "Umsatz"],
|
|
"values": [["Jan", 100], ["Feb", 150]]
|
|
})
|
|
cols, rows, x, y = parse_input_data(raw)
|
|
self.assertEqual(cols, ["Monat", "Umsatz"])
|
|
self.assertEqual(len(rows), 2)
|
|
self.assertEqual(x, "Monat")
|
|
self.assertEqual(y, "Umsatz")
|
|
|
|
|
|
class TestChartCLI(unittest.TestCase):
|
|
def run_cli(self, args, stdin_input=None):
|
|
cmd = [sys.executable, str(SCRIPTS_DIR / "chart.py")] + args
|
|
res = subprocess.run(
|
|
cmd,
|
|
input=stdin_input,
|
|
capture_output=True,
|
|
text=True,
|
|
timeout=30,
|
|
)
|
|
return res
|
|
|
|
def test_bar_chart_generation(self):
|
|
data = json.dumps([
|
|
{"label": "Projekt A", "stunden": 14},
|
|
{"label": "Projekt B", "stunden": 28},
|
|
{"label": "Projekt C", "stunden": 8},
|
|
])
|
|
out_path = Path("/tmp/charts/test_bar.png")
|
|
if out_path.exists():
|
|
out_path.unlink()
|
|
|
|
res = self.run_cli(["--type", "bar", "--title", "Projektstunden", "--data", data, "--output", str(out_path)])
|
|
self.assertEqual(res.returncode, 0, f"Stderr: {res.stderr}")
|
|
self.assertTrue(res.stdout.strip().startswith("MEDIA:"))
|
|
self.assertTrue(out_path.exists())
|
|
self.assertGreater(out_path.stat().st_size, 1000)
|
|
|
|
def test_line_chart_stdin(self):
|
|
data = json.dumps([
|
|
{"datum": "2026-09-10", "puls": 62},
|
|
{"datum": "2026-09-11", "puls": 58},
|
|
{"datum": "2026-09-12", "puls": 65},
|
|
{"datum": "2026-09-13", "puls": 59},
|
|
])
|
|
out_path = Path("/tmp/charts/test_line.png")
|
|
if out_path.exists():
|
|
out_path.unlink()
|
|
|
|
res = self.run_cli(["--type", "line", "--title", "Ruhepuls Verlauf", "--data", "-", "--output", str(out_path)], stdin_input=data)
|
|
self.assertEqual(res.returncode, 0, f"Stderr: {res.stderr}")
|
|
self.assertTrue(res.stdout.strip().startswith("MEDIA:"))
|
|
self.assertTrue(out_path.exists())
|
|
self.assertGreater(out_path.stat().st_size, 1000)
|
|
|
|
def test_donut_chart(self):
|
|
data = json.dumps([
|
|
{"bereich": "Wohnen", "kosten": 950},
|
|
{"bereich": "Essen", "kosten": 450},
|
|
{"bereich": "Mobilität", "kosten": 120},
|
|
{"bereich": "Freizeit", "kosten": 200},
|
|
])
|
|
out_path = Path("/tmp/charts/test_donut.png")
|
|
if out_path.exists():
|
|
out_path.unlink()
|
|
|
|
res = self.run_cli(["--type", "donut", "--title", "Ausgaben", "--data", data, "--output", str(out_path)])
|
|
self.assertEqual(res.returncode, 0, f"Stderr: {res.stderr}")
|
|
self.assertTrue(res.stdout.strip().startswith("MEDIA:"))
|
|
self.assertTrue(out_path.exists())
|
|
self.assertGreater(out_path.stat().st_size, 1000)
|
|
|
|
def test_quickchart_fallback_engine(self):
|
|
data = json.dumps([
|
|
{"fach": "Mathe", "note": 1.7},
|
|
{"fach": "Physik", "note": 2.0},
|
|
{"fach": "Informatik", "note": 1.0},
|
|
])
|
|
out_path = Path("/tmp/charts/test_quickchart.png")
|
|
if out_path.exists():
|
|
out_path.unlink()
|
|
|
|
res = self.run_cli(["--engine", "quickchart", "--type", "bar", "--title", "Notenspiegel", "--data", data, "--output", str(out_path)])
|
|
self.assertEqual(res.returncode, 0, f"Stderr: {res.stderr}")
|
|
self.assertTrue(res.stdout.strip().startswith("MEDIA:"))
|
|
self.assertTrue(out_path.exists())
|
|
self.assertGreater(out_path.stat().st_size, 1000)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|