def test_from_json(self):
plaintext_fields = [DeviceField('test_name', 'integer', 123), DeviceField('test2', 'string', 'something')]
encrypted_field1 = EncryptedDeviceField('test3', 'boolean', True)
encrypted_field2 = EncryptedDeviceField('test4', 'string', 'something else')
encrypted_fields = [encrypted_field1, encrypted_field2]
encrypted_field_json1 = encrypted_field1.as_json()
encrypted_field_json2 = encrypted_field2.as_json()
encrypted_field_json1_with_value = dict(encrypted_field_json1)
encrypted_field_json2_with_value = dict(encrypted_field_json2)
encrypted_field_json1_with_value['value'] = True
encrypted_field_json2_with_value['value'] = 'something_else'
fields_json = [field.as_json() for field in plaintext_fields]
fields_json.extend([encrypted_field_json1_with_value, encrypted_field_json2_with_value])
device = Device.from_json({'name': 'test', 'fields': fields_json, 'type': 'something', 'description': 'desc'})
expected_plaintext_fields_as_json = [field.as_json() for field in plaintext_fields]
expected_encrypted_fields_as_json = [field.as_json() for field in encrypted_fields]
self.assertEqual(device.name, 'test')
self.assertEqual(device.type, 'something')
self.assertEqual(device.description, 'desc')
for field in device.plaintext_fields:
self.assertIn(field.as_json(), expected_plaintext_fields_as_json)
for field in device.encrypted_fields:
self.assertIn(field.as_json(), expected_encrypted_fields_as_json)
def test_construct_fields_from_json(self):
plaintext_fields = [DeviceField('test_name', 'integer', 123), DeviceField('test2', 'string', 'something')]
encrypted_field1 = EncryptedDeviceField('test3', 'boolean', True)
encrypted_field2 = EncryptedDeviceField('test4', 'string', 'something else')
encrypted_field_json1 = encrypted_field1.as_json()
encrypted_field_json2 = encrypted_field2.as_json()
encrypted_field_json1['value'] = True
encrypted_field_json2['value'] = 'something_else'
fields_json = [field.as_json() for field in plaintext_fields]
fields_json.extend([encrypted_field_json1, encrypted_field_json2])
plaintext_fields, encrypted_fields = Device._construct_fields_from_json(fields_json)
self.assertSetEqual({field.name for field in plaintext_fields}, {'test_name', 'test2'})
self.assertSetEqual({field.name for field in encrypted_fields}, {'test3', 'test4'})
def test_from_json(self):
plaintext_fields = [
DeviceField('test_name', 'integer', 123),
DeviceField('test2', 'string', 'something')
]
encrypted_field1 = EncryptedDeviceField('test3', 'boolean', True)
encrypted_field2 = EncryptedDeviceField('test4', 'string',
'something else')
encrypted_fields = [encrypted_field1, encrypted_field2]
encrypted_field_json1 = encrypted_field1.as_json()
encrypted_field_json2 = encrypted_field2.as_json()
encrypted_field_json1_with_value = dict(encrypted_field_json1)
encrypted_field_json2_with_value = dict(encrypted_field_json2)
encrypted_field_json1_with_value['value'] = True
encrypted_field_json2_with_value['value'] = 'something_else'
fields_json = [field.as_json() for field in plaintext_fields]
fields_json.extend([
encrypted_field_json1_with_value, encrypted_field_json2_with_value
])
device = Device.from_json({
'name': 'test',
'fields': fields_json,
'type': 'something',
'description': 'desc'
})
expected_plaintext_fields_as_json = [
field.as_json() for field in plaintext_fields
]
expected_encrypted_fields_as_json = [
field.as_json() for field in encrypted_fields
]
self.assertEqual(device.name, 'test')
self.assertEqual(device.type, 'something')
self.assertEqual(device.description, 'desc')
for field in device.plaintext_fields:
self.assertIn(field.as_json(), expected_plaintext_fields_as_json)
for field in device.encrypted_fields:
self.assertIn(field.as_json(), expected_encrypted_fields_as_json)
def test_construct_fields_from_json(self):
plaintext_fields = [
DeviceField('test_name', 'integer', 123),
DeviceField('test2', 'string', 'something')
]
encrypted_field1 = EncryptedDeviceField('test3', 'boolean', True)
encrypted_field2 = EncryptedDeviceField('test4', 'string',
'something else')
encrypted_field_json1 = encrypted_field1.as_json()
encrypted_field_json2 = encrypted_field2.as_json()
encrypted_field_json1['value'] = True
encrypted_field_json2['value'] = 'something_else'
fields_json = [field.as_json() for field in plaintext_fields]
fields_json.extend([encrypted_field_json1, encrypted_field_json2])
plaintext_fields, encrypted_fields = Device._construct_fields_from_json(
fields_json)
self.assertSetEqual({field.name
for field in plaintext_fields},
{'test_name', 'test2'})
self.assertSetEqual({field.name
for field in encrypted_fields},
{'test3', 'test4'})
def test_as_json_basic_encrypted(self):
field = EncryptedDeviceField('test_name', 'integer', 123)
self.assertDictEqual(field.as_json(), {'name': 'test_name', 'type': 'integer', 'encrypted': True})
def test_as_json_basic_encrypted(self):
field = EncryptedDeviceField('test_name', 'integer', 123)
self.assertDictEqual(field.as_json(), {'name': 'test_name', 'type': 'integer', 'encrypted': True})