#!/usr/bin/env python3
# -*- coding: utf-8 -*-

from __future__ import annotations

import unittest
import zlib

from core.ksse_crc import ksse_crc32


class KsseCrcTests(unittest.TestCase):
    def test_matches_standard_crc32_seed_zero(self) -> None:
        data = b"0123456789ABCDEF@DFP@NEBULA"
        self.assertEqual(ksse_crc32(data), zlib.crc32(data) & 0xFFFFFFFF)

    def test_matches_standard_crc32_with_seed(self) -> None:
        data = b"ro.product.model=OnePlus(PJZ110)"
        seed = 0x12345678
        self.assertEqual(ksse_crc32(data, seed), zlib.crc32(data, seed) & 0xFFFFFFFF)


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