Peter M. Groen / oletools

""" Test that oleobj detects external links in relationships files.
"""

import unittest
import os
from os import path

# Directory with test data, independent of current working directory
from tests.test_utils import DATA_BASE_DIR
from oletools import oleobj

BASE_DIR = path.join(DATA_BASE_DIR, 'oleobj', 'external_link')


class TestExternalLinks(unittest.TestCase):
    def test_external_links(self):
        """
        loop through sample files asserting that external links are found
        """

        for dirpath, _, filenames in os.walk(BASE_DIR):
            for filename in filenames:
                file_path = path.join(dirpath, filename)

                ret_val = oleobj.main([file_path])
                self.assertEqual(ret_val, oleobj.RETURN_DID_DUMP)


# just in case somebody calls this file as a script
if __name__ == '__main__':
    unittest.main()