def test_get_encrypted_field_dne(self):
fields = [EncryptedDeviceField('test_name', 'integer', 123),
EncryptedDeviceField('test2', 'string', 'something')]
device = Device('test', [], fields, 'type')
with self.assertRaises(UnknownDeviceField):
device.get_encrypted_field('invalid')
def test_get_encrypted_field(self):
fields = [EncryptedDeviceField('test_name', 'integer', 123),
EncryptedDeviceField('test2', 'string', 'something')]
device = Device('test', [], fields, 'type')
self.assertEqual(device.get_encrypted_field('test_name'), 123)