test_filehash.py

#!/usr/bin/python3

import unittest
import filehash


class TestFilehash(unittest.TestCase):
    def test_ed2k_empty(self):
        ed2k = filehash.Ed2k()
        ed2k.update(b"")
        self.assertEqual(ed2k.hexdigest(), "31d6cfe0d16ae931b73c59d7e0c089c0")

    def test_ed2k_chunk_of_zeroes(self):
        ed2k = filehash.Ed2k()
        ed2k.update(b"\00" * 9728000)
        self.assertEqual(ed2k.hexdigest(), "fc21d9af828f92a8df64beac3357425d")

    def test_ed2k_double_chunk_of_zeroes(self):
        ed2k = filehash.Ed2k()
        ed2k.update(b"\00" * 9728000 * 2)
        self.assertEqual(ed2k.hexdigest(), "114b21c63a74b6ca922291a11177dd5c")

    def test_ed2k_lonely_zero(self):
        ed2k = filehash.Ed2k()
        ed2k.update(b"\00" * 1)
        self.assertEqual(ed2k.hexdigest(), "47c61a0fa8738ba77308a8a600f88e4b")

    def test_ed2k_over_nine_thousand_zeroes(self):
        ed2k = filehash.Ed2k()
        ed2k.update(b"\00" * 9001)
        self.assertEqual(ed2k.hexdigest(), "7248d472369fe4e27fe7571b5c4e9a24")

    def test_crc32_empty(self):
        crc32_empty = filehash.Crc32()
        crc32_empty.update(b"")
        self.assertEqual(crc32_empty.hexdigest(), "00000000")

    def test_crc32_chunk_of_zeroes(self):
        crc32_chunk_of_zeroes = filehash.Crc32()
        crc32_chunk_of_zeroes.update(b"\00" * 9728000)
        self.assertEqual(crc32_chunk_of_zeroes.hexdigest(), "3abc06ba")


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