test_tar.py

import unittest

from polydet import PolyglotLevel
from polydet.plugins import tar


# TODO: look at https://sourceforge.net/projects/s-tar/files/testscripts/ for tests
class TestTARDetector(unittest.TestCase):
    def test_is_tar(self):
        self.assertEqual(PolyglotLevel(), tar.check('tests/samples/tar/regular.tar'))

    def test_is_not_tar(self):
        # From https://sourceforge.net/projects/s-tar/files/testscripts/
        self.assertIsNone(tar.check('tests/samples/tar/not_a_tar_file'))
        self.assertIsNone(tar.check('tests/samples/zip/regular.zip'))

    def test_payload_in_filename_field(self):
        self.assertEqual(PolyglotLevel(suspicious_chunks=[(0x410, 0x54)]),
                         tar.check('tests/samples/tar/payload_in_filename_field.tar'))

    def test_garbage_at_end(self):
        self.assertEqual(PolyglotLevel(suspicious_chunks=[(0x2800, 0x400)]),
                         tar.check('tests/samples/tar/garbage_at_end.tar'))