diff --git a/ereuse_devicehub/parser/schemas.py b/ereuse_devicehub/parser/schemas.py deleted file mode 100644 index 557dc10d..00000000 --- a/ereuse_devicehub/parser/schemas.py +++ /dev/null @@ -1,32 +0,0 @@ -from flask import current_app as app -from marshmallow import Schema as MarshmallowSchema -from marshmallow import ValidationError, validates_schema -from marshmallow.fields import Dict, List, Nested, String - - -class Snapshot_lite_data(MarshmallowSchema): - dmidecode = String(required=False) - hwinfo = String(required=False) - smart = List(Dict(), required=False) - lshw = Dict(required=False) - - -class Snapshot_lite(MarshmallowSchema): - uuid = String(required=True) - version = String(required=True) - software = String(required=True) - wbid = String(required=True) - type = String(required=True) - timestamp = String(required=True) - data = Nested(Snapshot_lite_data) - - @validates_schema - def validate_workbench_version(self, data: dict): - if data['version'] not in app.config['WORKBENCH_LITE']: - raise ValidationError( - 'Min. supported Workbench version is ' - '{} but yours is {}.'.format( - app.config['WORKBENCH_LITE'][0], data['version'] - ), - field_names=['version'], - ) diff --git a/tests/conftest.py b/tests/conftest.py index b4e97af0..536bbfb8 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -23,6 +23,7 @@ from ereuse_devicehub.resources.tag import Tag from ereuse_devicehub.resources.user.models import User from ereuse_devicehub.resources.user.models import Session from ereuse_devicehub.resources.enums import SessionType +from ereuse_devicehub.api.views import api STARTT = datetime(year=2000, month=1, day=1, hour=1) """A dummy starting time to use in tests.""" @@ -69,6 +70,7 @@ def app(request, _app: Devicehub) -> Devicehub: tag_token=uuid.UUID('52dacef0-6bcb-4919-bfed-f10d2c96ecee'), erase=False, common=True) + _app.register_blueprint(api) with _app.app_context(): try: