Skip to content

Commit b25fc49

Browse files
committed
More tests fixup
1 parent 3de8200 commit b25fc49

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

tests/test_parse_pe.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
import pytest
22
import time
33
import zlib
4-
import base64
54
import ctypes
65
import windows
76

@@ -26,7 +25,7 @@ def pe(request):
2625
@pytest.fixture(scope="session")
2726
def pe_dotnet32(tmpdir_factory):
2827
tmpdir = tmpdir_factory.mktemp("pe_dotnet32_test_dir")
29-
pe_dotnet32_data = zlib.decompress(base64.decodebytes(PE_DOTNET32_DLL_BASE64))
28+
pe_dotnet32_data = zlib.decompress(b64decode(PE_DOTNET32_DLL_BASE64))
3029
fullpath = str(tmpdir.join(PE_DOTNET32_DLL_NAME))
3130
with open(fullpath, "wb") as f:
3231
f.write(pe_dotnet32_data)

windows/rpc/ndr.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -554,6 +554,7 @@ class NdrStream(object):
554554
def __init__(self, data):
555555
self.fulldata = data
556556
self.data = data
557+
# self.data = memoryview(data) # FAST but need some code rewrite at some places
557558

558559
def partial_unpack(self, format):
559560
size = struct.calcsize(format)

0 commit comments

Comments
 (0)