From 0f68a242daba0837f50b9aa5d721cfab12b5d107 Mon Sep 17 00:00:00 2001
From: Dobin <dobin.rutishauser@gmail.com>
Date: Mon, 19 Feb 2024 16:35:59 +0000
Subject: [PATCH] feature: disable observer in tests

---
 observer.py                 | 3 +++
 tests/test_asm.py           | 5 +++++
 tests/test_derbackdoorer.py | 5 +++++
 3 files changed, 13 insertions(+)

diff --git a/observer.py b/observer.py
index b9bb0ad..37ce946 100644
--- a/observer.py
+++ b/observer.py
@@ -11,6 +11,7 @@ class Observer():
     def __init__(self):
         self.logs = []
         self.idx = 0
+        self.active = True
 
     def add_text(self, name, data):
         self.write_to_file(name + ".txt", data)
@@ -28,6 +29,8 @@ class Observer():
         self.idx += 1
 
     def write_to_file(self, filename, data):
+        if not self.active:
+            return
         with open("logs/{}-{}".format(self.idx, filename), "w") as f:
             f.write(data)
 
diff --git a/tests/test_asm.py b/tests/test_asm.py
index 00cd841..e52db1f 100644
--- a/tests/test_asm.py
+++ b/tests/test_asm.py
@@ -6,9 +6,14 @@ import logging
 from phases.compiler import fixup_asm_file, fixup_iat_reuse
 from model import ExeInfo
 from defs import *
+from observer import observer
 
 
 class AsmTest(unittest.TestCase):
+    @classmethod
+    def setUpClass(cls):
+        observer.active = False
+        
     def test_asm_fixup(self):
         path_in: FilePath = "tests/data/peb_walk_pre_fixup.asm"
         path_working: FilePath = "tests/data/peb_walk_pre_fixup.asm.test"
diff --git a/tests/test_derbackdoorer.py b/tests/test_derbackdoorer.py
index 11fd7f6..001f474 100644
--- a/tests/test_derbackdoorer.py
+++ b/tests/test_derbackdoorer.py
@@ -7,6 +7,7 @@ from model import ExeInfo
 from defs import *
 from pehelper import extract_code_from_exe
 from helper import hexdump
+from observer import observer
 
 from derbackdoorer.derbackdoorer import PeBackdoor
 
@@ -21,6 +22,10 @@ from derbackdoorer.derbackdoorer import PeBackdoor
 #   * The call has been patched
 
 class DerBackdoorerTest(unittest.TestCase):
+    @classmethod
+    def setUpClass(cls):
+        observer.active = False
+
     def test_backdoor_ep(self):
         # Write example shellcode
         shellcode_path = "exes/shellcode.test"