This repository has been archived on 2024-05-31. You can view files and clone it, but cannot push or open issues or pull requests.
devicehub-teal/tests/test_inventory.py

105 lines
4.0 KiB
Python
Raw Normal View History

2018-06-12 14:50:05 +00:00
import pytest
2018-06-14 13:14:23 +00:00
from ereuse_devicehub.client import UserClient
from ereuse_devicehub.db import db
from ereuse_devicehub.devicehub import Devicehub
2018-06-26 13:35:13 +00:00
from ereuse_devicehub.resources.device.models import Desktop, Device, Laptop, SolidStateDrive
from ereuse_devicehub.resources.enums import ComputerChassis
2018-07-17 17:00:07 +00:00
from ereuse_devicehub.resources.event.models import Snapshot
2018-06-14 13:14:23 +00:00
from ereuse_devicehub.resources.inventory import Filters, Inventory, Sorting
2018-06-12 14:50:05 +00:00
from teal.utils import compiled
2018-07-17 17:00:07 +00:00
from tests.conftest import file
2018-06-12 14:50:05 +00:00
@pytest.mark.usefixtures('app_context')
def test_inventory_filters():
schema = Filters()
q = schema.load({
2018-06-14 13:14:23 +00:00
'type': ['Computer', 'Laptop'],
2018-06-12 14:50:05 +00:00
'manufacturer': 'Dell',
'rating': {
'rating': [3, 6],
'appearance': [2, 4]
},
'tag': {
'id': ['bcn-', 'activa-02']
}
})
s, params = compiled(Device, q)
# Order between query clauses can change
2018-06-26 13:35:13 +00:00
assert '(device.type IN (%(type_1)s, %(type_2)s, %(type_3)s, %(type_4)s) ' \
'OR device.type IN (%(type_5)s))' in s
2018-06-12 14:50:05 +00:00
assert 'device.manufacturer ILIKE %(manufacturer_1)s' in s
assert 'rate.rating BETWEEN %(rating_1)s AND %(rating_2)s' in s
assert 'rate.appearance BETWEEN %(appearance_1)s AND %(appearance_2)s' in s
assert '(tag.id ILIKE %(id_1)s OR tag.id ILIKE %(id_2)s)' in s
2018-06-14 13:14:23 +00:00
# type_x can be assigned at different values
# ex: type_1 can be 'Desktop' in one execution but the next one 'Laptop'
2018-06-26 13:35:13 +00:00
assert set(params.keys()) == {'id_2', 'appearance_1', 'type_1', 'type_4', 'rating_2', 'type_5',
'type_3', 'type_2', 'appearance_2', 'id_1', 'rating_1',
'manufacturer_1'}
assert set(params.values()) == {2.0, 'Laptop', 4.0, 3.0, 6.0, 'Desktop', 'activa-02%',
'Server', 'Dell%', 'Computer', 'bcn-%'}
2018-06-12 14:50:05 +00:00
@pytest.mark.usefixtures('app_context')
2018-06-14 13:14:23 +00:00
def test_inventory_sort():
schema = Sorting()
r = next(schema.load({'created': True}))
assert str(r) == 'device.created ASC'
@pytest.fixture()
def inventory_query_dummy(app: Devicehub):
with app.app_context():
db.session.add_all(( # The order matters ;-)
2018-06-26 13:35:13 +00:00
Desktop(serial_number='s1',
model='ml1',
manufacturer='mr1',
chassis=ComputerChassis.Tower),
Laptop(serial_number='s3',
model='ml3',
manufacturer='mr3',
chassis=ComputerChassis.Detachable),
Desktop(serial_number='s2',
model='ml2',
manufacturer='mr2',
chassis=ComputerChassis.Microtower),
2018-06-14 13:14:23 +00:00
SolidStateDrive(serial_number='s4', model='ml4', manufacturer='mr4')
))
db.session.commit()
2018-07-17 17:00:07 +00:00
@pytest.mark.usefixtures(inventory_query_dummy.__name__)
2018-06-14 13:14:23 +00:00
def test_inventory_query_no_filters(user: UserClient):
i, _ = user.get(res=Inventory)
assert tuple(d['type'] for d in i['devices']) == (
2018-06-26 13:35:13 +00:00
'SolidStateDrive', 'Desktop', 'Laptop', 'Desktop'
2018-06-14 13:14:23 +00:00
)
2018-07-17 17:00:07 +00:00
@pytest.mark.usefixtures(inventory_query_dummy.__name__)
2018-06-14 13:14:23 +00:00
def test_inventory_query_filter_type(user: UserClient):
2018-06-26 13:35:13 +00:00
i, _ = user.get(res=Inventory, query=[('filter', {'type': ['Desktop', 'Laptop']})])
assert tuple(d['type'] for d in i['devices']) == ('Desktop', 'Laptop', 'Desktop')
2018-06-14 13:14:23 +00:00
2018-07-17 17:00:07 +00:00
@pytest.mark.usefixtures(inventory_query_dummy.__name__)
2018-06-14 13:14:23 +00:00
def test_inventory_query_filter_sort(user: UserClient):
i, _ = user.get(res=Inventory, query=[
('sort', {'created': Sorting.ASCENDING}),
('filter', {'type': ['Computer']})
])
2018-06-26 13:35:13 +00:00
assert tuple(d['type'] for d in i['devices']) == ('Desktop', 'Laptop', 'Desktop')
2018-07-17 17:00:07 +00:00
def test_inventory_query(user: UserClient):
"""Checks result of inventory."""
user.post(file('basic.snapshot'), res=Snapshot)
i, _ = user.get(res=Inventory)
pc = next(d for d in i['devices'] if d['type'] == 'Desktop')
assert len(pc['events']) == 4
assert len(pc['components']) == 3
assert not pc['tags']