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_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
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
