# -*- coding: utf-8 -*-
# :-
# SPDX-FileCopyrightText: © 2025 Franz Glasner
# SPDX-License-Identifier: BSD-3-Clause
# :-
r"""Unit tests for treesum

"""

from __future__ import absolute_import, print_function

import os
import shutil
import sys
import unittest

from _test_setup import (DATADIR, TMPDIR)

import cutils.treesum


class TaggedTreesumTests(unittest.TestCase):

    def setUp(self):
        if not os.path.isdir(TMPDIR):
            os.mkdir(TMPDIR)

    def tearDown(self):
        if os.path.isdir(TMPDIR):
            shutil.rmtree(TMPDIR)

    def test_gen_and_info_P_utf8(self):
        digest_file = os.path.join(
            TMPDIR,
            "%d__test_gen_and_info_P_utf8.info" % (os.getpid(),))
        rel_tmpdir = os.path.relpath(TMPDIR, DATADIR)
        gen_opts = cutils.treesum.gen_generate_opts(
            directories=[DATADIR],
            algorithm="SHA1",
            fnmatch_filters=[("exclude", "path:%s" % (rel_tmpdir,)),
                             ("exclude", "glob:*.treesum")],
            generator="full",
            grouping_separator="_",
            output=digest_file,
            output_style="tagged",
            print_size=True,
            utf8=True)
        info_opts = cutils.treesum.gen_info_opts(digest_files=[digest_file],
                                                 last=True)
        cutils.treesum.generate_treesum(gen_opts)
        cutils.treesum.print_treesum_digestfile_infos(info_opts)
        info = cutils.treesum.TreesumInfo.collect_last_from_file(digest_file)
        self.assertEqual(
            b"\xcb\x04\x39\x44\x60\x08\xd7\xfa\x0a\x25\xfd\xc6\xbb\x74\x8e\x2c\x12\x13\xf8\x31",    # noqa: E501 line too long
            info.digest)
        self.assertEqual(55, info.size)
        self.assertEqual("SHA1", info.algorithm)

    def test_gen_and_info_P_native(self):
        digest_file = os.path.join(
            TMPDIR,
            "%d__test_gen_and_info_P_native.info" % (os.getpid(),))
        rel_tmpdir = os.path.relpath(TMPDIR, DATADIR)
        gen_opts = cutils.treesum.gen_generate_opts(
            directories=[DATADIR],
            algorithm="SHA1",
            fnmatch_filters=[("exclude", "path:%s" % (rel_tmpdir,)),
                             ("exclude", "glob:*.treesum")],
            generator="full",
            grouping_separator="_",
            output=digest_file,
            output_style="tagged",
            print_size=True,
            utf8=False)
        info_opts = cutils.treesum.gen_info_opts(digest_files=[digest_file],
                                                 last=True)
        cutils.treesum.generate_treesum(gen_opts)
        cutils.treesum.print_treesum_digestfile_infos(info_opts)
        info = cutils.treesum.TreesumInfo.collect_last_from_file(digest_file)
        self.assertEqual(
            b"\xcb\x04\x39\x44\x60\x08\xd7\xfa\x0a\x25\xfd\xc6\xbb\x74\x8e\x2c\x12\x13\xf8\x31",    # noqa: E501 line too long
            info.digest)
        self.assertEqual(55, info.size)
        self.assertEqual("SHA1", info.algorithm)

    def test_gen_and_info_L_utf8(self):
        digest_file = os.path.join(
            TMPDIR,
            "%d__test_gen_and_info_L_utf8.info" % (os.getpid(),))
        rel_tmpdir = os.path.relpath(TMPDIR, DATADIR)
        gen_opts = cutils.treesum.gen_generate_opts(
            directories=[DATADIR],
            algorithm="SHA1",
            fnmatch_filters=[("exclude", "path:%s" % (rel_tmpdir,)),
                             ("exclude", "glob:*.treesum")],
            follow_symlinks=cutils.treesum.FollowSymlinkConfig(
                True, True, True),
            generator="full",
            grouping_separator="_",
            output=digest_file,
            output_style="tagged",
            print_size=True,
            utf8=True)
        info_opts = cutils.treesum.gen_info_opts(digest_files=[digest_file],
                                                 last=True)
        cutils.treesum.generate_treesum(gen_opts)
        cutils.treesum.print_treesum_digestfile_infos(info_opts)
        info = cutils.treesum.TreesumInfo.collect_last_from_file(digest_file)
        self.assertEqual(
            b"\xfe\x21\x4c\xfa\xdc\xc5\x1f\xad\x63\x19\x74\x95\xf5\xb3\x6d\x32\x45\xec\x73\x3e",    # noqa: E501 line too long
            info.digest)
        self.assertEqual(110, info.size)
        self.assertEqual("SHA1", info.algorithm)

    def test_gen_and_info_L_native(self):
        digest_file = os.path.join(
            TMPDIR,
            "%d__test_gen_and_info_L_native.info" % (os.getpid(),))
        rel_tmpdir = os.path.relpath(TMPDIR, DATADIR)
        gen_opts = cutils.treesum.gen_generate_opts(
            directories=[DATADIR],
            algorithm="SHA1",
            fnmatch_filters=[("exclude", "path:%s" % (rel_tmpdir,)),
                             ("exclude", "glob:*.treesum")],
            follow_symlinks=cutils.treesum.FollowSymlinkConfig(
                True, True, True),
            generator="full",
            grouping_separator="_",
            output=digest_file,
            output_style="tagged",
            print_size=True,
            utf8=False)
        info_opts = cutils.treesum.gen_info_opts(digest_files=[digest_file],
                                                 last=True)
        cutils.treesum.generate_treesum(gen_opts)
        cutils.treesum.print_treesum_digestfile_infos(info_opts)
        info = cutils.treesum.TreesumInfo.collect_last_from_file(digest_file)
        self.assertEqual(
            b"\xfe\x21\x4c\xfa\xdc\xc5\x1f\xad\x63\x19\x74\x95\xf5\xb3\x6d\x32\x45\xec\x73\x3e",    # noqa: E501 line too long
            info.digest)
        self.assertEqual(110, info.size)
        self.assertEqual("SHA1", info.algorithm)

    def test_gen_and_info_P_utf8_with_treesum(self):
        digest_file = os.path.join(
            TMPDIR,
            "%d__test_gen_and_info_P_native.info" % (os.getpid(),))
        rel_tmpdir = os.path.relpath(TMPDIR, DATADIR)
        gen_opts = cutils.treesum.gen_generate_opts(
            directories=[DATADIR],
            algorithm="SHA1",
            fnmatch_filters=[("exclude", "path:%s" % (rel_tmpdir,))],
            generator="full",
            grouping_separator="_",
            output=digest_file,
            output_style="tagged",
            print_size=True,
            utf8=True)
        info_opts = cutils.treesum.gen_info_opts(digest_files=[digest_file],
                                                 last=True)
        cutils.treesum.generate_treesum(gen_opts)
        cutils.treesum.print_treesum_digestfile_infos(info_opts)
        info = cutils.treesum.TreesumInfo.collect_last_from_file(digest_file)
        self.assertEqual(
            b"\x78\xdd\xcd\xd2\xbe\xa5\x2c\x8c\xc9\x1e\x3a\xd4\x26\xda\x35\x2e\xa6\xc2\x0a\x06",    # noqa: E501 line too long
            info.digest)
        self.assertEqual(2620, info.size)
        self.assertEqual("SHA1", info.algorithm)

    def test_gen_and_info_P_utf8_accept_treesum(self):
        digest_file = os.path.join(
            TMPDIR,
            "%d__test_gen_and_info_P_native.info" % (os.getpid(),))
        rel_tmpdir = os.path.relpath(TMPDIR, DATADIR)
        gen_opts = cutils.treesum.gen_generate_opts(
            directories=[DATADIR],
            algorithm="SHA1",
            fnmatch_filters=[("exclude", "path:%s" % (rel_tmpdir,)),
                             ("accept-treesum", "glob:*.treesum")],
            generator="full",
            grouping_separator="_",
            output=digest_file,
            output_style="tagged",
            print_size=True,
            utf8=True)
        info_opts = cutils.treesum.gen_info_opts(digest_files=[digest_file],
                                                 last=True)
        cutils.treesum.generate_treesum(gen_opts)
        cutils.treesum.print_treesum_digestfile_infos(info_opts)
        info = cutils.treesum.TreesumInfo.collect_last_from_file(digest_file)
        self.assertEqual(
            b"\x69\x6f\xe2\x51\xbe\x94\xbe\xcc\x76\xa5\x91\x24\x1d\x46\x83\xbb\x44\x36\xc7\x9b\x5b\x7b\x62\xb3\xe0\x4a\x0e\xdc\x7e\xc4\x07\xcb",    # noqa: E501 line too long
            info.digest)
        # accepts the size within the .treesum file
        self.assertEqual(67, info.size)
        # accepts uses the digest algorithm from the .treesum file
        self.assertEqual("SHA256", info.algorithm)


if __name__ == "__main__":
    sys.exit(unittest.main(buffer=True))