from distutils.version import StrictVersion import pytest from ereuse_devicehub.db import db from ereuse_devicehub.resources.device.models import Computer, Desktop, HardDrive, Processor, \ RamModule from ereuse_devicehub.resources.enums import AppearanceRange, Bios, ComputerChassis, \ FunctionalityRange, RatingSoftware from ereuse_devicehub.resources.event.models import AggregateRate, BenchmarkDataStorage, \ BenchmarkProcessor, EreusePrice, WorkbenchRate from ereuse_devicehub.resources.event.rate import main from tests import conftest @pytest.mark.usefixtures(conftest.auth_app_context.__name__) def test_workbench_rate_db(): rate = WorkbenchRate(processor=0.1, ram=1.0, bios_range=Bios.A, labelling=False, graphic_card=0.1, data_storage=4.1, software=RatingSoftware.ECost, version=StrictVersion('1.0'), device=Computer(serial_number='24', chassis=ComputerChassis.Tower)) db.session.add(rate) db.session.commit() @pytest.mark.xfail(reason='AggreagteRate only takes data from WorkbenchRate as for now') def test_rate_workbench_then_manual(): """Checks that a new AggregateRate is generated with a new rate value when a ManualRate is performed after performing a WorkbenchRate. The new AggregateRate needs to be computed by the values of the WorkbenchRate + new values from ManualRate. """ pass @pytest.mark.usefixtures(conftest.app_context.__name__) def test_rate(): """Test generating an AggregateRate for a given PC / components / WorkbenchRate ensuring results and relationships between pc - rate - workbenchRate - price. """ rate = WorkbenchRate( appearance_range=AppearanceRange.A, functionality_range=FunctionalityRange.A ) pc = Desktop() hdd = HardDrive(size=476940) hdd.events_one.add(BenchmarkDataStorage(read_speed=126, write_speed=29.8)) cpu = Processor(cores=2, speed=3.4) cpu.events_one.add(BenchmarkProcessor(rate=27136.44)) pc.components |= { hdd, RamModule(size=4096, speed=1600), RamModule(size=2048, speed=1067), cpu } rate.device = pc events = main.main(rate, RatingSoftware.ECost, StrictVersion('1.0')) price = next(e for e in events if isinstance(e, EreusePrice)) assert price.price == 92.2 assert price.retailer.standard.amount == 40.97 assert price.platform.standard.amount == 18.84 assert price.refurbisher.standard.amount == 32.38 assert price.price >= price.retailer.standard.amount \ + price.platform.standard.amount \ + price.refurbisher.standard.amount assert price.retailer.warranty2.amount == 55.30 assert price.platform.warranty2.amount == 25.43 assert price.refurbisher.warranty2.amount == 43.72 assert price.warranty2 == 124.45 # Checks relationships workbench_rate = next(e for e in events if isinstance(e, WorkbenchRate)) aggregate_rate = next(e for e in events if isinstance(e, AggregateRate)) assert price.rating == aggregate_rate assert aggregate_rate.workbench == workbench_rate assert aggregate_rate.rating == workbench_rate.rating == 4.61 assert aggregate_rate.software == workbench_rate.software == RatingSoftware.ECost assert aggregate_rate.version == StrictVersion('1.0') assert aggregate_rate.appearance == workbench_rate.appearance assert aggregate_rate.functionality == workbench_rate.functionality assert aggregate_rate.rating_range == workbench_rate.rating_range assert cpu.rate == pc.rate == hdd.rate == aggregate_rate assert cpu.price == pc.price == aggregate_rate.price == hdd.price == price