trustchain-oc1-orchestral
/
devicehub-teal
Archived
10
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

validate fields for the new snapshot

This commit is contained in:
Cayo Puigdefabregas 2022-03-18 12:07:22 +01:00
parent 6bad17013d
commit 18883b2b8d
3 changed files with 485 additions and 263 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

275
ereuse_devicehub/resources/action/schemas.py
View File

@ -1,14 +1,28 @@
import copy import copy
from datetime import datetime, timedelta from datetime import datetime, timedelta
from dateutil.tz import tzutc from dateutil.tz import tzutc
from flask import current_app as app, g from flask import current_app as app
from marshmallow import Schema as MarshmallowSchema, ValidationError, fields as f, validates_schema, pre_load, post_load from flask import g
from marshmallow.fields import Boolean, DateTime, Decimal, Float, Integer, Nested, String, \ from marshmallow import Schema as MarshmallowSchema
TimeDelta, UUID from marshmallow import ValidationError
from marshmallow import fields as f
from marshmallow import post_load, pre_load, validates_schema
from marshmallow.fields import (
UUID,
Boolean,
DateTime,
Decimal,
Float,
Integer,
Nested,
String,
TimeDelta,
)
from marshmallow.validate import Length, OneOf, Range from marshmallow.validate import Length, OneOf, Range
from sqlalchemy.util import OrderedSet from sqlalchemy.util import OrderedSet
from teal.enums import Country, Currency, Subdivision from teal.enums import Country, Currency, Subdivision
from teal.marshmallow import EnumField, IP, SanitizedStr, URL, Version from teal.marshmallow import IP, URL, EnumField, SanitizedStr, Version
from teal.resource import Schema from teal.resource import Schema
from ereuse_devicehub.marshmallow import NestedOn from ereuse_devicehub.marshmallow import NestedOn
@ -16,24 +30,32 @@ from ereuse_devicehub.resources import enums
from ereuse_devicehub.resources.action import models as m from ereuse_devicehub.resources.action import models as m
from ereuse_devicehub.resources.agent import schemas as s_agent from ereuse_devicehub.resources.agent import schemas as s_agent
from ereuse_devicehub.resources.device import schemas as s_device from ereuse_devicehub.resources.device import schemas as s_device
from ereuse_devicehub.resources.tradedocument import schemas as s_document
from ereuse_devicehub.resources.documents import schemas as s_generic_document from ereuse_devicehub.resources.documents import schemas as s_generic_document
from ereuse_devicehub.resources.enums import AppearanceRange, BiosAccessRange, FunctionalityRange, \ from ereuse_devicehub.resources.enums import (
PhysicalErasureMethod, R_POSITIVE, RatingRange, \ R_POSITIVE,
Severity, SnapshotSoftware, TestDataStorageLength AppearanceRange,
BiosAccessRange,
FunctionalityRange,
PhysicalErasureMethod,
RatingRange,
Severity,
SnapshotSoftware,
TestDataStorageLength,
)
from ereuse_devicehub.resources.models import STR_BIG_SIZE, STR_SIZE from ereuse_devicehub.resources.models import STR_BIG_SIZE, STR_SIZE
from ereuse_devicehub.resources.schemas import Thing from ereuse_devicehub.resources.schemas import Thing
from ereuse_devicehub.resources.tradedocument import schemas as s_document
from ereuse_devicehub.resources.tradedocument.models import TradeDocument
from ereuse_devicehub.resources.user import schemas as s_user from ereuse_devicehub.resources.user import schemas as s_user
from ereuse_devicehub.resources.user.models import User from ereuse_devicehub.resources.user.models import User
from ereuse_devicehub.resources.tradedocument.models import TradeDocument
class Action(Thing): class Action(Thing):
__doc__ = m.Action.__doc__ __doc__ = m.Action.__doc__
id = UUID(dump_only=True) id = UUID(dump_only=True)
name = SanitizedStr(default='', name = SanitizedStr(
validate=Length(max=STR_BIG_SIZE), default='', validate=Length(max=STR_BIG_SIZE), description=m.Action.name.comment
description=m.Action.name.comment) )
closed = Boolean(missing=True, description=m.Action.closed.comment) closed = Boolean(missing=True, description=m.Action.closed.comment)
severity = EnumField(Severity, description=m.Action.severity.comment) severity = EnumField(Severity, description=m.Action.severity.comment)
description = SanitizedStr(default='', description=m.Action.description.comment) description = SanitizedStr(default='', description=m.Action.description.comment)
@ -43,7 +65,9 @@ class Action(Thing):
agent = NestedOn(s_agent.Agent, description=m.Action.agent_id.comment) agent = NestedOn(s_agent.Agent, description=m.Action.agent_id.comment)
author = NestedOn(s_user.User, dump_only=True, exclude=('token',)) author = NestedOn(s_user.User, dump_only=True, exclude=('token',))
components = NestedOn(s_device.Component, dump_only=True, many=True) components = NestedOn(s_device.Component, dump_only=True, many=True)
parent = NestedOn(s_device.Computer, dump_only=True, description=m.Action.parent_id.comment) parent = NestedOn(
s_device.Computer, dump_only=True, description=m.Action.parent_id.comment
)
url = URL(dump_only=True, description=m.Action.url.__doc__) url = URL(dump_only=True, description=m.Action.url.__doc__)
@validates_schema @validates_schema
@ -67,24 +91,27 @@ class ActionWithOneDevice(Action):
class ActionWithMultipleDocuments(Action): class ActionWithMultipleDocuments(Action):
__doc__ = m.ActionWithMultipleTradeDocuments.__doc__ __doc__ = m.ActionWithMultipleTradeDocuments.__doc__
documents = NestedOn(s_document.TradeDocument, documents = NestedOn(
s_document.TradeDocument,
many=True, many=True,
required=True, # todo test ensuring len(devices) >= 1 required=True, # todo test ensuring len(devices) >= 1
only_query='id', only_query='id',
collection_class=OrderedSet) collection_class=OrderedSet,
)
class ActionWithMultipleDevices(Action): class ActionWithMultipleDevices(Action):
__doc__ = m.ActionWithMultipleDevices.__doc__ __doc__ = m.ActionWithMultipleDevices.__doc__
devices = NestedOn(s_device.Device, devices = NestedOn(
s_device.Device,
many=True, many=True,
required=True, # todo test ensuring len(devices) >= 1 required=True, # todo test ensuring len(devices) >= 1
only_query='id', only_query='id',
collection_class=OrderedSet) collection_class=OrderedSet,
)
class ActionWithMultipleDevicesCheckingOwner(ActionWithMultipleDevices): class ActionWithMultipleDevicesCheckingOwner(ActionWithMultipleDevices):
@post_load @post_load
def check_owner_of_device(self, data): def check_owner_of_device(self, data):
for dev in data['devices']: for dev in data['devices']:
@ -102,20 +129,29 @@ class Remove(ActionWithOneDevice):
class Allocate(ActionWithMultipleDevicesCheckingOwner): class Allocate(ActionWithMultipleDevicesCheckingOwner):
__doc__ = m.Allocate.__doc__ __doc__ = m.Allocate.__doc__
start_time = DateTime(data_key='startTime', required=True, start_time = DateTime(
description=m.Action.start_time.comment) data_key='startTime', required=True, description=m.Action.start_time.comment
end_time = DateTime(data_key='endTime', required=False, )
description=m.Action.end_time.comment) end_time = DateTime(
final_user_code = SanitizedStr(data_key="finalUserCode", data_key='endTime', required=False, description=m.Action.end_time.comment
)
final_user_code = SanitizedStr(
data_key="finalUserCode",
validate=Length(min=1, max=STR_BIG_SIZE), validate=Length(min=1, max=STR_BIG_SIZE),
required=False, required=False,
description='This is a internal code for mainteing the secrets of the \ description='This is a internal code for mainteing the secrets of the \
personal datas of the new holder') personal datas of the new holder',
transaction = SanitizedStr(validate=Length(min=1, max=STR_BIG_SIZE), )
transaction = SanitizedStr(
validate=Length(min=1, max=STR_BIG_SIZE),
required=False, required=False,
description='The code used from the owner for \ description='The code used from the owner for \
relation with external tool.') relation with external tool.',
end_users = Integer(data_key='endUsers', validate=[Range(min=1, error="Value must be greater than 0")]) )
end_users = Integer(
data_key='endUsers',
validate=[Range(min=1, error="Value must be greater than 0")],
)
@validates_schema @validates_schema
def validate_allocate(self, data: dict): def validate_allocate(self, data: dict):
@ -136,12 +172,15 @@ class Allocate(ActionWithMultipleDevicesCheckingOwner):
class Deallocate(ActionWithMultipleDevicesCheckingOwner): class Deallocate(ActionWithMultipleDevicesCheckingOwner):
__doc__ = m.Deallocate.__doc__ __doc__ = m.Deallocate.__doc__
start_time = DateTime(data_key='startTime', required=True, start_time = DateTime(
description=m.Action.start_time.comment) data_key='startTime', required=True, description=m.Action.start_time.comment
transaction = SanitizedStr(validate=Length(min=1, max=STR_BIG_SIZE), )
transaction = SanitizedStr(
validate=Length(min=1, max=STR_BIG_SIZE),
required=False, required=False,
description='The code used from the owner for \ description='The code used from the owner for \
relation with external tool.') relation with external tool.',
)
@validates_schema @validates_schema
def validate_deallocate(self, data: dict): def validate_deallocate(self, data: dict):
@ -232,7 +271,9 @@ class MeasureBattery(Test):
__doc__ = m.MeasureBattery.__doc__ __doc__ = m.MeasureBattery.__doc__
size = Integer(required=True, description=m.MeasureBattery.size.comment) size = Integer(required=True, description=m.MeasureBattery.size.comment)
voltage = Integer(required=True, description=m.MeasureBattery.voltage.comment) voltage = Integer(required=True, description=m.MeasureBattery.voltage.comment)
cycle_count = Integer(data_key='cycleCount', description=m.MeasureBattery.cycle_count.comment) cycle_count = Integer(
data_key='cycleCount', description=m.MeasureBattery.cycle_count.comment
)
health = EnumField(enums.BatteryHealth, description=m.MeasureBattery.health.comment) health = EnumField(enums.BatteryHealth, description=m.MeasureBattery.health.comment)
@ -289,28 +330,32 @@ class TestBios(Test):
class VisualTest(Test): class VisualTest(Test):
__doc__ = m.VisualTest.__doc__ __doc__ = m.VisualTest.__doc__
appearance_range = EnumField(AppearanceRange, data_key='appearanceRange') appearance_range = EnumField(AppearanceRange, data_key='appearanceRange')
functionality_range = EnumField(FunctionalityRange, functionality_range = EnumField(FunctionalityRange, data_key='functionalityRange')
data_key='functionalityRange')
labelling = Boolean() labelling = Boolean()
class Rate(ActionWithOneDevice): class Rate(ActionWithOneDevice):
__doc__ = m.Rate.__doc__ __doc__ = m.Rate.__doc__
rating = Integer(validate=Range(*R_POSITIVE), rating = Integer(
validate=Range(*R_POSITIVE), dump_only=True, description=m.Rate._rating.comment
)
version = Version(dump_only=True, description=m.Rate.version.comment)
appearance = Integer(
validate=Range(enums.R_NEGATIVE),
dump_only=True, dump_only=True,
description=m.Rate._rating.comment) description=m.Rate._appearance.comment,
version = Version(dump_only=True, )
description=m.Rate.version.comment) functionality = Integer(
appearance = Integer(validate=Range(enums.R_NEGATIVE), validate=Range(enums.R_NEGATIVE),
dump_only=True, dump_only=True,
description=m.Rate._appearance.comment) description=m.Rate._functionality.comment,
functionality = Integer(validate=Range(enums.R_NEGATIVE), )
dump_only=True, rating_range = EnumField(
description=m.Rate._functionality.comment) RatingRange,
rating_range = EnumField(RatingRange,
dump_only=True, dump_only=True,
data_key='ratingRange', data_key='ratingRange',
description=m.Rate.rating_range.__doc__) description=m.Rate.rating_range.__doc__,
)
class RateComputer(Rate): class RateComputer(Rate):
@ -320,19 +365,25 @@ class RateComputer(Rate):
data_storage = Float(dump_only=True, data_key='dataStorage') data_storage = Float(dump_only=True, data_key='dataStorage')
graphic_card = Float(dump_only=True, data_key='graphicCard') graphic_card = Float(dump_only=True, data_key='graphicCard')
data_storage_range = EnumField(RatingRange, dump_only=True, data_key='dataStorageRange') data_storage_range = EnumField(
RatingRange, dump_only=True, data_key='dataStorageRange'
)
ram_range = EnumField(RatingRange, dump_only=True, data_key='ramRange') ram_range = EnumField(RatingRange, dump_only=True, data_key='ramRange')
processor_range = EnumField(RatingRange, dump_only=True, data_key='processorRange') processor_range = EnumField(RatingRange, dump_only=True, data_key='processorRange')
graphic_card_range = EnumField(RatingRange, dump_only=True, data_key='graphicCardRange') graphic_card_range = EnumField(
RatingRange, dump_only=True, data_key='graphicCardRange'
)
class Price(ActionWithOneDevice): class Price(ActionWithOneDevice):
__doc__ = m.Price.__doc__ __doc__ = m.Price.__doc__
currency = EnumField(Currency, required=True, description=m.Price.currency.comment) currency = EnumField(Currency, required=True, description=m.Price.currency.comment)
price = Decimal(places=m.Price.SCALE, price = Decimal(
places=m.Price.SCALE,
rounding=m.Price.ROUND, rounding=m.Price.ROUND,
required=True, required=True,
description=m.Price.price.comment) description=m.Price.price.comment,
)
version = Version(dump_only=True, description=m.Price.version.comment) version = Version(dump_only=True, description=m.Price.version.comment)
rating = NestedOn(Rate, dump_only=True, description=m.Price.rating_id.comment) rating = NestedOn(Rate, dump_only=True, description=m.Price.rating_id.comment)
@ -356,13 +407,33 @@ class EreusePrice(Price):
class Install(ActionWithOneDevice): class Install(ActionWithOneDevice):
__doc__ = m.Install.__doc__ __doc__ = m.Install.__doc__
name = SanitizedStr(validate=Length(min=4, max=STR_BIG_SIZE), name = SanitizedStr(
validate=Length(min=4, max=STR_BIG_SIZE),
required=True, required=True,
description='The name of the OS installed.') description='The name of the OS installed.',
)
elapsed = TimeDelta(precision=TimeDelta.SECONDS, required=True) elapsed = TimeDelta(precision=TimeDelta.SECONDS, required=True)
address = Integer(validate=OneOf({8, 16, 32, 64, 128, 256})) address = Integer(validate=OneOf({8, 16, 32, 64, 128, 256}))
class Snapshot2(MarshmallowSchema):
uuid = UUID()
version = Version(required=True, description='The version of the software.')
type = String()
endTime = DateTime('iso', dump_only=True, description=m.Thing.updated.comment)
@validates_schema
def validate_workbench_version(self, data: dict):
if data['version'] < app.config['MIN_WORKBENCH']:
raise ValidationError(
'Min. supported Workbench version is '
'{} but yours is {}.'.format(
app.config['MIN_WORKBENCH'], data['version']
),
field_names=['version'],
)
class Snapshot(ActionWithOneDevice): class Snapshot(ActionWithOneDevice):
__doc__ = m.Snapshot.__doc__ __doc__ = m.Snapshot.__doc__
""" """
@ -372,18 +443,22 @@ class Snapshot(ActionWithOneDevice):
See docs for more info. See docs for more info.
""" """
uuid = UUID() uuid = UUID()
software = EnumField(SnapshotSoftware, software = EnumField(
SnapshotSoftware,
required=True, required=True,
description='The software that generated this Snapshot.') description='The software that generated this Snapshot.',
)
version = Version(required=True, description='The version of the software.') version = Version(required=True, description='The version of the software.')
actions = NestedOn(Action, many=True, dump_only=True) actions = NestedOn(Action, many=True, dump_only=True)
elapsed = TimeDelta(precision=TimeDelta.SECONDS) elapsed = TimeDelta(precision=TimeDelta.SECONDS)
components = NestedOn(s_device.Component, components = NestedOn(
s_device.Component,
many=True, many=True,
description='A list of components that are inside of the device' description='A list of components that are inside of the device'
'at the moment of this Snapshot.' 'at the moment of this Snapshot.'
'Order is preserved, so the component num 0 when' 'Order is preserved, so the component num 0 when'
'submitting is the component num 0 when returning it back.') 'submitting is the component num 0 when returning it back.',
)
@validates_schema @validates_schema
def validate_workbench_version(self, data: dict): def validate_workbench_version(self, data: dict):
@ -391,16 +466,21 @@ class Snapshot(ActionWithOneDevice):
if data['version'] < app.config['MIN_WORKBENCH']: if data['version'] < app.config['MIN_WORKBENCH']:
raise ValidationError( raise ValidationError(
'Min. supported Workbench version is ' 'Min. supported Workbench version is '
'{} but yours is {}.'.format(app.config['MIN_WORKBENCH'], data['version']), '{} but yours is {}.'.format(
field_names=['version'] app.config['MIN_WORKBENCH'], data['version']
),
field_names=['version'],
) )
@validates_schema @validates_schema
def validate_components_only_workbench(self, data: dict): def validate_components_only_workbench(self, data: dict):
if (data['software'] != SnapshotSoftware.Workbench) and (data['software'] != SnapshotSoftware.WorkbenchAndroid): if (data['software'] != SnapshotSoftware.Workbench) and (
data['software'] != SnapshotSoftware.WorkbenchAndroid
):
if data.get('components', None) is not None: if data.get('components', None) is not None:
raise ValidationError('Only Workbench can add component info', raise ValidationError(
field_names=['components']) 'Only Workbench can add component info', field_names=['components']
)
@validates_schema @validates_schema
def validate_only_workbench_fields(self, data: dict): def validate_only_workbench_fields(self, data: dict):
@ -408,22 +488,32 @@ class Snapshot(ActionWithOneDevice):
# todo test # todo test
if data['software'] == SnapshotSoftware.Workbench: if data['software'] == SnapshotSoftware.Workbench:
if not data.get('uuid', None): if not data.get('uuid', None):
raise ValidationError('Snapshots from Workbench and WorkbenchAndroid must have uuid', raise ValidationError(
field_names=['uuid']) 'Snapshots from Workbench and WorkbenchAndroid must have uuid',
field_names=['uuid'],
)
if data.get('elapsed', None) is None: if data.get('elapsed', None) is None:
raise ValidationError('Snapshots from Workbench must have elapsed', raise ValidationError(
field_names=['elapsed']) 'Snapshots from Workbench must have elapsed',
field_names=['elapsed'],
)
elif data['software'] == SnapshotSoftware.WorkbenchAndroid: elif data['software'] == SnapshotSoftware.WorkbenchAndroid:
if not data.get('uuid', None): if not data.get('uuid', None):
raise ValidationError('Snapshots from Workbench and WorkbenchAndroid must have uuid', raise ValidationError(
field_names=['uuid']) 'Snapshots from Workbench and WorkbenchAndroid must have uuid',
field_names=['uuid'],
)
else: else:
if data.get('uuid', None): if data.get('uuid', None):
raise ValidationError('Only Snapshots from Workbench or WorkbenchAndroid can have uuid', raise ValidationError(
field_names=['uuid']) 'Only Snapshots from Workbench or WorkbenchAndroid can have uuid',
field_names=['uuid'],
)
if data.get('elapsed', None): if data.get('elapsed', None):
raise ValidationError('Only Snapshots from Workbench can have elapsed', raise ValidationError(
field_names=['elapsed']) 'Only Snapshots from Workbench can have elapsed',
field_names=['elapsed'],
)
class ToRepair(ActionWithMultipleDevicesCheckingOwner): class ToRepair(ActionWithMultipleDevicesCheckingOwner):
@ -440,16 +530,20 @@ class Ready(ActionWithMultipleDevicesCheckingOwner):
class ActionStatus(Action): class ActionStatus(Action):
rol_user = NestedOn(s_user.User, dump_only=True, exclude=('token',)) rol_user = NestedOn(s_user.User, dump_only=True, exclude=('token',))
devices = NestedOn(s_device.Device, devices = NestedOn(
s_device.Device,
many=True, many=True,
required=False, # todo test ensuring len(devices) >= 1 required=False, # todo test ensuring len(devices) >= 1
only_query='id', only_query='id',
collection_class=OrderedSet) collection_class=OrderedSet,
documents = NestedOn(s_document.TradeDocument, )
documents = NestedOn(
s_document.TradeDocument,
many=True, many=True,
required=False, # todo test ensuring len(devices) >= 1 required=False, # todo test ensuring len(devices) >= 1
only_query='id', only_query='id',
collection_class=OrderedSet) collection_class=OrderedSet,
)
@pre_load @pre_load
def put_devices(self, data: dict): def put_devices(self, data: dict):
@ -508,20 +602,28 @@ class Live(ActionWithOneDevice):
See docs for more info. See docs for more info.
""" """
uuid = UUID() uuid = UUID()
software = EnumField(SnapshotSoftware, software = EnumField(
SnapshotSoftware,
required=True, required=True,
description='The software that generated this Snapshot.') description='The software that generated this Snapshot.',
)
version = Version(required=True, description='The version of the software.') version = Version(required=True, description='The version of the software.')
final_user_code = SanitizedStr(data_key="finalUserCode", dump_only=True) final_user_code = SanitizedStr(data_key="finalUserCode", dump_only=True)
licence_version = Version(required=True, description='The version of the software.') licence_version = Version(required=True, description='The version of the software.')
components = NestedOn(s_device.Component, components = NestedOn(
s_device.Component,
many=True, many=True,
description='A list of components that are inside of the device' description='A list of components that are inside of the device'
'at the moment of this Snapshot.' 'at the moment of this Snapshot.'
'Order is preserved, so the component num 0 when' 'Order is preserved, so the component num 0 when'
'submitting is the component num 0 when returning it back.') 'submitting is the component num 0 when returning it back.',
usage_time_allocate = TimeDelta(data_key='usageTimeAllocate', required=False, )
precision=TimeDelta.HOURS, dump_only=True) usage_time_allocate = TimeDelta(
data_key='usageTimeAllocate',
required=False,
precision=TimeDelta.HOURS,
dump_only=True,
)
class Organize(ActionWithMultipleDevices): class Organize(ActionWithMultipleDevices):
@ -691,26 +793,23 @@ class Trade(ActionWithMultipleDevices):
validate=Length(max=STR_SIZE), validate=Length(max=STR_SIZE),
data_key='userToEmail', data_key='userToEmail',
missing='', missing='',
required=False required=False,
) )
user_to = NestedOn(s_user.User, dump_only=True, data_key='userTo') user_to = NestedOn(s_user.User, dump_only=True, data_key='userTo')
user_from_email = SanitizedStr( user_from_email = SanitizedStr(
validate=Length(max=STR_SIZE), validate=Length(max=STR_SIZE),
data_key='userFromEmail', data_key='userFromEmail',
missing='', missing='',
required=False required=False,
) )
user_from = NestedOn(s_user.User, dump_only=True, data_key='userFrom') user_from = NestedOn(s_user.User, dump_only=True, data_key='userFrom')
code = SanitizedStr(validate=Length(max=STR_SIZE), data_key='code', required=False) code = SanitizedStr(validate=Length(max=STR_SIZE), data_key='code', required=False)
confirm = Boolean( confirm = Boolean(
data_key='confirms', data_key='confirms',
missing=True, missing=True,
description="""If you need confirmation of the user you need actevate this field""" description="""If you need confirmation of the user you need actevate this field""",
) )
lot = NestedOn('Lot', lot = NestedOn('Lot', many=False, required=True, only_query='id')
many=False,
required=True,
only_query='id')
@pre_load @pre_load
def adding_devices(self, data: dict): def adding_devices(self, data: dict):

39
ereuse_devicehub/resources/action/views/snapshot.py
View File

@ -1,18 +1,21 @@
""" This is the view for Snapshots """ """ This is the view for Snapshots """
import os
import json import json
import os
import shutil import shutil
from datetime import datetime from datetime import datetime
from flask import current_app as app, g from flask import current_app as app
from flask import g
from flask.json import jsonify
from sqlalchemy.util import OrderedSet from sqlalchemy.util import OrderedSet
from ereuse_devicehub.db import db from ereuse_devicehub.db import db
from ereuse_devicehub.resources.action.models import RateComputer, Snapshot from ereuse_devicehub.resources.action.models import RateComputer, Snapshot
from ereuse_devicehub.resources.device.models import Computer
from ereuse_devicehub.resources.action.rate.v1_0 import CannotRate from ereuse_devicehub.resources.action.rate.v1_0 import CannotRate
from ereuse_devicehub.resources.enums import SnapshotSoftware, Severity from ereuse_devicehub.resources.action.schemas import Snapshot2
from ereuse_devicehub.resources.device.models import Computer
from ereuse_devicehub.resources.enums import Severity, SnapshotSoftware
from ereuse_devicehub.resources.user.exceptions import InsufficientPermission from ereuse_devicehub.resources.user.exceptions import InsufficientPermission
@ -59,11 +62,12 @@ def move_json(tmp_snapshots, path_name, user, live=False):
os.remove(path_name) os.remove(path_name)
class SnapshotView(): class SnapshotView:
"""Performs a Snapshot. """Performs a Snapshot.
See `Snapshot` section in docs for more info. See `Snapshot` section in docs for more info.
""" """
# Note that if we set the device / components into the snapshot # Note that if we set the device / components into the snapshot
# model object, when we flush them to the db we will flush # model object, when we flush them to the db we will flush
# snapshot, and we want to wait to flush snapshot at the end # snapshot, and we want to wait to flush snapshot at the end
@ -74,6 +78,10 @@ class SnapshotView():
self.tmp_snapshots = app.config['TMP_SNAPSHOTS'] self.tmp_snapshots = app.config['TMP_SNAPSHOTS']
self.path_snapshot = save_json(snapshot_json, self.tmp_snapshots, g.user.email) self.path_snapshot = save_json(snapshot_json, self.tmp_snapshots, g.user.email)
snapshot_json.pop('debug', None) snapshot_json.pop('debug', None)
if snapshot_json.get('version') in ["14.0.0"]:
self.validate_json(snapshot_json)
self.response = self.build2()
else:
self.snapshot_json = resource_def.schema.load(snapshot_json) self.snapshot_json = resource_def.schema.load(snapshot_json)
self.response = self.build() self.response = self.build()
move_json(self.tmp_snapshots, self.path_snapshot, g.user.email) move_json(self.tmp_snapshots, self.path_snapshot, g.user.email)
@ -84,8 +92,12 @@ class SnapshotView():
def build(self): def build(self):
device = self.snapshot_json.pop('device') # type: Computer device = self.snapshot_json.pop('device') # type: Computer
components = None components = None
if self.snapshot_json['software'] == (SnapshotSoftware.Workbench or SnapshotSoftware.WorkbenchAndroid): if self.snapshot_json['software'] == (
components = self.snapshot_json.pop('components', None) # type: List[Component] SnapshotSoftware.Workbench or SnapshotSoftware.WorkbenchAndroid
):
components = self.snapshot_json.pop(
'components', None
) # type: List[Component]
if isinstance(device, Computer) and device.hid: if isinstance(device, Computer) and device.hid:
device.add_mac_to_hid(components_snap=components) device.add_mac_to_hid(components_snap=components)
snapshot = Snapshot(**self.snapshot_json) snapshot = Snapshot(**self.snapshot_json)
@ -94,7 +106,9 @@ class SnapshotView():
actions_device = set(e for e in device.actions_one) actions_device = set(e for e in device.actions_one)
device.actions_one.clear() device.actions_one.clear()
if components: if components:
actions_components = tuple(set(e for e in c.actions_one) for c in components) actions_components = tuple(
set(e for e in c.actions_one) for c in components
)
for component in components: for component in components:
component.actions_one.clear() component.actions_one.clear()
@ -141,3 +155,12 @@ class SnapshotView():
ret.status_code = 201 ret.status_code = 201
db.session.commit() db.session.commit()
return ret return ret
def validate_json(self, snapshot_json):
self.schema2 = Snapshot2()
self.snapshot_json = self.schema2.load(snapshot_json)
def build2(self):
res = jsonify("Ok")
res.status_code = 201
return res

264
tests/test_snapshot.py
View File

@ -1,38 +1,48 @@
import os
import json import json
import os
import shutil import shutil
import pytest
import uuid import uuid
from datetime import datetime, timedelta, timezone from datetime import datetime, timedelta, timezone
from requests.exceptions import HTTPError
from operator import itemgetter from operator import itemgetter
from typing import List, Tuple from typing import List, Tuple
from uuid import uuid4 from uuid import uuid4
import pytest
from boltons import urlutils from boltons import urlutils
from teal.db import UniqueViolation, DBError
from teal.marshmallow import ValidationError
from ereuse_utils.test import ANY from ereuse_utils.test import ANY
from requests.exceptions import HTTPError
from teal.db import DBError, UniqueViolation
from teal.marshmallow import ValidationError
from ereuse_devicehub.client import UserClient from ereuse_devicehub.client import UserClient
from ereuse_devicehub.db import db from ereuse_devicehub.db import db
from ereuse_devicehub.devicehub import Devicehub from ereuse_devicehub.devicehub import Devicehub
from ereuse_devicehub.resources.action.models import Action, BenchmarkDataStorage, \ from ereuse_devicehub.resources.action.models import (
BenchmarkProcessor, EraseSectors, RateComputer, Snapshot, SnapshotRequest, VisualTest, \ Action,
EreusePrice, Ready BenchmarkDataStorage,
BenchmarkProcessor,
EraseSectors,
EreusePrice,
RateComputer,
Ready,
Snapshot,
SnapshotRequest,
VisualTest,
)
from ereuse_devicehub.resources.action.views.snapshot import save_json
from ereuse_devicehub.resources.device import models as m from ereuse_devicehub.resources.device import models as m
from ereuse_devicehub.resources.device.exceptions import NeedsId from ereuse_devicehub.resources.device.exceptions import NeedsId
from ereuse_devicehub.resources.device.models import SolidStateDrive from ereuse_devicehub.resources.device.models import SolidStateDrive
from ereuse_devicehub.resources.device.sync import MismatchBetweenProperties, \ from ereuse_devicehub.resources.device.sync import (
MismatchBetweenTagsAndHid MismatchBetweenProperties,
MismatchBetweenTagsAndHid,
)
from ereuse_devicehub.resources.documents import documents
from ereuse_devicehub.resources.enums import ComputerChassis, SnapshotSoftware from ereuse_devicehub.resources.enums import ComputerChassis, SnapshotSoftware
from ereuse_devicehub.resources.tag import Tag from ereuse_devicehub.resources.tag import Tag
from ereuse_devicehub.resources.user.models import User from ereuse_devicehub.resources.user.models import User
from ereuse_devicehub.resources.action.views.snapshot import save_json
from ereuse_devicehub.resources.documents import documents
from tests.conftest import file, yaml2json, json_encode
from tests import conftest from tests import conftest
from tests.conftest import file, json_encode, yaml2json
@pytest.mark.mvp @pytest.mark.mvp
@ -43,18 +53,22 @@ def test_snapshot_model():
""" """
device = m.Desktop(serial_number='a1', chassis=ComputerChassis.Tower) device = m.Desktop(serial_number='a1', chassis=ComputerChassis.Tower)
# noinspection PyArgumentList # noinspection PyArgumentList
snapshot = Snapshot(uuid=uuid4(), snapshot = Snapshot(
uuid=uuid4(),
end_time=datetime.now(timezone.utc), end_time=datetime.now(timezone.utc),
version='1.0', version='1.0',
software=SnapshotSoftware.DesktopApp, software=SnapshotSoftware.DesktopApp,
elapsed=timedelta(seconds=25)) elapsed=timedelta(seconds=25),
)
snapshot.device = device snapshot.device = device
snapshot.request = SnapshotRequest(request={'foo': 'bar'}) snapshot.request = SnapshotRequest(request={'foo': 'bar'})
db.session.add(snapshot) db.session.add(snapshot)
db.session.commit() db.session.commit()
device = m.Desktop.query.one() # type: m.Desktop device = m.Desktop.query.one() # type: m.Desktop
e1 = device.actions[0] e1 = device.actions[0]
assert isinstance(e1, Snapshot), 'Creation order must be preserved: 1. snapshot, 2. WR' assert isinstance(
e1, Snapshot
), 'Creation order must be preserved: 1. snapshot, 2. WR'
db.session.delete(device) db.session.delete(device)
db.session.commit() db.session.commit()
assert Snapshot.query.one_or_none() is None assert Snapshot.query.one_or_none() is None
@ -63,7 +77,9 @@ def test_snapshot_model():
assert m.Desktop.query.one_or_none() is None assert m.Desktop.query.one_or_none() is None
assert m.Device.query.one_or_none() is None assert m.Device.query.one_or_none() is None
# Check properties # Check properties
assert device.url == urlutils.URL('http://localhost/devices/%s' % device.devicehub_id) assert device.url == urlutils.URL(
'http://localhost/devices/%s' % device.devicehub_id
)
@pytest.mark.mvp @pytest.mark.mvp
@ -78,13 +94,12 @@ def test_snapshot_post(user: UserClient):
"""Tests the post snapshot endpoint (validation, etc), data correctness, """Tests the post snapshot endpoint (validation, etc), data correctness,
and relationship correctness. and relationship correctness.
""" """
snapshot = snapshot_and_check(user, yaml2json('basic.snapshot'), snapshot = snapshot_and_check(
action_types=( user,
BenchmarkProcessor.t, yaml2json('basic.snapshot'),
VisualTest.t, action_types=(BenchmarkProcessor.t, VisualTest.t, RateComputer.t),
RateComputer.t perform_second_snapshot=False,
), )
perform_second_snapshot=False)
assert snapshot['software'] == 'Workbench' assert snapshot['software'] == 'Workbench'
assert snapshot['version'] == '11.0' assert snapshot['version'] == '11.0'
assert snapshot['uuid'] == 'f5efd26e-8754-46bc-87bf-fbccc39d60d9' assert snapshot['uuid'] == 'f5efd26e-8754-46bc-87bf-fbccc39d60d9'
@ -98,8 +113,11 @@ def test_snapshot_post(user: UserClient):
device['components'].sort(key=key) device['components'].sort(key=key)
assert snapshot['components'] == device['components'] assert snapshot['components'] == device['components']
assert {c['type'] for c in snapshot['components']} == {m.GraphicCard.t, m.RamModule.t, assert {c['type'] for c in snapshot['components']} == {
m.Processor.t} m.GraphicCard.t,
m.RamModule.t,
m.Processor.t,
}
rate = next(e for e in snapshot['actions'] if e['type'] == RateComputer.t) rate = next(e for e in snapshot['actions'] if e['type'] == RateComputer.t)
rate, _ = user.get(res=Action, item=rate['id']) rate, _ = user.get(res=Action, item=rate['id'])
assert rate['device']['id'] == snapshot['device']['id'] assert rate['device']['id'] == snapshot['device']['id']
@ -127,20 +145,26 @@ def test_same_device_tow_users(user: UserClient, user2: UserClient):
assert pc['ownerID'] != pc2['ownerID'] assert pc['ownerID'] != pc2['ownerID']
assert pc['hid'] == pc2['hid'] assert pc['hid'] == pc2['hid']
@pytest.mark.mvp @pytest.mark.mvp
def test_snapshot_update_timefield_updated(user: UserClient): def test_snapshot_update_timefield_updated(user: UserClient):
""" """
Tests for check if one computer have the time mark updated when one component of it is updated Tests for check if one computer have the time mark updated when one component of it is updated
""" """
computer1 = yaml2json('1-device-with-components.snapshot') computer1 = yaml2json('1-device-with-components.snapshot')
snapshot = snapshot_and_check(user, snapshot = snapshot_and_check(
user,
computer1, computer1,
action_types=(BenchmarkProcessor.t, action_types=(BenchmarkProcessor.t, RateComputer.t),
RateComputer.t), perform_second_snapshot=False,
perform_second_snapshot=False) )
computer2 = yaml2json('2-second-device-with-components-of-first.snapshot') computer2 = yaml2json('2-second-device-with-components-of-first.snapshot')
snapshot_and_check(user, computer2, action_types=('Remove', 'RateComputer'), snapshot_and_check(
perform_second_snapshot=False) user,
computer2,
action_types=('Remove', 'RateComputer'),
perform_second_snapshot=False,
)
pc1_devicehub_id = snapshot['device']['devicehubID'] pc1_devicehub_id = snapshot['device']['devicehubID']
pc1, _ = user.get(res=m.Device, item=pc1_devicehub_id) pc1, _ = user.get(res=m.Device, item=pc1_devicehub_id)
assert pc1['updated'] != snapshot['device']['updated'] assert pc1['updated'] != snapshot['device']['updated']
@ -165,7 +189,10 @@ def test_snapshot_power_on_hours(user: UserClient):
test_data_storage = ac test_data_storage = ac
break break
assert test_data_storage.lifetime.total_seconds()/3600 == test_data_storage.power_on_hours assert (
test_data_storage.lifetime.total_seconds() / 3600
== test_data_storage.power_on_hours
)
@pytest.mark.mvp @pytest.mark.mvp
@ -176,10 +203,7 @@ def test_snapshot_component_add_remove(user: UserClient):
def get_actions_info(actions: List[dict]) -> tuple: def get_actions_info(actions: List[dict]) -> tuple:
return tuple( return tuple(
( (e['type'], [c['serialNumber'] for c in e['components']])
e['type'],
[c['serialNumber'] for c in e['components']]
)
for e in user.get_many(res=Action, resources=actions, key='id') for e in user.get_many(res=Action, resources=actions, key='id')
) )
@ -198,7 +222,11 @@ def test_snapshot_component_add_remove(user: UserClient):
pc1, _ = user.get(res=m.Device, item=pc1_devicehub_id) pc1, _ = user.get(res=m.Device, item=pc1_devicehub_id)
update1_pc1 = pc1['updated'] update1_pc1 = pc1['updated']
# Parent contains components # Parent contains components
assert tuple(c['serialNumber'] for c in pc1['components']) == ('p1c1s', 'p1c2s', 'p1c3s') assert tuple(c['serialNumber'] for c in pc1['components']) == (
'p1c1s',
'p1c2s',
'p1c3s',
)
# Components contain parent # Components contain parent
assert all(c['parent'] == pc1_id for c in pc1['components']) assert all(c['parent'] == pc1_id for c in pc1['components'])
# pc has three actions: Snapshot, BenchmarkProcessor and RateComputer # pc has three actions: Snapshot, BenchmarkProcessor and RateComputer
@ -228,7 +256,12 @@ def test_snapshot_component_add_remove(user: UserClient):
# PC1 # PC1
assert tuple(c['serialNumber'] for c in pc1['components']) == ('p1c1s', 'p1c3s') assert tuple(c['serialNumber'] for c in pc1['components']) == ('p1c1s', 'p1c3s')
assert all(c['parent'] == pc1_id for c in pc1['components']) assert all(c['parent'] == pc1_id for c in pc1['components'])
assert tuple(e['type'] for e in pc1['actions']) == ('BenchmarkProcessor', 'Snapshot', 'RateComputer', 'Remove') assert tuple(e['type'] for e in pc1['actions']) == (
'BenchmarkProcessor',
'Snapshot',
'RateComputer',
'Remove',
)
# PC2 # PC2
assert tuple(c['serialNumber'] for c in pc2['components']) == ('p1c2s', 'p2c1s') assert tuple(c['serialNumber'] for c in pc2['components']) == ('p1c2s', 'p2c1s')
assert all(c['parent'] == pc2_id for c in pc2['components']) assert all(c['parent'] == pc2_id for c in pc2['components'])
@ -236,15 +269,24 @@ def test_snapshot_component_add_remove(user: UserClient):
# p1c2s has two Snapshots, a Remove and an Add # p1c2s has two Snapshots, a Remove and an Add
p1c2s, _ = user.get(res=m.Device, item=pc2['components'][0]['devicehubID']) p1c2s, _ = user.get(res=m.Device, item=pc2['components'][0]['devicehubID'])
assert tuple(e['type'] for e in p1c2s['actions']) == ( assert tuple(e['type'] for e in p1c2s['actions']) == (
'BenchmarkProcessor', 'Snapshot', 'RateComputer', 'Snapshot', 'Remove', 'RateComputer' 'BenchmarkProcessor',
'Snapshot',
'RateComputer',
'Snapshot',
'Remove',
'RateComputer',
) )
# We register the first device again, but removing motherboard # We register the first device again, but removing motherboard
# and moving processor from the second device to the first. # and moving processor from the second device to the first.
# We have created 1 Remove (from PC2's processor back to PC1) # We have created 1 Remove (from PC2's processor back to PC1)
# PC 0: p1c2s, p1c3s. PC 1: p2c1s # PC 0: p1c2s, p1c3s. PC 1: p2c1s
s3 = yaml2json('3-first-device-but-removing-motherboard-and-adding-processor-from-2.snapshot') s3 = yaml2json(
snapshot_and_check(user, s3, ('Remove', 'RateComputer'), perform_second_snapshot=False) '3-first-device-but-removing-motherboard-and-adding-processor-from-2.snapshot'
)
snapshot_and_check(
user, s3, ('Remove', 'RateComputer'), perform_second_snapshot=False
)
pc1, _ = user.get(res=m.Device, item=pc1_devicehub_id) pc1, _ = user.get(res=m.Device, item=pc1_devicehub_id)
pc2, _ = user.get(res=m.Device, item=pc2_devicehub_id) pc2, _ = user.get(res=m.Device, item=pc2_devicehub_id)
# Check if the update_timestamp is updated # Check if the update_timestamp is updated
@ -263,7 +305,7 @@ def test_snapshot_component_add_remove(user: UserClient):
('RateComputer', ['p1c1s', 'p1c2s', 'p1c3s']), ('RateComputer', ['p1c1s', 'p1c2s', 'p1c3s']),
('Remove', ['p1c2s']), # Remove Processor in Snapshot2 ('Remove', ['p1c2s']), # Remove Processor in Snapshot2
('Snapshot', ['p1c2s', 'p1c3s']), # This Snapshot3 ('Snapshot', ['p1c2s', 'p1c3s']), # This Snapshot3
('RateComputer', ['p1c2s', 'p1c3s']) ('RateComputer', ['p1c2s', 'p1c3s']),
) )
# PC2 # PC2
assert tuple(c['serialNumber'] for c in pc2['components']) == ('p2c1s',) assert tuple(c['serialNumber'] for c in pc2['components']) == ('p2c1s',)
@ -271,7 +313,7 @@ def test_snapshot_component_add_remove(user: UserClient):
assert tuple(e['type'] for e in pc2['actions']) == ( assert tuple(e['type'] for e in pc2['actions']) == (
'Snapshot', # Second Snapshot 'Snapshot', # Second Snapshot
'RateComputer', 'RateComputer',
'Remove' # the processor we added in 2. 'Remove', # the processor we added in 2.
) )
# p1c2s has Snapshot, Remove and Add # p1c2s has Snapshot, Remove and Add
p1c2s, _ = user.get(res=m.Device, item=pc1['components'][0]['devicehubID']) p1c2s, _ = user.get(res=m.Device, item=pc1['components'][0]['devicehubID'])
@ -284,13 +326,17 @@ def test_snapshot_component_add_remove(user: UserClient):
('RateComputer', ['p1c2s', 'p2c1s']), ('RateComputer', ['p1c2s', 'p2c1s']),
('Snapshot', ['p1c2s', 'p1c3s']), # The third Snapshot to PC1 ('Snapshot', ['p1c2s', 'p1c3s']), # The third Snapshot to PC1
('Remove', ['p1c2s']), # ...which caused p1c2 to be removed from PC2 ('Remove', ['p1c2s']), # ...which caused p1c2 to be removed from PC2
('RateComputer', ['p1c2s', 'p1c3s']) ('RateComputer', ['p1c2s', 'p1c3s']),
) )
# We register the first device but without the processor, # We register the first device but without the processor,
# adding a graphic card and adding a new component # adding a graphic card and adding a new component
s4 = yaml2json('4-first-device-but-removing-processor.snapshot-and-adding-graphic-card') s4 = yaml2json(
snapshot4 = snapshot_and_check(user, s4, ('RateComputer',), perform_second_snapshot=False) '4-first-device-but-removing-processor.snapshot-and-adding-graphic-card'
)
snapshot4 = snapshot_and_check(
user, s4, ('RateComputer',), perform_second_snapshot=False
)
pc1, _ = user.get(res=m.Device, item=pc1_devicehub_id) pc1, _ = user.get(res=m.Device, item=pc1_devicehub_id)
pc2, _ = user.get(res=m.Device, item=pc2_devicehub_id) pc2, _ = user.get(res=m.Device, item=pc2_devicehub_id)
# Check if the update_timestamp is updated # Check if the update_timestamp is updated
@ -308,6 +354,7 @@ def test_snapshot_component_add_remove(user: UserClient):
assert tuple(c['serialNumber'] for c in pc2['components']) == ('p2c1s',) assert tuple(c['serialNumber'] for c in pc2['components']) == ('p2c1s',)
assert all(c['parent'] == pc2_id for c in pc2['components']) assert all(c['parent'] == pc2_id for c in pc2['components'])
@pytest.mark.mvp @pytest.mark.mvp
def test_snapshot_post_without_hid(user: UserClient): def test_snapshot_post_without_hid(user: UserClient):
"""Tests the post snapshot endpoint (validation, etc), data correctness, """Tests the post snapshot endpoint (validation, etc), data correctness,
@ -338,15 +385,18 @@ def test_snapshot_tag_inner_tag(user: UserClient, tag_id: str, app: Devicehub):
b = yaml2json('basic.snapshot') b = yaml2json('basic.snapshot')
b['device']['tags'] = [{'type': 'Tag', 'id': tag_id}] b['device']['tags'] = [{'type': 'Tag', 'id': tag_id}]
snapshot_and_check(user, b, snapshot_and_check(
action_types=(RateComputer.t, BenchmarkProcessor.t, VisualTest.t)) user, b, action_types=(RateComputer.t, BenchmarkProcessor.t, VisualTest.t)
)
with app.app_context(): with app.app_context():
tag = Tag.query.all()[0] # type: Tag tag = Tag.query.all()[0] # type: Tag
assert tag.device_id == 3, 'Tag should be linked to the first device' assert tag.device_id == 3, 'Tag should be linked to the first device'
@pytest.mark.mvp @pytest.mark.mvp
def test_snapshot_tag_inner_tag_mismatch_between_tags_and_hid(user: UserClient, tag_id: str): def test_snapshot_tag_inner_tag_mismatch_between_tags_and_hid(
user: UserClient, tag_id: str
):
"""Ensures one device cannot 'steal' the tag from another one.""" """Ensures one device cannot 'steal' the tag from another one."""
pc1 = yaml2json('basic.snapshot') pc1 = yaml2json('basic.snapshot')
pc1['device']['tags'] = [{'type': 'Tag', 'id': tag_id}] pc1['device']['tags'] = [{'type': 'Tag', 'id': tag_id}]
@ -396,7 +446,7 @@ def test_snapshot_component_containing_components(user: UserClient):
'type': 'Processor', 'type': 'Processor',
'serialNumber': 'foo', 'serialNumber': 'foo',
'manufacturer': 'bar', 'manufacturer': 'bar',
'model': 'baz' 'model': 'baz',
} }
user.post(json_encode(s), res=Snapshot, status=ValidationError) user.post(json_encode(s), res=Snapshot, status=ValidationError)
@ -435,13 +485,15 @@ def test_not_remove_ram_in_same_computer(user: UserClient):
snap1, _ = user.post(json_encode(s), res=Snapshot) snap1, _ = user.post(json_encode(s), res=Snapshot)
s['uuid'] = '74caa7eb-2bad-4333-94f6-6f1b031d0774' s['uuid'] = '74caa7eb-2bad-4333-94f6-6f1b031d0774'
s['components'].append({ s['components'].append(
{
"actions": [], "actions": [],
"manufacturer": "Intel Corporation", "manufacturer": "Intel Corporation",
"model": "NM10/ICH7 Family High Definition Audio Controller", "model": "NM10/ICH7 Family High Definition Audio Controller",
"serialNumber": "mp2pc", "serialNumber": "mp2pc",
"type": "SoundCard" "type": "SoundCard",
}) }
)
dev1 = m.Device.query.filter_by(id=snap1['device']['id']).one() dev1 = m.Device.query.filter_by(id=snap1['device']['id']).one()
ram1 = [x.id for x in dev1.components if x.type == 'RamModule'][0] ram1 = [x.id for x in dev1.components if x.type == 'RamModule'][0]
snap2, _ = user.post(json_encode(s), res=Snapshot) snap2, _ = user.post(json_encode(s), res=Snapshot)
@ -466,13 +518,18 @@ def test_ereuse_price(user: UserClient):
s = yaml2json('erase-sectors.snapshot') s = yaml2json('erase-sectors.snapshot')
assert s['components'][0]['actions'][0]['endTime'] == '2018-06-01T09:12:06+02:00' assert s['components'][0]['actions'][0]['endTime'] == '2018-06-01T09:12:06+02:00'
s['device']['type'] = 'Server' s['device']['type'] = 'Server'
snapshot = snapshot_and_check(user, s, action_types=( snapshot = snapshot_and_check(
user,
s,
action_types=(
EraseSectors.t, EraseSectors.t,
BenchmarkDataStorage.t, BenchmarkDataStorage.t,
BenchmarkProcessor.t, BenchmarkProcessor.t,
RateComputer.t, RateComputer.t,
EreusePrice.t EreusePrice.t,
), perform_second_snapshot=False) ),
perform_second_snapshot=False,
)
ereuse_price = snapshot['actions'][-1] ereuse_price = snapshot['actions'][-1]
assert len(ereuse_price) > 0 assert len(ereuse_price) > 0
@ -487,33 +544,54 @@ def test_erase_privacy_standards_endtime_sort(user: UserClient):
""" """
s = yaml2json('erase-sectors.snapshot') s = yaml2json('erase-sectors.snapshot')
assert s['components'][0]['actions'][0]['endTime'] == '2018-06-01T09:12:06+02:00' assert s['components'][0]['actions'][0]['endTime'] == '2018-06-01T09:12:06+02:00'
snapshot = snapshot_and_check(user, s, action_types=( snapshot = snapshot_and_check(
user,
s,
action_types=(
EraseSectors.t, EraseSectors.t,
BenchmarkDataStorage.t, BenchmarkDataStorage.t,
BenchmarkProcessor.t, BenchmarkProcessor.t,
RateComputer.t, RateComputer.t,
EreusePrice.t EreusePrice.t,
), perform_second_snapshot=False) ),
perform_second_snapshot=False,
)
# Perform a new snapshot changing the erasure time, as if # Perform a new snapshot changing the erasure time, as if
# it is a new erasure performed after. # it is a new erasure performed after.
erase = next(e for e in snapshot['actions'] if e['type'] == EraseSectors.t) erase = next(e for e in snapshot['actions'] if e['type'] == EraseSectors.t)
assert erase['endTime'] == '2018-06-01T07:12:06+00:00' assert erase['endTime'] == '2018-06-01T07:12:06+00:00'
s['uuid'] = uuid4() s['uuid'] = uuid4()
s['components'][0]['actions'][0]['endTime'] = '2018-06-01T07:14:00+00:00' s['components'][0]['actions'][0]['endTime'] = '2018-06-01T07:14:00+00:00'
snapshot = snapshot_and_check(user, s, action_types=( snapshot = snapshot_and_check(
user,
s,
action_types=(
EraseSectors.t, EraseSectors.t,
BenchmarkDataStorage.t, BenchmarkDataStorage.t,
BenchmarkProcessor.t, BenchmarkProcessor.t,
RateComputer.t, RateComputer.t,
EreusePrice.t EreusePrice.t,
), perform_second_snapshot=False) ),
perform_second_snapshot=False,
)
# The actual test # The actual test
storage = next(e for e in snapshot['components'] if e['type'] == SolidStateDrive.t) storage = next(e for e in snapshot['components'] if e['type'] == SolidStateDrive.t)
storage, _ = user.get(res=m.Device, item=storage['devicehubID']) # Let's get storage actions too storage, _ = user.get(
res=m.Device, item=storage['devicehubID']
) # Let's get storage actions too
# order: endTime ascending # order: endTime ascending
# erasure1/2 have an user defined time and others actions endTime = created # erasure1/2 have an user defined time and others actions endTime = created
erasure1, erasure2, benchmark_hdd1, _snapshot1, _, _, benchmark_hdd2, _snapshot2 = storage['actions'][:8] (
erasure1,
erasure2,
benchmark_hdd1,
_snapshot1,
_,
_,
benchmark_hdd2,
_snapshot2,
) = storage['actions'][:8]
assert erasure1['type'] == erasure2['type'] == 'EraseSectors' assert erasure1['type'] == erasure2['type'] == 'EraseSectors'
assert benchmark_hdd1['type'] == benchmark_hdd2['type'] == 'BenchmarkDataStorage' assert benchmark_hdd1['type'] == benchmark_hdd2['type'] == 'BenchmarkDataStorage'
assert _snapshot1['type'] == _snapshot2['type'] == 'Snapshot' assert _snapshot1['type'] == _snapshot2['type'] == 'Snapshot'
@ -555,8 +633,7 @@ def test_test_data_storage(user: UserClient):
s = file('erase-sectors-2-hdd.snapshot') s = file('erase-sectors-2-hdd.snapshot')
snapshot, _ = user.post(res=Snapshot, data=s) snapshot, _ = user.post(res=Snapshot, data=s)
incidence_test = next( incidence_test = next(
ev for ev in snapshot['actions'] ev for ev in snapshot['actions'] if ev.get('reallocatedSectorCount', None) == 15
if ev.get('reallocatedSectorCount', None) == 15
) )
assert incidence_test['severity'] == 'Error' assert incidence_test['severity'] == 'Error'
@ -584,10 +661,12 @@ def assert_similar_components(components1: List[dict], components2: List[dict]):
assert_similar_device(c1, c2) assert_similar_device(c1, c2)
def snapshot_and_check(user: UserClient, def snapshot_and_check(
user: UserClient,
input_snapshot: dict, input_snapshot: dict,
action_types: Tuple[str, ...] = tuple(), action_types: Tuple[str, ...] = tuple(),
perform_second_snapshot=True) -> dict: perform_second_snapshot=True,
) -> dict:
"""Performs a Snapshot and then checks if the result is ok: """Performs a Snapshot and then checks if the result is ok:
- There have been performed the types of actions and in the same - There have been performed the types of actions and in the same
@ -610,18 +689,22 @@ def snapshot_and_check(user: UserClient,
if action['type'] == 'Add': if action['type'] == 'Add':
found_add = True found_add = True
if found_add: if found_add:
assert action['type'] != 'Receive', 'All Remove actions must be before the Add ones' assert (
action['type'] != 'Receive'
), 'All Remove actions must be before the Add ones'
assert input_snapshot['device'] assert input_snapshot['device']
assert_similar_device(input_snapshot['device'], snapshot['device']) assert_similar_device(input_snapshot['device'], snapshot['device'])
if input_snapshot.get('components', None): if input_snapshot.get('components', None):
assert_similar_components(input_snapshot['components'], snapshot['components']) assert_similar_components(input_snapshot['components'], snapshot['components'])
assert all(c['parent'] == snapshot['device']['id'] for c in snapshot['components']), \ assert all(
'Components must be in their parent' c['parent'] == snapshot['device']['id'] for c in snapshot['components']
), 'Components must be in their parent'
if perform_second_snapshot: if perform_second_snapshot:
if 'uuid' in input_snapshot: if 'uuid' in input_snapshot:
input_snapshot['uuid'] = uuid4() input_snapshot['uuid'] = uuid4()
return snapshot_and_check(user, input_snapshot, action_types, return snapshot_and_check(
perform_second_snapshot=False) user, input_snapshot, action_types, perform_second_snapshot=False
)
else: else:
return snapshot return snapshot
@ -642,12 +725,12 @@ def test_erase_changing_hdd_between_pcs(user: UserClient):
db.session.commit() db.session.commit()
assert dev2.components[1].actions[2].parent == dev1 assert dev2.components[1].actions[2].parent == dev1
doc1, response = user.get(res=documents.DocumentDef.t, doc1, response = user.get(
item='erasures/{}'.format(dev1.id), res=documents.DocumentDef.t, item='erasures/{}'.format(dev1.id), accept=ANY
accept=ANY) )
doc2, response = user.get(res=documents.DocumentDef.t, doc2, response = user.get(
item='erasures/{}'.format(dev2.id), res=documents.DocumentDef.t, item='erasures/{}'.format(dev2.id), accept=ANY
accept=ANY) )
assert 'dev1' in doc2 assert 'dev1' in doc2
assert 'dev2' in doc2 assert 'dev2' in doc2
@ -667,7 +750,6 @@ def test_pc_2(user: UserClient):
snapshot, _ = user.post(res=Snapshot, data=s) snapshot, _ = user.post(res=Snapshot, data=s)
@pytest.mark.mvp @pytest.mark.mvp
def test_save_snapshot_in_file(app: Devicehub, user: UserClient): def test_save_snapshot_in_file(app: Devicehub, user: UserClient):
"""This test check if works the function save_snapshot_in_file""" """This test check if works the function save_snapshot_in_file"""
@ -712,6 +794,7 @@ def test_action_no_snapshot_without_save_file(app: Devicehub, user: UserClient):
assert os.path.exists(tmp_snapshots) == False assert os.path.exists(tmp_snapshots) == False
@pytest.mark.mvp @pytest.mark.mvp
def test_save_snapshot_with_debug(app: Devicehub, user: UserClient): def test_save_snapshot_with_debug(app: Devicehub, user: UserClient):
"""This test check if works the function save_snapshot_in_file""" """This test check if works the function save_snapshot_in_file"""
@ -870,6 +953,7 @@ def test_snapshot_failed_end_time_bug(app: Devicehub, user: UserClient):
tmp_snapshots = app.config['TMP_SNAPSHOTS'] tmp_snapshots = app.config['TMP_SNAPSHOTS']
shutil.rmtree(tmp_snapshots) shutil.rmtree(tmp_snapshots)
@pytest.mark.mvp @pytest.mark.mvp
def test_snapshot_not_failed_end_time_bug(app: Devicehub, user: UserClient): def test_snapshot_not_failed_end_time_bug(app: Devicehub, user: UserClient):
"""This test check if the end_time != 0001-01-01 00:00:00+00:00 """This test check if the end_time != 0001-01-01 00:00:00+00:00
@ -926,3 +1010,19 @@ def test_bug_141(user: UserClient):
""" """
dev = file('2021-5-4-13-41_time_out_test_datastorage') dev = file('2021-5-4-13-41_time_out_test_datastorage')
user.post(dev, res=Snapshot) user.post(dev, res=Snapshot)
@pytest.mark.mvp
def test_min_validate_fields(user: UserClient):
"""This test check the minimum validation of json that come from snapshot"""
snapshot = {
"type": "Snapshot",
"uuid": "d1b70cb8-8929-4f36-99b7-fe052cec0abd",
"version": "14.0.0",
"endTime": "2016-11-03T17:17:17.266543+00:00",
"dmidecode": '',
"smartctl": '',
}
body, res = user.post(snapshot, res=Snapshot)
assert body == 'Ok'
assert res.status == '201 CREATED'