Peter M. Groen / oletools

Commit b8ba847bd09c84ce6973eca1ffce7c2dfb0516da

Authored by Christian Herdtweck
1 parent f899797d

tests: speed up longest test 

On my test VM this reduces test time for this one test from 33s to
10s
Inline Side-by-side
Showing 1 changed file with 6 additions and 2 deletions
tests/olevba/test_basic.py
  View file @b8ba847
... ... @@ -76,14 +76,14 @@ class TestOlevbaBasic(unittest.TestCase):
76 76  
77 77 def test_crypt_return(self):
78 78 """
79   - Tests that encrypted files give a certain return code.
  79 + Test that encrypted files give a certain return code.
80 80  
81 81 Currently, only the encryption applied by Office 2010 (CryptoApi RC4
82 82 Encryption) is tested.
83 83 """
84 84 CRYPT_DIR = join(DATA_BASE_DIR, 'encrypted')
85 85 CRYPT_RETURN_CODE = 9
86   - ADD_ARGS = [], ['-d', ], ['-a', ], ['-j', ], ['-t', ]
  86 + ADD_ARGS = [], ['-d', ], ['-a', ], ['-j', ], ['-t', ] # only 1st file
87 87 EXCEPTIONS = ['autostart-encrypt-standardpassword.xls', # These ...
88 88 'autostart-encrypt-standardpassword.xlsm', # files ...
89 89 'autostart-encrypt-standardpassword.xlsb', # are ...
... ... @@ -103,6 +103,10 @@ class TestOlevbaBasic(unittest.TestCase):
103 103 msg='Wrong return code {} for args {}'\
104 104 .format(ret_code, args + [filename, ]))
105 105  
  106 + # test only first file with all arg combinations, others just
  107 + # without arg (test takes too long otherwise
  108 + ADD_ARGS = ([], )
  109 +
106 110  
107 111 # just in case somebody calls this file as a script
108 112 if __name__ == '__main__':
... ...