diff options
| -rw-r--r-- | test/test_core.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/test/test_core.py b/test/test_core.py index f60f4d2..ad2a2a6 100644 --- a/test/test_core.py +++ b/test/test_core.py @@ -4,6 +4,7 @@ from moto import mock_aws import boto3 import csv import random +import json obfuscator = Obfuscator() @@ -40,7 +41,9 @@ def test_main_integration(): path = f"s3://{bucket}/{key}" - result = obfuscator.process_s3(path, ["name"]) + json_input = json.dumps({"file_path": path, "pii_fields": ["name"]}) + + result = obfuscator.process_s3(json_input) result_str = result.decode("utf-8") assert rand_name not in result_str |