devicehub-teal/tests/test_documents.py

import csv
import hashlib
from datetime import datetime
from io import BytesIO, StringIO
from pathlib import Path

import pytest
import teal.marshmallow
from ereuse_utils.test import ANY
from flask import url_for
from werkzeug.exceptions import Unauthorized

from ereuse_devicehub import auth
from ereuse_devicehub.client import Client, UserClient
from ereuse_devicehub.db import db
from ereuse_devicehub.devicehub import Devicehub
from ereuse_devicehub.resources.action.models import Allocate, Live, Snapshot
from ereuse_devicehub.resources.device import models as d
from ereuse_devicehub.resources.documents import documents
from ereuse_devicehub.resources.enums import SessionType
from ereuse_devicehub.resources.hash_reports import ReportHash
from ereuse_devicehub.resources.lot.models import Lot
from ereuse_devicehub.resources.tag.model import Tag
from ereuse_devicehub.resources.tradedocument.models import TradeDocument
from ereuse_devicehub.resources.user.models import Session
from tests import conftest
from tests.conftest import file, json_encode, yaml2json


@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', {'ids': [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', {'ids': [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)
    dev_id = snapshot['device']['id']
    csv_user, _ = user.get(
        res=documents.DocumentDef.t,
        item='devices/',
        accept='text/csv',
        query=[('filter', {'type': ['Computer'], 'ids': [dev_id]})],
    )

    csv_user2, _ = user2.get(
        res=documents.DocumentDef.t,
        item='devices/',
        accept='text/csv',
        query=[('filter', {'type': ['Computer'], 'ids': [dev_id]})],
    )

    _, res = client.get(
        res=documents.DocumentDef.t,
        item='devices/',
        accept='text/csv',
        query=[('filter', {'type': ['Computer'], 'ids': [dev_id]})],
        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 = yaml2json('acer.happy.battery.snapshot')
    snapshot, _ = user.post(json_encode(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'], 'ids': [device_id]})],
    )

    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 = yaml2json('acer-happy.snapshot-test1')
    snapshot, _ = user.post(json_encode(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 = yaml2json('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'], 'ids': [device_id]})],
    )

    assert "4692" in csv_user
    assert "8692" in csv_user
    assert "DHID" 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 = yaml2json('acer-happy.snapshot-test1')
    snapshot, _ = user.post(json_encode(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 = yaml2json('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)
    dev_id = snapshot['device']['id']
    csv_str, _ = user.get(
        res=documents.DocumentDef.t,
        item='devices/',
        accept='text/csv',
        query=[('filter', {'type': ['Computer'], 'ids': [dev_id]})],
    )

    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][19], '%c'), datetime
    ), 'Register in field is not a datetime'

    assert fixture_csv[0] == export_csv[0], 'Headers are not equal'
    assert (
        fixture_csv[1][:19] == export_csv[1][:19]
    ), 'Computer information are not equal'
    assert fixture_csv[1][20] == export_csv[1][20], 'Computer information are not equal'
    assert (
        fixture_csv[1][22:] == export_csv[1][22:]
    ), '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

    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)
    dev1_id = snapshot1['device']['id']
    dev2_id = snapshot2['device']['id']
    with app.app_context():
        # Create a pc with a tag
        pc = d.Device.query.filter_by(id=dev1_id).first()
        db.session.add(pc)
        db.session.commit()

    csv_str, _ = user.get(
        res=documents.DocumentDef.t,
        item='devices/',
        accept='text/csv',
        query=[('filter', {'type': ['Computer'], 'ids': [dev1_id, dev2_id]})],
    )

    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][19], '%c'), datetime
    ), 'Register in field is not a datetime'

    assert fixture_csv[0] == export_csv[0], 'Headers are not equal'
    assert (
        fixture_csv[1][:19] == export_csv[1][:19]
    ), 'Computer information are not equal'
    assert fixture_csv[1][20] == export_csv[1][20], 'Computer information are not equal'
    assert (
        fixture_csv[1][22:83] == export_csv[1][22:83]
    ), 'Computer information are not equal'
    assert fixture_csv[1][84] == export_csv[1][84], 'Computer information are not equal'
    assert (
        fixture_csv[1][87:] == export_csv[1][87:]
    ), 'Computer information are not equal'
    assert (
        fixture_csv[2][:19] == export_csv[2][:19]
    ), 'Computer information are not equal'
    assert fixture_csv[2][20] == export_csv[2][20], 'Computer information are not equal'
    assert (
        fixture_csv[2][22:83] == export_csv[2][22:83]
    ), 'Computer information are not equal'
    assert fixture_csv[2][84] == export_csv[2][84], 'Computer information are not equal'
    assert (
        fixture_csv[2][87:107] == export_csv[2][87:107]
    ), 'Computer information are not equal'
    assert (
        fixture_csv[2][110] == export_csv[2][110]
    ), 'Computer information are not equal'
    assert (
        fixture_csv[2][113:134] == export_csv[2][113:134]
    ), 'Computer information are not equal'
    assert (
        fixture_csv[2][136] == export_csv[2][136]
    ), 'Computer information are not equal'
    assert (
        fixture_csv[2][139:] == export_csv[2][139:]
    ), '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(';')
    fixture_csv[0] = fixture_csv[0][0].split(';')
    fixture_csv[1] = fixture_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_verify_stamp(user: UserClient, client: Client):
    """Test verify stamp of one 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']})],
    )

    response, _ = client.post(
        res=documents.DocumentDef.t,
        item='stamps/',
        content_type='multipart/form-data',
        accept='text/html',
        data={'docUpload': [(BytesIO(bytes(csv_str, 'utf-8')), 'example.csv')]},
        status=200,
    )
    assert "alert alert-info" in response
    assert not "alert alert-danger" in response

    response, _ = client.post(
        res=documents.DocumentDef.t,
        item='stamps/',
        content_type='multipart/form-data',
        accept='text/html',
        data={'docUpload': [(BytesIO(b'abc'), 'example.csv')]},
        status=200,
    )

    assert not "alert alert-info" in response
    assert "alert alert-danger" in response

    response, _ = client.get(
        res=documents.DocumentDef.t, item='stamps/', accept='text/html', status=200
    )

    assert not "alert alert-info" in response
    assert not "alert alert-danger" in response


@pytest.mark.mvp
def test_verify_stamp_log_info(user: UserClient, client: Client):
    """Test verify stamp of one export lots-info in a csv file."""

    l, _ = user.post(
        {'name': 'Lot1', 'description': 'comments,lot1,testcomment-lot1,'}, res=Lot
    )
    l, _ = user.post(
        {'name': 'Lot2', 'description': 'comments,lot2,testcomment-lot2,'}, res=Lot
    )

    csv_str, _ = user.get(res=documents.DocumentDef.t, item='lots/', accept='text/csv')

    response, _ = client.post(
        res=documents.DocumentDef.t,
        item='stamps/',
        content_type='multipart/form-data',
        accept='text/html',
        data={'docUpload': [(BytesIO(bytes(csv_str, 'utf-8')), 'example.csv')]},
        status=200,
    )
    assert "alert alert-info" in response


@pytest.mark.mvp
def test_verify_stamp_devices_stock(user: UserClient, client: Client):
    """Test verify stamp of one export device information in a csv file."""

    snapshot, _ = user.post(file('basic.snapshot'), res=Snapshot)

    csv_str, _ = user.get(
        res=documents.DocumentDef.t,
        item='stock/',
        accept='text/csv',
        query=[('filter', {'type': ['Computer']})],
    )

    response, _ = client.post(
        res=documents.DocumentDef.t,
        item='stamps/',
        content_type='multipart/form-data',
        accept='text/html',
        data={'docUpload': [(BytesIO(bytes(csv_str, 'utf-8')), 'example.csv')]},
        status=200,
    )
    assert "alert alert-info" in response


@pytest.mark.mvp
def test_verify_stamp_csv_actions(user: UserClient, client: Client):
    """Test verify stamp of one export device information in a csv file with others users."""
    acer = yaml2json('acer.happy.battery.snapshot')
    snapshot, _ = user.post(json_encode(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_str, _ = user.get(
        res=documents.DocumentDef.t,
        item='actions/',
        accept='text/csv',
        query=[('filter', {'type': ['Computer']})],
    )

    response, _ = client.post(
        res=documents.DocumentDef.t,
        item='stamps/',
        content_type='multipart/form-data',
        accept='text/html',
        data={'docUpload': [(BytesIO(bytes(csv_str, 'utf-8')), 'example.csv')]},
        status=200,
    )
    assert "alert alert-info" in response


@pytest.mark.mvp
def test_verify_stamp_erasure_certificate(user: UserClient, client: Client):
    """Test verify stamp of one export certificate in PDF."""
    s = file('erase-sectors.snapshot')
    snapshot, response = user.post(s, res=Snapshot)

    doc, _ = user.get(
        res=documents.DocumentDef.t,
        item='erasures/',
        query=[('filter', {'ids': [snapshot['device']['id']]})],
        accept=ANY,
    )

    response, _ = client.post(
        res=documents.DocumentDef.t,
        item='stamps/',
        content_type='multipart/form-data',
        accept='text/html',
        data={'docUpload': [(BytesIO(bytes(doc, 'utf-8')), 'example.csv')]},
        status=200,
    )
    assert "alert alert-danger" in response

    doc, _ = user.get(
        res=documents.DocumentDef.t,
        item='erasures/',
        query=[('filter', {'ids': [snapshot['device']['id']]}), ('format', 'PDF')],
        accept='application/pdf',
    )

    response, _ = client.post(
        res=documents.DocumentDef.t,
        item='stamps/',
        content_type='multipart/form-data',
        accept='text/html',
        data={'docUpload': [(BytesIO(doc), 'example.csv')]},
        status=200,
    )
    assert "alert alert-info" in response


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
@pytest.mark.usefixtures(conftest.app_context.__name__)
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 =' in env

    env, _ = user.get(res=documents.DocumentDef.t, item='wbconf/usodywipe', accept=ANY)
    assert 'WB_ERASE =' in env
    # assert 'WB_ERASE = True' in env

    session = Session.query.filter_by(
        user_id=user.user['id'], type=SessionType.Internal
    ).first()
    token = session.token
    token = auth.Auth.encode(session.token)
    assert token in env
    user.user['token'] = token
    snapshot, _ = user.post(file('basic.snapshot'), res=Snapshot)


@pytest.mark.mvp
@pytest.mark.usefixtures(conftest.app_context.__name__)
def test_trade_documents(user: UserClient):
    """Tests upload one document"""

    lot, _ = user.post({'name': 'MyLot'}, res=Lot)
    request_post = {
        'filename': 'test.pdf',
        'hash': 'bbbbbbbb',
        'url': 'http://www.ereuse.org/',
        'lot': lot['id'],
    }
    doc, _ = user.post(res=TradeDocument, data=request_post)
    assert doc['filename'] == request_post['filename']
    assert doc['url'] == request_post['url']
    assert doc['hash'] == request_post['hash']
    assert doc['lot']['name'] == lot['name']


@pytest.mark.mvp
@pytest.mark.usefixtures(conftest.app_context.__name__)
def test_trade_documents_with_weight(user: UserClient):
    """Tests upload one document"""

    lot, _ = user.post({'name': 'MyLot'}, res=Lot)
    # long url
    url = (
        'http://www.ereuse.org/apapaapaapaapaapaapaapaapaapaapapaapaapaapaapaapaapaapaapaapapaapaapaapaapaapaapaapaapaapaaaa',
    )
    request_post = {
        'filename': 'test.pdf',
        'hash': 'bbbbbbbb',
        'url': url,
        'weight': 15,
        'lot': lot['id'],
    }
    doc, _ = user.post(res=TradeDocument, data=request_post)
    assert doc['weight'] == request_post['weight']