mirror of
https://github.com/dobin/avred
synced 2026-06-08 13:54:13 +00:00
31 lines
963 B
Python
31 lines
963 B
Python
#!/usr/bin/env python
|
|
|
|
import unittest
|
|
from plugins.pe.file_pe import FilePe
|
|
from filehelper import getFileScannerTypeFor, FileType, getFileInfo
|
|
|
|
|
|
class UtilsTest(unittest.TestCase):
|
|
def test_magic(self):
|
|
filename = "tests/data/dotnet-test.dll"
|
|
type = getFileScannerTypeFor(filename)
|
|
self.assertEqual(type, FileType.DOTNET)
|
|
|
|
filename = "tests/data/P5-5h3ll.docm"
|
|
type = getFileScannerTypeFor(filename)
|
|
self.assertEqual(type, FileType.OFFICE)
|
|
|
|
filename = "tests/data/test.exe"
|
|
type = getFileScannerTypeFor(filename)
|
|
self.assertEqual(type, FileType.EXE)
|
|
|
|
|
|
def test_fileInfo(self):
|
|
file = FilePe()
|
|
file.loadFromFile('tests/data/test.exe')
|
|
fileInfo = getFileInfo(file)
|
|
|
|
self.assertEqual(fileInfo.name, 'test.exe')
|
|
self.assertEqual(fileInfo.size, 89062)
|
|
self.assertEqual(fileInfo.hash, b'\xcai\xed\x146.\xfe\x01\xb0|\x9a\xd4uv\x07\xd1')
|