def test_extract_header(): with open(os.path.join(utils.SAMPLES_DIR, 'example1.exe'), 'rb') \ as nsis_file: firstheader = fileform._find_firstheader(nsis_file) header = fileform._extract_header(nsis_file, firstheader) assert header is not None assert len(header.blocks) == fileform.BLOCKS_COUNT assert len(header.install_types) == 33
def test_extract_blocks(): with open(os.path.join(utils.SAMPLES_DIR, 'example1.exe'), 'rb') \ as nsis_file: firstheader = fileform._find_firstheader(nsis_file) header = fileform._extract_header(nsis_file, firstheader) for block_id in [fileform.NB_PAGES, fileform.NB_SECTIONS, fileform.NB_ENTRIES, fileform.NB_STRINGS, fileform.NB_LANGTABLES, fileform.NB_CTLCOLORS]: pages_block = fileform._extract_block(nsis_file, firstheader, block_id) assert pages_block is not None
def test_extract_blocks(): with open(os.path.join(utils.SAMPLES_DIR, 'example1.exe'), 'rb') \ as nsis_file: firstheader = fileform._find_firstheader(nsis_file) header = fileform._extract_header(nsis_file, firstheader) for block_id in [ fileform.NB_PAGES, fileform.NB_SECTIONS, fileform.NB_ENTRIES, fileform.NB_STRINGS, fileform.NB_LANGTABLES, fileform.NB_CTLCOLORS ]: pages_block = fileform._extract_block(nsis_file, firstheader, block_id) assert pages_block is not None
def test_extract_lzma_solid(): with open(os.path.join(utils.SAMPLES_DIR, 'example_lzma_solid.exe'), 'rb') \ as nsis_file: firstheader = fileform._find_firstheader(nsis_file) header = fileform._extract_header(nsis_file, firstheader) assert header is not None