fix: unittest paths

This commit is contained in:
Dobin
2024-03-30 21:54:48 +00:00
parent 75f3a76786
commit ae198cda33
3 changed files with 14 additions and 13 deletions
+9 -9
View File
@@ -6,9 +6,9 @@ import logging
from model.exehost import ExeHost
from model.defs import *
from pe.pehelper import extract_code_from_exe_file
from helper import hexdump
from utils import hexdump
from observer import observer
from model.defs import *
from pe.derbackdoorer import PeBackdoor
@@ -28,13 +28,13 @@ class DerBackdoorerTest(unittest.TestCase):
def test_backdoor_ep(self):
# Write example shellcode
shellcode_path = "data/exes/shellcode.test"
shellcode_path = PATH_EXES + "shellcode.test"
shellcode = b"\x90" * 200
with open(shellcode_path, "wb") as f:
f.write(shellcode)
exe_path = "data/exes/iattest-full.exe"
exe_out_path = "data/exes/iattest-full-test.exe"
exe_path = PATH_EXES + "iattest-full.exe"
exe_out_path = PATH_EXES + "iattest-full.test.exe"
shutil.copyfile(exe_path, exe_out_path)
@@ -59,12 +59,12 @@ class DerBackdoorerTest(unittest.TestCase):
def test_backdoor_hijack(self):
# Write example shellcode
shellcode = b"\x90" * 200
with open("data/exes/shellcode.test", "wb") as f:
with open(PATH_EXES + "shellcode.test", "wb") as f:
f.write(shellcode)
shellcode_path = "data/exes/shellcode.test"
exe_path = "data/exes/7z.exe"
exe_out_path = "data/exes/7z-test.exe"
shellcode_path = PATH_EXES + "shellcode.test"
exe_path = PATH_EXES + "7z.exe"
exe_out_path = PATH_EXES + "7z.test.exe"
shutil.copyfile(exe_path, exe_out_path)