From 1e836b7b23fbe87252423d279920f370884b3196 Mon Sep 17 00:00:00 2001 From: Alex Schofield Date: Wed, 26 Feb 2025 16:37:27 +0000 Subject: update test_core.py to use json string as input --- test/test_core.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'test/test_core.py') 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 -- cgit v1.2.3