@@ -236,21 +236,23 @@ def test_sensitive_fields_are_removed(self):
236
236
"""
237
237
Ensure some sensitive fields are never present in output of OAI-PMH apis
238
238
"""
239
+ sensitive_field_values = [
'[email protected] ' ,
'999-123-123' ,
'999-456-456' ]
239
240
240
241
def _check_fields (content ):
241
242
"""
242
- Verify sensitive fields are not in the content
243
+ Verify sensitive fields values are not in the content. Checking for field value, instead
244
+ of field name, since the field names might be different in Datacite etc other formats.
243
245
"""
244
- for sensitive_field in [ 'email' , 'telephone' , 'phone' ] :
245
- self .assertEqual (sensitive_field not in str (content ), True ,
246
- 'field %s should have been stripped' % sensitive_field )
246
+ for sensitive_field_value in sensitive_field_values :
247
+ self .assertEqual (sensitive_field_value not in str (content ), True ,
248
+ 'field %s should have been stripped' % sensitive_field_value )
247
249
248
250
# setup some records to have sensitive fields
249
251
for cr in CatalogRecord .objects .filter (pk__in = (1 , 2 , 3 )):
250
252
cr .research_dataset ['curator' ][0 ].update ({
251
-
252
- 'phone' : '123124' ,
253
- 'telephone' : '123124' ,
253
+ 'email' : sensitive_field_values [ 0 ] ,
254
+ 'phone' : sensitive_field_values [ 1 ] ,
255
+ 'telephone' : sensitive_field_values [ 2 ] ,
254
256
})
255
257
cr .force_save ()
256
258
0 commit comments