def test_exposed_value(self):
mock_type = MagicMock()
mock_type.get_value = lambda user: '******'
mock_type.from_value_secure = lambda value, spec: mock_type
info = {'gob_type': mock_type, 'spec': 'any spec'}
value = Authority.exposed_value('any value', info)
self.assertEqual(value, 'protected value')
value = Authority.exposed_value(None, info)
self.assertEqual(value, None)
def test_handle_secured_columns(self):
authority = Authority('secure catalog', 'any col')
authority.get_secured_columns = lambda: {'col': 'any info'}
authority.exposed_value = lambda value, info: 'exposed value'
row = {}
authority._handle_secured_columns({}, row)
self.assertEqual(row, {})
row = {'col': 'any value', 'any other col': 'any value'}
authority._handle_secured_columns({}, row)
self.assertEqual(row, {
'col': 'exposed value',
'any other col': 'any value'
})
row = {'mapped col': 'any value', 'any other col': 'any value'}
mapping = {'col': 'mapped col'}
authority._handle_secured_columns(mapping, row)
self.assertEqual(row, {
'mapped col': 'exposed value',
'any other col': 'any value'
})