From dcb40cd87546619c48055b4cc6440ccea6bb1ab1 Mon Sep 17 00:00:00 2001
From: Alex Schofield <git@ajschof.me>
Date: Fri, 21 Feb 2025 02:50:47 +0000
Subject: fix imports and instantiate Obfuscator()

---
 test/test_init.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

(limited to 'test')

diff --git a/test/test_init.py b/test/test_init.py
index b717e35..0915f13 100644
--- a/test/test_init.py
+++ b/test/test_init.py
@@ -1,9 +1,12 @@
-from obfuscator.__init__ import main
+from gdpr_obfuscator import Obfuscator
+import moto
+
+obfuscator = Obfuscator()
 
 
 def test_main_integration():
     test_source = "s3://test-bucket/data.csv"
-    result = main(test_source, ["name"], log_level="DEBUG")
+    result = obfuscator.process_s3(test_source, ["name"])
     result_str = result.decode("utf-8")
     assert "John" not in result_str
     assert "***" in result_str
-- 
cgit v1.2.3