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})