Peter M. Groen / oletools

Commit 4cbf6aa985171dfc29bf9040020065e4dfaac27f

Authored by Philippe Lagadec
Committed by GitHub
2 parents e150f893 560b9032

Merge pull request #769 from christian-intra2net/fix-unittests 

Fix unittests
Inline Side-by-side
Showing 2 changed files with 2 additions and 5 deletions
oletools/olevba.py
  View file @4cbf6aa
... ... @@ -3112,7 +3112,7 @@ class VBA_Parser(object):
3112 3112 xlm_macro_found = False
3113 3113 xlm_macros = []
3114 3114 xlm_macros.append('Formulas and XLM/Excel 4 macros extracted from SLK file:')
3115   - for line in data.splitlines(keepends=False):
  3115 + for line in data.splitlines(False):
3116 3116 if line.startswith(b'O'):
3117 3117 # Option: "O;E" indicates a macro sheet, must appear before NN and C rows
3118 3118 for s in line.split(b';'):
... ...
tests/olevba/test_basic.py
  View file @4cbf6aa
... ... @@ -120,12 +120,9 @@ class TestOlevbaBasic(unittest.TestCase):
120 120 args=[full_name, ] + ADD_ARGS,
121 121 accept_nonzero_exit=True)
122 122 output = json.loads(out_str)
123   - self.assertEqual(len(output), 3)
  123 + self.assertEqual(len(output), 2)
124 124 self.assertEqual(output[0]['type'], 'MetaInformation')
125 125 self.assertEqual(output[0]['script_name'], 'olevba')
126   - self.assertEqual(output[-1]['type'], 'MetaInformation')
127   - self.assertEqual(output[-1]['n_processed'], 1)
128   - self.assertEqual(output[-1]['return_code'], 0)
129 126 result = output[1]
130 127 self.assertTrue(result['json_conversion_successful'])
131 128 if suffix in ('.xlsb', '.xltm', '.xlsm'):
... ...