66 lines
2.0 KiB
Python
66 lines
2.0 KiB
Python
|
import pytest
|
||
|
|
from flask import g
|
||
|
|
from sqlalchemy_utils import Ltree
|
||
|
|
|
||
|
|
from ereuse_devicehub.db import db
|
||
|
|
from ereuse_devicehub.resources.device.models import Desktop
|
||
|
|
from ereuse_devicehub.resources.enums import ComputerChassis
|
||
|
|
from ereuse_devicehub.resources.lot.models import Edge, Lot, LotDevice
|
||
|
|
from tests import conftest
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.usefixtures(conftest.auth_app_context.__name__)
|
||
|
|
def test_lot_device_relationship():
|
||
|
|
device = Desktop(serial_number='foo',
|
||
|
|
model='bar',
|
||
|
|
manufacturer='foobar',
|
||
|
|
chassis=ComputerChassis.Lunchbox)
|
||
|
|
lot = Lot(name='lot1')
|
||
|
|
lot.devices.add(device)
|
||
|
|
db.session.add(lot)
|
||
|
|
db.session.flush()
|
||
|
|
|
||
|
|
lot_device = LotDevice.query.one() # type: LotDevice
|
||
|
|
assert lot_device.device_id == device.id
|
||
|
|
assert lot_device.lot_id == lot.id
|
||
|
|
assert lot_device.created
|
||
|
|
assert lot_device.author_id == g.user.id
|
||
|
|
assert device.parents == {lot}
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.usefixtures(conftest.auth_app_context.__name__)
|
||
|
|
def test_add_edge():
|
||
|
|
child = Lot(name='child')
|
||
|
|
parent = Lot(name='parent')
|
||
|
|
db.session.add(child)
|
||
|
|
db.session.add(parent)
|
||
|
|
db.session.flush()
|
||
|
|
# todo edges should automatically be created when the lot is created
|
||
|
|
child.edges.add(Edge(path=Ltree(str(child.id).replace('-', '_'))))
|
||
|
|
parent.edges.add(Edge(path=Ltree(str(parent.id).replace('-', '_'))))
|
||
|
|
db.session.flush()
|
||
|
|
|
||
|
|
parent.add_child(child)
|
||
|
|
|
||
|
|
assert child in parent
|
||
|
|
assert len(child.edges) == 1
|
||
|
|
assert len(parent.edges) == 1
|
||
|
|
|
||
|
|
parent.remove_child(child)
|
||
|
|
assert child not in parent
|
||
|
|
assert len(child.edges) == 1
|
||
|
|
assert len(parent.edges) == 1
|
||
|
|
|
||
|
|
grandparent = Lot(name='grandparent')
|
||
|
|
db.session.add(grandparent)
|
||
|
|
db.session.flush()
|
||
|
|
grandparent.edges.add(Edge(path=Ltree(str(grandparent.id).replace('-', '_'))))
|
||
|
|
db.session.flush()
|
||
|
|
|
||
|
|
grandparent.add_child(parent)
|
||
|
|
parent.add_child(child)
|
||
|
|
|
||
|
|
assert parent in grandparent
|
||
|
|
assert child in parent
|
||
|
|
assert child in grandparent
|