test_wav.py

from unittest import TestCase

from polydet import PolyglotLevel
from polydet.plugins import wav


class TestWAVDetector(TestCase):
    def test_regular_file(self):
        self.assertEqual(PolyglotLevel(),
                         wav.check('tests/samples/wav/regular.wav'))

    def test_not_wav(self):
        self.assertIsNone(wav.check('tests/samples/zip/regular.zip'))

    def test_garbage_at_beginning(self):
        result = wav.check('tests/samples/wav/garbage_at_beginning.wav')
        self.assertEqual(PolyglotLevel(suspicious_chunks=[(0, 0xF)]), result)

    def test_garbage_at_end(self):
        result = wav.check('tests/samples/wav/garbage_at_end.wav')
        self.assertEqual(PolyglotLevel(suspicious_chunks=[(0x56414, 0x2E8C)]),
                         result)

    def test_garbage_at_beginning_and_end(self):
        result = wav.check('tests/samples/wav/garbage_at_beginning_and_end.wav')
        self.assertEqual(PolyglotLevel(suspicious_chunks=[(0, 0x15), (0x56429, 0x2E8C)]),
                         result)