unittest: add simple csv file and test it

Christian Herdtweck
1 parent 4ac29b53
Showing 2 changed files with 15 additions and 1 deletions
tests/msodde/test_csv.py
tests/test-data/msodde/dde-in-csv.csv
@@ -6,9 +6,10 @@
 import unittest
 from tempfile import mkstemp
 import os
+from os.path import join
  
 from oletools import msodde
-from tests.test_utils import OutputCapture
+from tests.test_utils import OutputCapture, DATA_BASE_DIR
  
  
 class TestCSV(unittest.TestCase):
@@ -67,6 +68,18 @@ class TestCSV(unittest.TestCase):
                             if self.DO_DEBUG:
                                 print('Worked: ' + desc)
  
+    def test_file(self):
+        """ test simple small example file """
+        filename = join(DATA_BASE_DIR, 'msodde', 'dde-in-csv.csv')
+        with OutputCapture() as capturer:
+            capturer.reload_module(msodde)    # re-create logger
+            ret_code = msodde.main([filename, ])
+        self.assertEqual(ret_code, 0)
+        links = self.get_dde_from_output(capturer)
+        self.assertEqual(len(links), 1)
+        self.assertEqual(links[0],
+                         "cmd '/k \..\..\..\Windows\System32\calc.exe'")
+
     def write_and_run(self, sample_text):
         """ helper for test_texts: save text to file, run through msodde """
         filename = None
+=cmd|'/k \..\..\..\Windows\System32\calc.exe'!A0