From b9f5d73a95062834318fb13693643bf1f3994a2e Mon Sep 17 00:00:00 2001 From: Alex Schofield Date: Sun, 2 Mar 2025 23:58:20 +0000 Subject: fix obfuscator test where pii fields are missing --- test/test_obfuscator.py | 14 +++----------- 1 file changed, 3 insertions(+), 11 deletions(-) (limited to 'test/test_obfuscator.py') diff --git a/test/test_obfuscator.py b/test/test_obfuscator.py index 66d0312..d90d905 100644 --- a/test/test_obfuscator.py +++ b/test/test_obfuscator.py @@ -1,4 +1,5 @@ from gdpr_obfuscator.obfuscate import obfuscate_data +import pytest def test_obfuscate_data_with_valid_pii_fields(): @@ -47,18 +48,9 @@ def test_obfuscate_data_with_missing_pii_field(): }, ] pii_fields = ["name", "email_address"] - expected = [ - {"student_id": "1234", "name": "***", "course": "Software"}, - { - "student_id": "5678", - "name": "***", - "course": "Data Science", - "email_address": "***", - }, - ] - result = obfuscate_data(data, pii_fields) - assert result == expected + with pytest.raises(Exception): + obfuscate_data(data, pii_fields) def test_obfuscate_data_with_no_data(): -- cgit v1.2.3