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