503 lines
20 KiB
Python
503 lines
20 KiB
Python
import csv
|
|
import hashlib
|
|
from datetime import datetime
|
|
from io import StringIO
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
from werkzeug.exceptions import Unauthorized
|
|
import teal.marshmallow
|
|
from ereuse_utils.test import ANY
|
|
|
|
from ereuse_devicehub.client import Client, UserClient
|
|
from ereuse_devicehub.devicehub import Devicehub
|
|
from ereuse_devicehub.resources.action.models import Snapshot, Allocate, Live
|
|
from ereuse_devicehub.resources.documents import documents
|
|
from ereuse_devicehub.resources.device import models as d
|
|
from ereuse_devicehub.resources.lot.models import Lot
|
|
from ereuse_devicehub.resources.tag.model import Tag
|
|
from ereuse_devicehub.resources.hash_reports import ReportHash
|
|
from ereuse_devicehub.db import db
|
|
from tests import conftest
|
|
from tests.conftest import file
|
|
|
|
|
|
@pytest.mark.mvp
|
|
def test_erasure_certificate_public_one(user: UserClient, client: Client):
|
|
"""Public user can get certificate from one device as HTML or PDF."""
|
|
s = file('erase-sectors.snapshot')
|
|
snapshot, _ = user.post(s, res=Snapshot)
|
|
|
|
doc, response = user.get(res=documents.DocumentDef.t,
|
|
item='erasures/{}'.format(snapshot['device']['id']),
|
|
accept=ANY)
|
|
assert 'html' in response.content_type
|
|
assert '<html' in doc
|
|
assert '2018' in doc
|
|
|
|
doc, response = client.get(res=documents.DocumentDef.t,
|
|
item='erasures/{}'.format(snapshot['device']['id']),
|
|
query=[('format', 'PDF')],
|
|
accept='application/pdf')
|
|
assert 'application/pdf' == response.content_type
|
|
|
|
erasure = next(e for e in snapshot['actions'] if e['type'] == 'EraseSectors')
|
|
|
|
doc, response = client.get(res=documents.DocumentDef.t,
|
|
item='erasures/{}'.format(erasure['id']),
|
|
accept=ANY)
|
|
assert 'html' in response.content_type
|
|
assert '<html' in doc
|
|
assert '2018' in doc
|
|
|
|
|
|
@pytest.mark.mvp
|
|
def test_erasure_certificate_private_query(user: UserClient):
|
|
"""Logged-in user can get certificates using queries as HTML and
|
|
PDF.
|
|
"""
|
|
s = file('erase-sectors.snapshot')
|
|
snapshot, response = user.post(s, res=Snapshot)
|
|
|
|
doc, response = user.get(res=documents.DocumentDef.t,
|
|
item='erasures/',
|
|
query=[('filter', {'id': [snapshot['device']['id']]})],
|
|
accept=ANY)
|
|
assert 'html' in response.content_type
|
|
assert '<html' in doc
|
|
assert '2018' in doc
|
|
|
|
doc, response = user.get(res=documents.DocumentDef.t,
|
|
item='erasures/',
|
|
query=[
|
|
('filter', {'id': [snapshot['device']['id']]}),
|
|
('format', 'PDF')
|
|
],
|
|
accept='application/pdf')
|
|
assert 'application/pdf' == response.content_type
|
|
|
|
|
|
@pytest.mark.mvp
|
|
def test_erasure_certificate_wrong_id(client: Client):
|
|
client.get(res=documents.DocumentDef.t, item='erasures/this-is-not-an-id',
|
|
status=teal.marshmallow.ValidationError)
|
|
|
|
|
|
@pytest.mark.mvp
|
|
def test_export_csv_permitions(user: UserClient, user2: UserClient, client: Client):
|
|
"""test export device information in a csv file with others users."""
|
|
snapshot, _ = user.post(file('basic.snapshot'), res=Snapshot)
|
|
csv_user, _ = user.get(res=documents.DocumentDef.t,
|
|
item='devices/',
|
|
accept='text/csv',
|
|
query=[('filter', {'type': ['Computer']})])
|
|
|
|
csv_user2, _ = user2.get(res=documents.DocumentDef.t,
|
|
item='devices/',
|
|
accept='text/csv',
|
|
query=[('filter', {'type': ['Computer']})])
|
|
|
|
_, res = client.get(res=documents.DocumentDef.t,
|
|
item='devices/',
|
|
accept='text/csv',
|
|
query=[('filter', {'type': ['Computer']})], status=401)
|
|
assert res.status_code == 401
|
|
|
|
assert len(csv_user) > 0
|
|
assert len(csv_user2) == 0
|
|
|
|
|
|
@pytest.mark.mvp
|
|
def test_export_csv_actions(user: UserClient, user2: UserClient, client: Client):
|
|
"""Test export device information in a csv file with others users."""
|
|
acer = file('acer.happy.battery.snapshot')
|
|
snapshot, _ = user.post(acer, res=Snapshot)
|
|
device_id = snapshot['device']['id']
|
|
post_request = {"transaction": "ccc", "name": "John", "endUsers": 1,
|
|
"devices": [device_id], "description": "aaa",
|
|
"finalUserCode": "abcdefjhi",
|
|
"startTime": "2020-11-01T02:00:00+00:00",
|
|
"endTime": "2020-12-01T02:00:00+00:00"
|
|
}
|
|
|
|
user.post(res=Allocate, data=post_request)
|
|
hdd = [c for c in acer['components'] if c['type'] == 'HardDrive'][0]
|
|
hdd_action = [a for a in hdd['actions'] if a['type'] == 'TestDataStorage'][0]
|
|
hdd_action['lifetime'] += 1000
|
|
acer.pop('elapsed')
|
|
acer['licence_version'] = '1.0.0'
|
|
snapshot, _ = client.post(acer, res=Live)
|
|
|
|
csv_user, _ = user.get(res=documents.DocumentDef.t,
|
|
item='actions/',
|
|
accept='text/csv',
|
|
query=[('filter', {'type': ['Computer']})])
|
|
|
|
csv_user2, _ = user2.get(res=documents.DocumentDef.t,
|
|
item='actions/',
|
|
accept='text/csv',
|
|
query=[('filter', {'type': ['Computer']})])
|
|
|
|
_, res = client.get(res=documents.DocumentDef.t,
|
|
item='actions/',
|
|
accept='text/csv',
|
|
query=[('filter', {'type': ['Computer']})], status=401)
|
|
assert res.status_code == 401
|
|
|
|
assert len(csv_user) > 0
|
|
assert len(csv_user2) == 0
|
|
|
|
|
|
@pytest.mark.mvp
|
|
@pytest.mark.usefixtures(conftest.app_context.__name__)
|
|
def test_live_export_csv2(user: UserClient, client: Client, app: Devicehub):
|
|
"""Tests inserting a Live into the database and GETting it."""
|
|
acer = file('acer-happy.snapshot-test1')
|
|
snapshot, _ = user.post(acer, res=Snapshot)
|
|
device_id = snapshot['device']['id']
|
|
post_request = {"transaction": "ccc", "name": "John", "endUsers": 1,
|
|
"devices": [device_id], "description": "aaa",
|
|
"finalUserCode": "abcdefjhi",
|
|
"startTime": "2020-11-01T02:00:00+00:00",
|
|
"endTime": "2020-12-01T02:00:00+00:00"
|
|
}
|
|
|
|
user.post(res=Allocate, data=post_request)
|
|
|
|
acer = file('acer-happy.live-test1')
|
|
live, _ = client.post(acer, res=Live)
|
|
csv_user, _ = user.get(res=documents.DocumentDef.t,
|
|
item='actions/',
|
|
accept='text/csv',
|
|
query=[('filter', {'type': ['Computer']})])
|
|
|
|
assert "4692" in csv_user
|
|
assert "8692" in csv_user
|
|
assert "SystemId" in csv_user
|
|
|
|
@pytest.mark.mvp
|
|
@pytest.mark.usefixtures(conftest.app_context.__name__)
|
|
def test_live_example2(user: UserClient, client: Client, app: Devicehub):
|
|
"""Tests inserting a Live into the database and GETting it."""
|
|
acer = file('acer-happy.snapshot-test1')
|
|
snapshot, _ = user.post(acer, res=Snapshot)
|
|
device_id = snapshot['device']['id']
|
|
post_request = {"transaction": "ccc", "name": "John", "endUsers": 1,
|
|
"devices": [device_id], "description": "aaa",
|
|
"finalUserCode": "abcdefjhi",
|
|
"startTime": "2020-11-01T02:00:00+00:00",
|
|
"endTime": "2020-12-01T02:00:00+00:00"
|
|
}
|
|
|
|
user.post(res=Allocate, data=post_request)
|
|
|
|
acer = file('acer-happy.live-test1')
|
|
live, _ = client.post(acer, res=Live)
|
|
db_device = d.Device.query.filter_by(id=device_id).one()
|
|
action_live = [a for a in db_device.actions if a.type == 'Live']
|
|
assert len(action_live) == 1
|
|
assert str(action_live[0].snapshot_uuid) == acer['uuid']
|
|
|
|
|
|
@pytest.mark.mvp
|
|
def test_export_basic_snapshot(user: UserClient):
|
|
"""Test export device information in a csv file."""
|
|
snapshot, _ = user.post(file('basic.snapshot'), res=Snapshot)
|
|
csv_str, _ = user.get(res=documents.DocumentDef.t,
|
|
item='devices/',
|
|
accept='text/csv',
|
|
query=[('filter', {'type': ['Computer']})])
|
|
f = StringIO(csv_str)
|
|
obj_csv = csv.reader(f, f, delimiter=';', quotechar='"')
|
|
export_csv = list(obj_csv)
|
|
|
|
# Open fixture csv and transform to list
|
|
with Path(__file__).parent.joinpath('files').joinpath('basic.csv').open() as csv_file:
|
|
obj_csv = csv.reader(csv_file, delimiter=';', quotechar='"')
|
|
fixture_csv = list(obj_csv)
|
|
|
|
assert isinstance(datetime.strptime(export_csv[1][18], '%c'), datetime), \
|
|
'Register in field is not a datetime'
|
|
|
|
assert fixture_csv[0] == export_csv[0], 'Headers are not equal'
|
|
assert fixture_csv[1][:18] == export_csv[1][:18], 'Computer information are not equal'
|
|
assert fixture_csv[1][19] == export_csv[1][19], 'Computer information are not equal'
|
|
assert fixture_csv[1][21:] == export_csv[1][21:], 'Computer information are not equal'
|
|
|
|
|
|
@pytest.mark.mvp
|
|
@pytest.mark.usefixtures(conftest.app_context.__name__)
|
|
def test_check_insert_hash(app: Devicehub, user: UserClient, client: Client):
|
|
"""Test export device information in a csv file."""
|
|
snapshot, _ = user.post(file('basic.snapshot'), res=Snapshot)
|
|
csv_str, _ = user.get(res=documents.DocumentDef.t,
|
|
item='devices/',
|
|
accept='text/csv',
|
|
query=[('filter', {'type': ['Computer']})])
|
|
hash3 = hashlib.sha3_256(csv_str.encode('utf-8')).hexdigest()
|
|
assert ReportHash.query.filter_by(hash3=hash3).count() == 1
|
|
result, status = client.get(res=documents.DocumentDef.t, item='check/', query=[('hash', hash3)])
|
|
assert status.status_code == 200
|
|
assert result == True
|
|
|
|
ff = open('/tmp/test.csv', 'w')
|
|
ff.write(csv_str)
|
|
ff.close()
|
|
|
|
a= open('/tmp/test.csv').read()
|
|
assert hash3 == hashlib.sha3_256(a.encode('utf-8')).hexdigest()
|
|
|
|
|
|
@pytest.mark.mvp
|
|
def test_export_extended(app: Devicehub, user: UserClient):
|
|
"""Test a export device with all information and a lot of components."""
|
|
snapshot1, _ = user.post(file('real-eee-1001pxd.snapshot.12'), res=Snapshot, status=201)
|
|
snapshot2, _ = user.post(file('complete.export.snapshot'), res=Snapshot, status=201)
|
|
with app.app_context():
|
|
# Create a pc with a tag
|
|
tag = Tag(id='foo', owner_id=user.user['id'])
|
|
# pc = Desktop(serial_number='sn1', chassis=ComputerChassis.Tower, owner_id=user.user['id'])
|
|
pc = d.Device.query.filter_by(id=snapshot1['device']['id']).first()
|
|
pc.tags.add(tag)
|
|
db.session.add(pc)
|
|
db.session.commit()
|
|
csv_str, _ = user.get(res=documents.DocumentDef.t,
|
|
item='devices/',
|
|
accept='text/csv',
|
|
query=[('filter', {'type': ['Computer']})])
|
|
|
|
f = StringIO(csv_str)
|
|
obj_csv = csv.reader(f, f, delimiter=';', quotechar='"')
|
|
export_csv = list(obj_csv)
|
|
|
|
# Open fixture csv and transform to list
|
|
with Path(__file__).parent.joinpath('files').joinpath(
|
|
'proposal_extended_csv_report.csv').open() as csv_file:
|
|
obj_csv = csv.reader(csv_file, delimiter=';', quotechar='"')
|
|
fixture_csv = list(obj_csv)
|
|
|
|
assert isinstance(datetime.strptime(export_csv[1][18], '%c'), datetime), \
|
|
'Register in field is not a datetime'
|
|
|
|
assert fixture_csv[0] == export_csv[0], 'Headers are not equal'
|
|
assert fixture_csv[1][:18] == export_csv[1][:18], 'Computer information are not equal'
|
|
assert fixture_csv[1][19] == export_csv[1][19], 'Computer information are not equal'
|
|
assert fixture_csv[1][21:80] == export_csv[1][21:80], 'Computer information are not equal'
|
|
assert fixture_csv[1][81] == export_csv[1][81], 'Computer information are not equal'
|
|
assert fixture_csv[1][84:] == export_csv[1][84:], 'Computer information are not equal'
|
|
assert fixture_csv[2][:18] == export_csv[2][:18], 'Computer information are not equal'
|
|
assert fixture_csv[2][19] == export_csv[2][19], 'Computer information are not equal'
|
|
assert fixture_csv[2][21:80] == export_csv[2][21:80], 'Computer information are not equal'
|
|
assert fixture_csv[2][81] == export_csv[2][81], 'Computer information are not equal'
|
|
assert fixture_csv[2][84:104] == export_csv[2][84:104], 'Computer information are not equal'
|
|
assert fixture_csv[2][105] == export_csv[2][105], 'Computer information are not equal'
|
|
assert fixture_csv[2][108:128] == export_csv[2][108:128], 'Computer information are not equal'
|
|
assert fixture_csv[2][129] == export_csv[2][129], 'Computer information are not equal'
|
|
assert fixture_csv[2][132:] == export_csv[2][132:], 'Computer information are not equal'
|
|
|
|
|
|
@pytest.mark.mvp
|
|
def test_export_empty(user: UserClient):
|
|
"""Test to check works correctly exporting csv without any information,
|
|
export a placeholder device.
|
|
"""
|
|
csv_str, _ = user.get(res=documents.DocumentDef.t,
|
|
accept='text/csv',
|
|
item='devices/')
|
|
f = StringIO(csv_str)
|
|
obj_csv = csv.reader(f, f)
|
|
export_csv = list(obj_csv)
|
|
|
|
assert len(export_csv) == 0, 'Csv is not empty'
|
|
|
|
|
|
@pytest.mark.xfail(reason='Feature not developed (Beta)')
|
|
def test_export_computer_monitor(user: UserClient):
|
|
"""Test a export device type computer monitor."""
|
|
snapshot, _ = user.post(file('computer-monitor.snapshot'), res=Snapshot)
|
|
csv_str, _ = user.get(res=documents.DocumentDef.t,
|
|
item='devices/',
|
|
accept='text/csv',
|
|
query=[('filter', {'type': ['ComputerMonitor']})])
|
|
f = StringIO(csv_str)
|
|
obj_csv = csv.reader(f, f)
|
|
export_csv = list(obj_csv)
|
|
# Open fixture csv and transform to list
|
|
with Path(__file__).parent.joinpath('files').joinpath('computer-monitor.csv').open() \
|
|
as csv_file:
|
|
obj_csv = csv.reader(csv_file)
|
|
fixture_csv = list(obj_csv)
|
|
|
|
# Pop dates fields from csv lists to compare them
|
|
fixture_csv[1] = fixture_csv[1][:8]
|
|
export_csv[1] = export_csv[1][:8]
|
|
|
|
assert fixture_csv[0] == export_csv[0], 'Headers are not equal'
|
|
assert fixture_csv[1] == export_csv[1], 'Component information are not equal'
|
|
|
|
|
|
@pytest.mark.xfail(reason='Feature not developed (Beta)')
|
|
def test_export_keyboard(user: UserClient):
|
|
"""Test a export device type keyboard."""
|
|
snapshot, _ = user.post(file('keyboard.snapshot'), res=Snapshot)
|
|
csv_str, _ = user.get(res=documents.DocumentDef.t,
|
|
item='devices/',
|
|
accept='text/csv',
|
|
query=[('filter', {'type': ['Keyboard']})])
|
|
f = StringIO(csv_str)
|
|
obj_csv = csv.reader(f, f)
|
|
export_csv = list(obj_csv)
|
|
# Open fixture csv and transform to list
|
|
with Path(__file__).parent.joinpath('files').joinpath('keyboard.csv').open() as csv_file:
|
|
obj_csv = csv.reader(csv_file)
|
|
fixture_csv = list(obj_csv)
|
|
|
|
# Pop dates fields from csv lists to compare them
|
|
fixture_csv[1] = fixture_csv[1][:8]
|
|
export_csv[1] = export_csv[1][:8]
|
|
|
|
assert fixture_csv[0] == export_csv[0], 'Headers are not equal'
|
|
assert fixture_csv[1] == export_csv[1], 'Component information are not equal'
|
|
|
|
|
|
@pytest.mark.xfail(reason='Feature not developed (Beta)')
|
|
def test_export_multiple_different_devices(user: UserClient):
|
|
"""Test function 'Export' of multiple different device types (like
|
|
computers, keyboards, monitors, etc..)
|
|
"""
|
|
# Open fixture csv and transform to list
|
|
with Path(__file__).parent.joinpath('files').joinpath('multiples_devices.csv').open() \
|
|
as csv_file:
|
|
fixture_csv = list(csv.reader(csv_file))
|
|
for row in fixture_csv:
|
|
del row[8] # We remove the 'Registered in' column
|
|
|
|
# Post all devices snapshots
|
|
snapshot_pc, _ = user.post(file('real-eee-1001pxd.snapshot.11'), res=Snapshot)
|
|
snapshot_empty, _ = user.post(file('basic.snapshot'), res=Snapshot)
|
|
snapshot_keyboard, _ = user.post(file('keyboard.snapshot'), res=Snapshot)
|
|
snapshot_monitor, _ = user.post(file('computer-monitor.snapshot'), res=Snapshot)
|
|
|
|
csv_str, _ = user.get(res=documents.DocumentDef.t,
|
|
item='devices/',
|
|
query=[('filter', {'type': ['Computer', 'Keyboard', 'Monitor']})],
|
|
accept='text/csv')
|
|
f = StringIO(csv_str)
|
|
obj_csv = csv.reader(f, f)
|
|
export_csv = list(obj_csv)
|
|
|
|
for row in export_csv:
|
|
del row[8]
|
|
|
|
assert fixture_csv == export_csv
|
|
|
|
|
|
@pytest.mark.mvp
|
|
def test_report_devices_stock_control(user: UserClient, user2: UserClient):
|
|
"""Test export device information in a csv file."""
|
|
snapshot, _ = user.post(file('basic.snapshot'), res=Snapshot)
|
|
snapshot2, _ = user2.post(file('basic.snapshot2'), res=Snapshot)
|
|
|
|
csv_str, _ = user.get(res=documents.DocumentDef.t,
|
|
item='stock/',
|
|
accept='text/csv',
|
|
query=[('filter', {'type': ['Computer']})])
|
|
f = StringIO(csv_str)
|
|
obj_csv = csv.reader(f, f)
|
|
export_csv = list(obj_csv)
|
|
|
|
# Open fixture csv and transform to list
|
|
with Path(__file__).parent.joinpath('files').joinpath('basic-stock.csv').open() as csv_file:
|
|
obj_csv = csv.reader(csv_file)
|
|
fixture_csv = list(obj_csv)
|
|
|
|
assert user.user['id'] != user2.user['id']
|
|
assert len(export_csv) == 2
|
|
|
|
export_csv[0] = export_csv[0][0].split(';')
|
|
export_csv[1] = export_csv[1][0].split(';')
|
|
assert isinstance(datetime.strptime(export_csv[1][5], '%c'), datetime), \
|
|
'Register in field is not a datetime'
|
|
|
|
# Pop dates fields from csv lists to compare them
|
|
fixture_csv[1] = fixture_csv[1][:5] + fixture_csv[1][6:]
|
|
export_csv[1] = export_csv[1][:5] + export_csv[1][6:]
|
|
|
|
assert fixture_csv[0] == export_csv[0], 'Headers are not equal'
|
|
assert fixture_csv[1] == export_csv[1], 'Computer information are not equal'
|
|
assert fixture_csv == export_csv
|
|
|
|
|
|
@pytest.mark.mvp
|
|
def test_get_document_lots(user: UserClient, user2: UserClient):
|
|
"""Tests submitting and retreiving all lots."""
|
|
|
|
l, _ = user.post({'name': 'Lot1', 'description': 'comments,lot1,testcomment-lot1,'}, res=Lot)
|
|
l, _ = user.post({'name': 'Lot2', 'description': 'comments,lot2,testcomment-lot2,'}, res=Lot)
|
|
l, _ = user2.post({'name': 'Lot3-User2', 'description': 'comments,lot3,testcomment-lot3,'}, res=Lot)
|
|
|
|
csv_str, _ = user.get(res=documents.DocumentDef.t,
|
|
item='lots/',
|
|
accept='text/csv')
|
|
|
|
csv2_str, _ = user2.get(res=documents.DocumentDef.t,
|
|
item='lots/',
|
|
accept='text/csv')
|
|
|
|
f = StringIO(csv_str)
|
|
obj_csv = csv.reader(f, f)
|
|
export_csv = list(obj_csv)
|
|
|
|
f = StringIO(csv2_str)
|
|
obj2_csv = csv.reader(f, f)
|
|
export2_csv = list(obj2_csv)
|
|
|
|
assert len(export_csv) == 3
|
|
assert len(export2_csv) == 2
|
|
|
|
assert export_csv[0] == export2_csv[0] == ['Id', 'Name', 'Registered in', 'Description']
|
|
|
|
assert export_csv[1][1] == 'Lot1' or 'Lot2'
|
|
assert export_csv[1][3] == 'comments,lot1,testcomment-lot1,' or 'comments,lot2,testcomment-lot2,'
|
|
assert export2_csv[1][1] == 'Lot3-User2'
|
|
assert export2_csv[1][3] == 'comments,lot3,testcomment-lot3,'
|
|
|
|
|
|
@pytest.mark.mvp
|
|
def test_get_document_internal_stats(user: UserClient, user2: UserClient):
|
|
"""Tests for get the internal stats."""
|
|
|
|
# csv_str, _ = user.get(res=documents.DocumentDef.t,
|
|
# item='internalstats/')
|
|
csv_str, _ = user.get(res=documents.DocumentDef.t,
|
|
item='internalstats/',
|
|
accept='text/csv',
|
|
query=[])
|
|
|
|
f = StringIO(csv_str)
|
|
obj_csv = csv.reader(f, f)
|
|
export_csv = list(obj_csv)
|
|
|
|
assert len(export_csv) == 1
|
|
|
|
csv_str, _ = user2.get(res=documents.DocumentDef.t,
|
|
item='internalstats/',
|
|
accept='text/csv',
|
|
query=[])
|
|
|
|
f = StringIO(csv_str)
|
|
obj_csv = csv.reader(f, f)
|
|
export_csv = list(obj_csv)
|
|
|
|
assert csv_str.strip() == '""'
|
|
|
|
@pytest.mark.mvp
|
|
def test_get_wbconf(user: UserClient):
|
|
"""Tests for get env file for usb wb."""
|
|
|
|
env, _ = user.get(res=documents.DocumentDef.t, item='wbconf/usodyrate', accept=ANY)
|
|
assert 'WB_ERASE = False' in env
|
|
|
|
env, _ = user.get(res=documents.DocumentDef.t, item='wbconf/usodywipe', accept=ANY)
|
|
assert 'WB_ERASE = True' in env
|