| 1 | import csv |
| 2 | import io |
| 3 | import pathlib |
| 4 | import unittest |
| 5 | from export import export |
| 6 | |
| 7 | |
| 8 | class Export(unittest.TestCase): |
| 9 | def test_matches_golden(self): |
| 10 | want = pathlib.Path("tests/golden/export.bin").read_bytes() |
| 11 | self.assertEqual(export(), want) |
| 12 | |
| 13 | def test_output_is_valid_csv(self): |
| 14 | rows = list(csv.reader(io.StringIO(export().decode("utf-8")))) |
| 15 | self.assertEqual(rows[0], ["id", "name"]) |
| 16 |