From e3b54a4e1fb1c8be10a064640a9b3e8a12b28969 Mon Sep 17 00:00:00 2001
From: msm <msm@cert.pl>
Date: Wed, 10 Apr 2024 17:51:43 +0200
Subject: [PATCH] Add tests

---
 tests/test_procmem.py    | 2 +-
 tests/test_procmemelf.py | 4 ++++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/tests/test_procmem.py b/tests/test_procmem.py
index 699a4e0..28d407b 100644
--- a/tests/test_procmem.py
+++ b/tests/test_procmem.py
@@ -271,4 +271,4 @@ def test_procmemdnpe():
         assert p.pe is not None
         assert p.pe.dn_metadata.struct.Version == b'v4.0.30319\x00\x00'
         assert p.pe.dn_metadata.struct.NumberOfStreams == len(p.pe.dn_metadata.streams)
-
+        assert p.pe.dn_user_string(1).value == "Hello World!"
diff --git a/tests/test_procmemelf.py b/tests/test_procmemelf.py
index c64a275..e3af9db 100644
--- a/tests/test_procmemelf.py
+++ b/tests/test_procmemelf.py
@@ -17,6 +17,7 @@ def test_hello_static():
         assert pelf.elf.elfclass == 64
         assert pelf.elf.get_machine_arch() == 'x64'
         assert pelf.elf.little_endian
+        assert pelf.imgend == 7159808
         
 
 def test_hello_32():
@@ -26,6 +27,7 @@ def test_hello_32():
         assert pelf.elf.elfclass == 32
         assert pelf.elf.get_machine_arch() == 'x86'
         assert pelf.elf.little_endian
+        assert pelf.imgend == 8192
         
 
 def test_hello_32_static():
@@ -35,6 +37,7 @@ def test_hello_32_static():
         assert pelf.elf.elfclass == 32
         assert pelf.elf.get_machine_arch() == 'x86'
         assert pelf.elf.little_endian
+        assert pelf.imgend == 135200768
 
 
 def test_hidden_32_static():
@@ -48,3 +51,4 @@ def test_hidden_32_static():
                       b"\x00\xcd\x80\x5a\x59\x5b\x58\x68\x73\x87\x04\x08\xc3\x28\x68\x69\x64\x64\x65\x6e\x20\x63\x6f"\
                       b"\x64\x65\x21\x29\x0a"
         assert pelf.readv(0x80ed200, len(hidden_code)) == hidden_code
+        assert pelf.imgend == 135200768