1 files changed, 9 insertions, 0 deletions
diff --git a/test/test_init.py b/test/test_init.py
new file mode 100644
index 0000000..b717e35
--- /dev/null
+++ b/test/test_init.py
@@ -0,0 +1,9 @@
+from obfuscator.__init__ import main
+
+
+def test_main_integration():
+    test_source = "s3://test-bucket/data.csv"
+    result = main(test_source, ["name"], log_level="DEBUG")
+    result_str = result.decode("utf-8")
+    assert "John" not in result_str
+    assert "***" in result_str