"""journal_images_lib 单元测试。"""
import json
import os
import tempfile
import unittest
from io import BytesIO

from lib.instance.journal_images_lib import (
    JOURNAL_UPLOAD_TFS,
    enrich_journal_api_item,
    images_json_dumps,
    journal_image_paths,
    journal_upload_field_name,
    parse_images_json,
    primary_journal_image,
    save_journal_slot_uploads,
)


class _FakeFile:
    def __init__(self, filename: str, data: bytes):
        self.filename = filename
        self._data = data

    def save(self, path: str) -> None:
        with open(path, "wb") as f:
            f.write(self._data)


class _FakeFiles:
    def __init__(self, mapping):
        self._mapping = mapping

    def get(self, key):
        return self._mapping.get(key)


class JournalImagesLibTest(unittest.TestCase):
    def test_field_names(self):
        self.assertEqual(journal_upload_field_name("5m"), "screenshot_5m")

    def test_save_slot_uploads_partial(self):
        with tempfile.TemporaryDirectory() as tmp:
            files = _FakeFiles(
                {
                    "screenshot_5m": _FakeFile("a.png", b"png5"),
                    "screenshot_1h": _FakeFile("b.jpg", b"jpg1"),
                }
            )
            saved = save_journal_slot_uploads(
                files,
                "abc123",
                tmp,
                secure_filename_fn=lambda x: x,
            )
            self.assertEqual(len(saved), 2)
            self.assertEqual(saved[0]["tf"], "5m")
            self.assertTrue(os.path.isfile(os.path.join(tmp, saved[0]["file"])))
            self.assertEqual(saved[1]["tf"], "1h")

    def test_parse_and_enrich(self):
        raw = images_json_dumps([{"tf": "5m", "file": "journal_x_5m.png"}])
        item = enrich_journal_api_item({"images_json": raw, "image": "legacy.png"})
        self.assertEqual(len(item["images"]), 1)
        self.assertEqual(item["images"][0]["tf"], "5m")

        legacy = enrich_journal_api_item({"image": "only.png"})
        self.assertEqual(legacy["images"][0]["file"], "only.png")

    def test_journal_image_paths_dedupe(self):
        with tempfile.TemporaryDirectory() as tmp:
            path = os.path.join(tmp, "same.png")
            with open(path, "wb") as f:
                f.write(b"x")
            row = {
                "image": "same.png",
                "images_json": json.dumps([{"tf": "5m", "file": "same.png"}]),
            }
            paths = journal_image_paths(row, tmp)
            self.assertEqual(len(paths), 1)

    def test_primary_journal_image(self):
        self.assertEqual(
            primary_journal_image([{"tf": "5m", "file": "a.png"}]),
            "a.png",
        )
        self.assertIsNone(primary_journal_image([]))


if __name__ == "__main__":
    unittest.main()