def test__get_pii_fields(self): """Test get pii fields""" # Setup table_meta = { 'fields': { 'foo': { 'type': 'categorical', 'pii': True, 'pii_category': 'email' }, 'bar': { 'type': 'categorical', 'pii_category': 'email' } } } # Run metadata = Mock(spec=Metadata) metadata.get_table_meta.return_value = table_meta table_name = 'test' result = Metadata._get_pii_fields(metadata, table_name) # Asserts expected = {'foo': 'email'} assert result == expected
def test__get_pii_fields(self): """Test get pii fields""" # Setup table_meta = { 'fields': { 'foo': { 'type': 'categorical', 'pii': True, 'pii_category': 'email' }, 'bar': { 'type': 'categorical', 'pii_category': 'email' } } } metadata = Mock(spec_set=Metadata) metadata.get_table_meta.return_value = table_meta # Run result = Metadata._get_pii_fields(metadata, 'test') # Asserts assert result == {'foo': 'email'}