from datetime import datetime, timedelta import pytest from flask import g from sqlalchemy.util import OrderedSet from ereuse_devicehub.db import db from ereuse_devicehub.resources.device.models import Computer, Device, GraphicCard, HardDrive, \ RamModule, SolidStateDrive from ereuse_devicehub.resources.enums import TestHardDriveLength from ereuse_devicehub.resources.event.models import BenchmarkDataStorage, EraseBasic, EraseSectors, \ EventWithOneDevice, Install, Ready, StepZero, StressTest, TestDataStorage from tests.conftest import create_user @pytest.mark.usefixtures('app_context') def test_author(): """ Checks the default created author. Note that the author can be accessed after inserting the row. """ user = create_user() g.user = user e = EventWithOneDevice(device=Device()) db.session.add(e) assert e.author is None assert e.author_id is None db.session.commit() assert e.author == user @pytest.mark.usefixtures('auth_app_context') def test_erase_basic(): erasure = EraseBasic( device=HardDrive(serial_number='foo', manufacturer='bar', model='foo-bar'), clean_with_zeros=True, start_time=datetime.now(), end_time=datetime.now(), secure_random_steps=25, error=False ) db.session.add(erasure) db.session.commit() db_erasure = EraseBasic.query.one() assert erasure == db_erasure assert next(iter(db_erasure.device.events)) == erasure @pytest.mark.usefixtures('auth_app_context') def test_validate_device_data_storage(): """Checks the validation for data-storage-only events works.""" # We can't set a GraphicCard with pytest.raises(TypeError, message='EraseBasic.device must be a DataStorage ' 'but you passed '): EraseBasic( device=GraphicCard(serial_number='foo', manufacturer='bar', model='foo-bar'), clean_with_zeros=True, start_time=datetime.now(), end_time=datetime.now(), secure_random_steps=25, error=False ) @pytest.mark.usefixtures('auth_app_context') def test_erase_sectors_steps(): erasure = EraseSectors( device=SolidStateDrive(serial_number='foo', manufacturer='bar', model='foo-bar'), clean_with_zeros=True, start_time=datetime.now(), end_time=datetime.now(), secure_random_steps=25, error=False, steps=[ StepZero(error=False, start_time=datetime.now(), end_time=datetime.now(), secure_random_steps=1, clean_with_zeros=True), StepZero(error=False, start_time=datetime.now(), end_time=datetime.now(), secure_random_steps=2, clean_with_zeros=True), StepZero(error=False, start_time=datetime.now(), end_time=datetime.now(), secure_random_steps=3, clean_with_zeros=True) ] ) db.session.add(erasure) db.session.commit() db_erasure = EraseSectors.query.one() # Steps are in order assert db_erasure.steps[0].secure_random_steps == 1 assert db_erasure.steps[0].num == 0 assert db_erasure.steps[1].secure_random_steps == 2 assert db_erasure.steps[1].num == 1 assert db_erasure.steps[2].secure_random_steps == 3 assert db_erasure.steps[2].num == 2 @pytest.mark.usefixtures('auth_app_context') def test_test_data_storage(): test = TestDataStorage( device=HardDrive(serial_number='foo', manufacturer='bar', model='foo-bar'), error=False, elapsed=timedelta(minutes=25), length=TestHardDriveLength.Short, status='OK!', lifetime=timedelta(days=120) ) db.session.add(test) db.session.commit() assert TestDataStorage.query.one() @pytest.mark.usefixtures('auth_app_context') def test_install(): hdd = HardDrive(serial_number='sn') install = Install(name='LinuxMint 18.04 es', elapsed=timedelta(seconds=25), device=hdd) db.session.add(install) db.session.commit() @pytest.mark.usefixtures('auth_app_context') def test_update_components_event_one(): computer = Computer(serial_number='sn1', model='ml1', manufacturer='mr1') hdd = HardDrive(serial_number='foo', manufacturer='bar', model='foo-bar') computer.components.add(hdd) # Add event test = StressTest(elapsed=timedelta(seconds=1)) computer.events_one.add(test) assert test.device == computer assert next(iter(test.components)) == hdd, 'Event has to have new components' # Remove event computer.events_one.clear() assert not test.device assert not test.components, 'Event has to loose the components' # If we add a component to a device AFTER assigning the event # to the device, the event doesn't get the new component computer.events_one.add(test) ram = RamModule() computer.components.add(ram) assert len(test.components) == 1 @pytest.mark.usefixtures('auth_app_context') def test_update_components_event_multiple(): computer = Computer(serial_number='sn1', model='ml1', manufacturer='mr1') hdd = HardDrive(serial_number='foo', manufacturer='bar', model='foo-bar') computer.components.add(hdd) ready = Ready() assert not ready.devices assert not ready.components # Add computer.events_multiple.add(ready) assert ready.devices == OrderedSet([computer]) assert next(iter(ready.components)) == hdd # Remove computer.events_multiple.remove(ready) assert not ready.devices assert not ready.components # init / replace collection ready.devices = OrderedSet([computer]) assert ready.devices assert ready.components @pytest.mark.usefixtures('auth_app_context') def test_update_parent(): computer = Computer(serial_number='sn1', model='ml1', manufacturer='mr1') hdd = HardDrive(serial_number='foo', manufacturer='bar', model='foo-bar') computer.components.add(hdd) # Add benchmark = BenchmarkDataStorage() benchmark.device = hdd assert benchmark.parent == computer assert not benchmark.components # Remove benchmark.device = None assert not benchmark.parent