Peter M. Groen / oletools

Commit a02c03d5776a4bd1cb8138082334d7e7ec7745ca

Authored by decalage2
1 parent ddf57c0e

oleform/test_basic: fixed test data, fixed olevba import

Inline Side-by-side
Showing 1 changed file with 68 additions and 5 deletions
tests/oleform/test_basic.py
  View file @a02c03d
... ... @@ -7,12 +7,75 @@ import sys
7 7 # Directory with test data, independent of current working directory
8 8 from tests.test_utils import DATA_BASE_DIR
9 9  
10   -if sys.version_info[0] <= 2:
11   - from oletools.olevba import VBA_Parser
12   -else:
13   - from oletools.olevba3 import VBA_Parser
  10 +from oletools.olevba import VBA_Parser
14 11  
15   -SAMPLES = [('oleform-PR314.docm', [('word/vbaProject.bin', u'UserFormTEST1', {'caption': 'Label1-test', 'control_tip_text': None, 'name': 'Label1', 'value': None, 'tag': 'l\x18sdf', 'ClsidCacheIndex': 21, 'id': 1, 'tabindex': 0}), ('word/vbaProject.bin', u'UserFormTEST1', {'caption': None, 'control_tip_text': None, 'name': 'TextBox1', 'value': 'heyhey', 'tag': '', 'ClsidCacheIndex': 23, 'id': 2, 'tabindex': 1}), ('word/vbaProject.bin', u'UserFormTEST1', {'caption': None, 'control_tip_text': None, 'name': 'ComboBox1', 'value': 'none dd', 'tag': '', 'ClsidCacheIndex': 25, 'id': 3, 'tabindex': 2}), ('word/vbaProject.bin', u'UserFormTEST1', {'caption': None, 'control_tip_text': None, 'name': 'CheckBox1', 'value': '1', 'tag': '', 'ClsidCacheIndex': 26, 'id': 5, 'tabindex': 4}), ('word/vbaProject.bin', u'UserFormTEST1', {'caption': None, 'control_tip_text': None, 'name': 'OptionButton1', 'value': '0', 'tag': '', 'ClsidCacheIndex': 27, 'id': 6, 'tabindex': 5}), ('word/vbaProject.bin', u'UserFormTEST1', {'caption': None, 'control_tip_text': None, 'name': 'ToggleButton1', 'value': '0', 'tag': '', 'ClsidCacheIndex': 28, 'id': 7, 'tabindex': 6}), ('word/vbaProject.bin', u'UserFormTEST1', {'caption': None, 'control_tip_text': None, 'name': 'Frame1', 'value': None, 'tag': '', 'ClsidCacheIndex': 14, 'id': 8, 'tabindex': 7}), ('word/vbaProject.bin', u'UserFormTEST1', {'caption': None, 'control_tip_text': None, 'name': 'TabStrip1', 'value': None, 'tag': '', 'ClsidCacheIndex': 18, 'id': 10, 'tabindex': 8}), ('word/vbaProject.bin', u'UserFormTEST1', {'caption': None, 'control_tip_text': None, 'name': 'CommandButton1', 'value': None, 'tag': '', 'ClsidCacheIndex': 17, 'id': 9, 'tabindex': 9}), ('word/vbaProject.bin', u'UserFormTEST1', {'caption': None, 'control_tip_text': None, 'name': 'MultiPage1', 'value': None, 'tag': '', 'ClsidCacheIndex': 57, 'id': 12, 'tabindex': 10}), ('word/vbaProject.bin', u'UserFormTEST1', {'caption': None, 'control_tip_text': None, 'name': 'ScrollBar1', 'value': None, 'tag': '', 'ClsidCacheIndex': 47, 'id': 16, 'tabindex': 11}), ('word/vbaProject.bin', u'UserFormTEST1', {'caption': None, 'control_tip_text': None, 'name': 'SpinButton1', 'value': None, 'tag': '', 'ClsidCacheIndex': 16, 'id': 17, 'tabindex': 12}), ('word/vbaProject.bin', u'UserFormTEST1', {'caption': None, 'control_tip_text': None, 'name': 'Image1', 'value': None, 'tag': '', 'ClsidCacheIndex': 12, 'id': 18, 'tabindex': 13}), ('word/vbaProject.bin', u'UserFormTEST1', {'caption': None, 'control_tip_text': None, 'name': 'ListBox1', 'value': '', 'tag': '', 'ClsidCacheIndex': 24, 'id': 4, 'tabindex': 3}), ('word/vbaProject.bin', u'UserFormTEST1/i08', {'caption': None, 'control_tip_text': None, 'name': 'TextBox2', 'value': 'abcd', 'tag': '', 'ClsidCacheIndex': 23, 'id': 20, 'tabindex': 0}), ('word/vbaProject.bin', u'UserFormTEST1/i12', {'caption': None, 'control_tip_text': None, 'name': '', 'value': None, 'tag': '', 'ClsidCacheIndex': 18, 'id': 13, 'tabindex': 2}), ('word/vbaProject.bin', u'UserFormTEST1/i12', {'caption': None, 'control_tip_text': None, 'name': 'Page1', 'value': None, 'tag': '', 'ClsidCacheIndex': 7, 'id': 14, 'tabindex': 0}), ('word/vbaProject.bin', u'UserFormTEST1/i12', {'caption': None, 'control_tip_text': None, 'name': 'Page2', 'value': None, 'tag': '', 'ClsidCacheIndex': 7, 'id': 15, 'tabindex': 1}), ('word/vbaProject.bin', u'UserFormTEST1/i12/i14', {'caption': None, 'control_tip_text': None, 'name': 'TextBox3', 'value': 'last one', 'tag': '', 'ClsidCacheIndex': 23, 'id': 24, 'tabindex': 0}), ('word/vbaProject.bin', u'UserFormTest2', {'caption': 'Label1', 'control_tip_text': None, 'name': 'Label1', 'value': None, 'tag': '', 'ClsidCacheIndex': 21, 'id': 1, 'tabindex': 0}), ('word/vbaProject.bin', u'UserFormTest2', {'caption': 'Label2', 'control_tip_text': None, 'name': 'Label2', 'value': None, 'tag': '', 'ClsidCacheIndex': 21, 'id': 2, 'tabindex': 1}), ('word/vbaProject.bin', u'UserFormTest2', {'caption': None, 'control_tip_text': None, 'name': 'TextBox1', 'value': '&\xe9"\'', 'tag': '', 'ClsidCacheIndex': 23, 'id': 3, 'tabindex': 2})])]
  12 +SAMPLES = [('oleform-PR314.docm',
  13 + [('word/vbaProject.bin', u'UserFormTEST1',
  14 + {'caption': 'Label1-test', 'control_tip_text': None, 'name': 'Label1', 'value': None, 'tag': 'l\x18sdf',
  15 + 'ClsidCacheIndex': 21, 'id': 1, 'tabindex': 0}),
  16 + ('word/vbaProject.bin', u'UserFormTEST1',
  17 + {'caption': None, 'control_tip_text': None, 'name': 'TextBox1', 'value': 'heyhey', 'tag': '',
  18 + 'ClsidCacheIndex': 23, 'id': 2, 'tabindex': 1}),
  19 + ('word/vbaProject.bin', u'UserFormTEST1',
  20 + {'caption': None, 'control_tip_text': None, 'name': 'ComboBox1', 'value': 'none dd', 'tag': '',
  21 + 'ClsidCacheIndex': 25, 'id': 3, 'tabindex': 2}),
  22 + ('word/vbaProject.bin', u'UserFormTEST1',
  23 + {'caption': None, 'control_tip_text': None, 'name': 'CheckBox1', 'value': '1', 'tag': '',
  24 + 'ClsidCacheIndex': 26, 'id': 5, 'tabindex': 4}),
  25 + ('word/vbaProject.bin', u'UserFormTEST1',
  26 + {'caption': None, 'control_tip_text': None, 'name': 'OptionButton1', 'value': '0', 'tag': '',
  27 + 'ClsidCacheIndex': 27, 'id': 6, 'tabindex': 5}),
  28 + ('word/vbaProject.bin', u'UserFormTEST1',
  29 + {'caption': None, 'control_tip_text': None, 'name': 'ToggleButton1', 'value': '0', 'tag': '',
  30 + 'ClsidCacheIndex': 28, 'id': 7, 'tabindex': 6}),
  31 + ('word/vbaProject.bin', u'UserFormTEST1',
  32 + {'caption': None, 'control_tip_text': None, 'name': 'Frame1', 'value': None, 'tag': '',
  33 + 'ClsidCacheIndex': 14, 'id': 8, 'tabindex': 7}),
  34 + ('word/vbaProject.bin', u'UserFormTEST1',
  35 + {'caption': None, 'control_tip_text': None, 'name': 'TabStrip1', 'value': None, 'tag': '',
  36 + 'ClsidCacheIndex': 18, 'id': 10, 'tabindex': 8}),
  37 + ('word/vbaProject.bin', u'UserFormTEST1',
  38 + {'caption': None, 'control_tip_text': None, 'name': 'CommandButton1', 'value': None, 'tag': '',
  39 + 'ClsidCacheIndex': 17, 'id': 9, 'tabindex': 9}),
  40 + ('word/vbaProject.bin', u'UserFormTEST1',
  41 + {'caption': None, 'control_tip_text': None, 'name': 'MultiPage1', 'value': None, 'tag': '',
  42 + 'ClsidCacheIndex': 57, 'id': 12, 'tabindex': 10}),
  43 + ('word/vbaProject.bin', u'UserFormTEST1',
  44 + {'caption': None, 'control_tip_text': None, 'name': 'ScrollBar1', 'value': None, 'tag': '',
  45 + 'ClsidCacheIndex': 47, 'id': 16, 'tabindex': 11}),
  46 + ('word/vbaProject.bin', u'UserFormTEST1',
  47 + {'caption': None, 'control_tip_text': None, 'name': 'SpinButton1', 'value': None, 'tag': '',
  48 + 'ClsidCacheIndex': 16, 'id': 17, 'tabindex': 12}),
  49 + ('word/vbaProject.bin', u'UserFormTEST1',
  50 + {'caption': None, 'control_tip_text': None, 'name': 'Image1', 'value': None, 'tag': '',
  51 + 'ClsidCacheIndex': 12, 'id': 18, 'tabindex': 13}),
  52 + ('word/vbaProject.bin', u'UserFormTEST1',
  53 + {'caption': None, 'control_tip_text': None, 'name': 'ListBox1', 'value': '', 'tag': '',
  54 + 'ClsidCacheIndex': 24, 'id': 4, 'tabindex': 3}),
  55 + ('word/vbaProject.bin', u'UserFormTEST1/i08',
  56 + {'caption': None, 'control_tip_text': None, 'name': 'TextBox2', 'value': 'abcd', 'tag': '',
  57 + 'ClsidCacheIndex': 23, 'id': 20, 'tabindex': 0}),
  58 + ('word/vbaProject.bin', u'UserFormTEST1/i12',
  59 + {'caption': None, 'control_tip_text': None, 'name': '', 'value': None, 'tag': '',
  60 + 'ClsidCacheIndex': 18, 'id': 13, 'tabindex': 2}),
  61 + ('word/vbaProject.bin', u'UserFormTEST1/i12',
  62 + {'caption': None, 'control_tip_text': None, 'name': 'Page1', 'value': None, 'tag': '',
  63 + 'ClsidCacheIndex': 7, 'id': 14, 'tabindex': 0}),
  64 + ('word/vbaProject.bin', u'UserFormTEST1/i12',
  65 + {'caption': None, 'control_tip_text': None, 'name': 'Page2', 'value': None, 'tag': '',
  66 + 'ClsidCacheIndex': 7, 'id': 15, 'tabindex': 1}),
  67 + ('word/vbaProject.bin', u'UserFormTEST1/i12/i14',
  68 + {'caption': None, 'control_tip_text': None, 'name': 'TextBox3', 'value': 'last one', 'tag': '',
  69 + 'ClsidCacheIndex': 23, 'id': 24, 'tabindex': 0}),
  70 + ('word/vbaProject.bin', u'UserFormTest2',
  71 + {'caption': 'Label1', 'control_tip_text': None, 'name': 'Label1', 'value': None, 'tag': '',
  72 + 'ClsidCacheIndex': 21, 'id': 1, 'tabindex': 0}),
  73 + ('word/vbaProject.bin', u'UserFormTest2',
  74 + {'caption': 'Label2', 'control_tip_text': None, 'name': 'Label2', 'value': None, 'tag': '',
  75 + 'ClsidCacheIndex': 21, 'id': 2, 'tabindex': 1}),
  76 + ('word/vbaProject.bin', u'UserFormTest2',
  77 + {'caption': None, 'control_tip_text': None, 'name': 'TextBox1', 'value': '&\xe9"\'', 'tag': '',
  78 + 'group_name': '', 'ClsidCacheIndex': 23, 'id': 3, 'tabindex': 2})])]
16 79  
17 80 class TestOleForm(unittest.TestCase):
18 81  
... ...