REFRESH_REQUEST_PAYLOAD_DICT = {
"jwtTokens": {"access": ACCESS_TOKEN_1, "refresh": REFRESH_TOKEN_1}
}
REFRESH_REQUEST_DICT = LOGIN_RESPONSE_DICT
REFRESH_REQUEST_OBJECT = AuthRefreshRequest(
access_token=ACCESS_TOKEN_1, refresh_token=REFRESH_TOKEN_1
)
REFRESH_RESPONSE_DICT = LOGIN_RESPONSE_DICT
REFRESH_RESPONSE_OBJECT = AuthRefreshResponse(
access_token=ACCESS_TOKEN_1, refresh_token=REFRESH_TOKEN_1
)
# OPENAPI
OPENAPI_REQUEST_OBJECT = OASRequest(mode="yaml")
OPENAPI_RESPONSE = "openapi stuff"
OPENAPI_RESPONSE_OBJECT = OASResponse(data=OPENAPI_RESPONSE)
# VERSION
VERSION_REQUEST_OBJECT = VersionRequest()
VERSION_RESPONSE_DICT = {
"api": API_VERSION_1,
"maru": MARU_VERSION_1,
"mythril": MYTHRIL_VERSION_1,
"harvey": HARVEY_VERSION_1,
"hash": HASHED_VERSION_1,
}
VERSION_RESPONSE_OBJECT = VersionResponse(
api_version=API_VERSION_1,
maru_version=MARU_VERSION_1,
mythril_version=MYTHRIL_VERSION_1,
harvey_version=HARVEY_VERSION_1,
def test_oas_response_invalid_type():
with pytest.raises(TypeError):
OASResponse(data=1)
def test_oas_response_from_valid_json():
resp = OASResponse.from_json("openapi stuff")
assert resp.data == "openapi stuff"
def test_oas_response_from_valid_dict():
resp = OASResponse.from_dict({"data": "openapi stuff"})
assert resp.data == "openapi stuff"
import json
import pytest
from mythx_models.response import OASResponse
OBJ_DATA = OASResponse.from_json("openapi stuff")
def test_oas_response_from_valid_json():
resp = OASResponse.from_json("openapi stuff")
assert resp.data == "openapi stuff"
def test_oas_response_from_valid_dict():
resp = OASResponse.from_dict({"data": "openapi stuff"})
assert resp.data == "openapi stuff"
# def test_oas_response_from_invalid_dict():
# with pytest.raises(ValidationError):
# OASResponse.from_dict({})
def test_oas_response_invalid_type():
with pytest.raises(TypeError):
OASResponse(data=1)
def test_oas_response_to_json():
assert OBJ_DATA.to_json() == json.dumps({"data": "openapi stuff"})
def test_oas_response_from_invalid_dict():
with pytest.raises(ValidationError):
OASResponse.from_dict({})
def test_oas_response_from_valid_dict():
resp = OASResponse.from_dict({"data": testdata.OPENAPI_RESPONSE})
assert resp.data == testdata.OPENAPI_RESPONSE
def test_oas_response_from_valid_json():
resp = OASResponse.from_json(testdata.OPENAPI_RESPONSE)
assert resp.data == testdata.OPENAPI_RESPONSE