def test_model_from_dict():
"""Test class Model `from_dict` method."""
exc = WazuhError(1000)
with pytest.raises(exc.__class__):
bm.Model.from_dict(exc)
dikt = {'test_key': 'test_value'}
assert bm.Model.from_dict(dikt) == deserialize_model(dikt, bm.Model)
def from_dict(cls, dikt) -> 'BasicInfo':
"""Returns the dict as a model
:param dikt: A dict.
:type: dict
:return: The BasicInfo of this BasicInfo.
:rtype: BasicInfo
"""
return util.deserialize_model(dikt, cls)
def from_dict(cls, dikt) -> 'ResultUUID':
"""Returns the dict as a model
:param dikt: A dict.
:type: dict
:return: The inline_response_200 of this ResultUUID.
:rtype: ResultUUID
"""
return util.deserialize_model(dikt, cls)
def test_deserialize_model():
"""Check that _deserialize_model function transforms item into desired object"""
test = {'data': 'test'}
result = util.deserialize_model(test, TestClass)
assert result.data == 'test'
assert isinstance(result, TestClass)
assert isinstance(result.attribute_map, dict)
assert isinstance(result.swagger_types, dict)
def from_dict(cls, dikt) -> 'TokenResponseModel':
"""Returns the dict as a model
:param dikt: A dict.
:type: dict
:return: The TokenResponse of this TokenResponse. # noqa: E501
:rtype: TokenResponseModel
"""
return util.deserialize_model(dikt, cls)
def from_dict(cls, dikt) -> 'Data':
"""Returns the dict as a model
:param dikt: A dict.
:type: dict
:return: The data of this Data. # noqa: E501
:rtype: Data
"""
return util.deserialize_model(dikt, cls)
def from_dict(cls, dikt) -> 'InlineResponse200':
"""Returns the dict as a model
:param dikt: A dict.
:type: dict
:return: The inline_response_200 of this InlineResponse200. # noqa: E501
:rtype: InlineResponse200
"""
return util.deserialize_model(dikt, cls)
def from_dict(cls, dikt) -> 'Items':
"""Returns the dict as a model
:param dikt: A dict.
:type: dict
:return: The Items of this Items. # noqa: E501
:rtype: Items
"""
return util.deserialize_model(dikt, cls)
def from_dict(cls, dikt) -> 'TextVector':
"""Returns the dict as a model
:param dikt: A dict.
:type: dict
:return: The TextVector of this TextVector. # noqa: E501
:rtype: TextVector
"""
return util.deserialize_model(dikt, cls)
def from_dict(cls, dikt):
"""Returns the dict as a model
:param dikt: A dict.
:type: dict
:return: The Agent of this Agent.
:rtype: dict
"""
return util.deserialize_model(dikt, cls)
def from_dict(cls, dikt) -> 'ListOfTexts':
"""Returns the dict as a model
:param dikt: A dict.
:type: dict
:return: The ListOfTexts of this ListOfTexts. # noqa: E501
:rtype: ListOfTexts
"""
return util.deserialize_model(dikt, cls)
def from_dict(cls, dikt) -> 'WebPage':
"""Returns the dict as a model
:param dikt: A dict.
:type: dict
:return: The WebPage of this WebPage. # noqa: E501
:rtype: WebPage
"""
return util.deserialize_model(dikt, cls)
def test_body_from_dict():
"""Test class Body `from_dict` method."""
test_dict = {
'test_key1': 'test_value1',
'test_key2': [{
'test_key21': 'test_value21'
}]
}
assert bm.Body.from_dict(test_dict) == deserialize_model(
test_dict, bm.Body)
async def get_kwargs(cls, request, additional_kwargs: dict = None):
try:
dikt = request if isinstance(request,
dict) else await request.json()
f_kwargs = util.deserialize_model(dikt, cls).to_dict()
except JSONDecodeError:
raise_if_exc(WazuhError(1018))
invalid = get_invalid_keys(dikt, f_kwargs)
if invalid:
raise ProblemException(
status=400,
title='Bad Request',
detail='Invalid field found {}'.format(invalid))
if additional_kwargs is not None:
f_kwargs.update(additional_kwargs)
return f_kwargs
async def get_kwargs(cls, request, additional_kwargs: dict = None):
try:
dikt = request if isinstance(request,
dict) else await request.json()
f_kwargs = util.deserialize_model(dikt, cls).to_dict()
except JSONDecodeError as e:
raise_if_exc(APIError(code=2005, details=e.msg))
invalid = {
key
for key in dikt.keys() if key not in list(f_kwargs.keys())
}
if invalid:
raise_if_exc(
WazuhError(
5005,
extra_message='Invalid field found {}'.format(invalid)))
if additional_kwargs is not None:
f_kwargs.update(additional_kwargs)
return f_kwargs
def from_dict(cls: typing.Type[T], dikt) -> T:
"""Returns the dict as a model"""
if isinstance(dikt, Exception):
raise dikt
return util.deserialize_model(dikt, cls)
def from_dict(cls: typing.Type[T], dikt) -> T:
"""Returns the dict as a model"""
return util.deserialize_model(dikt, cls)
def test_items_from_dict():
"""Test class Items `from_dict` class method."""
test_dict_list = [{'test_key': 'test_value'}, {'test_key2': 'test_value2'}]
assert bm.Items.from_dict(test_dict_list) == deserialize_model(
test_dict_list, bm.Items)
def from_dict(cls, dikt) -> 'Contact':
return util.deserialize_model(dikt, cls)
def test_data_from_dict():
"""Test class Data `from_dict` class method."""
test_dict = {'test_key': 'test_value'}
assert bm.Data.from_dict(test_dict) == deserialize_model(
test_dict, bm.Data)
