From 99d0a61e87d6575ed39864ec6c5eaa15966027e4 Mon Sep 17 00:00:00 2001 From: Cayo Puigdefabregas Date: Tue, 7 Jun 2022 15:24:45 +0200 Subject: [PATCH] return jwt.encode to snapshot code --- ereuse_devicehub/dummy/dummy.py | 205 +++++++++++++++++++------------- 1 file changed, 122 insertions(+), 83 deletions(-) diff --git a/ereuse_devicehub/dummy/dummy.py b/ereuse_devicehub/dummy/dummy.py index dbe96805..a1d38854 100644 --- a/ereuse_devicehub/dummy/dummy.py +++ b/ereuse_devicehub/dummy/dummy.py @@ -1,41 +1,37 @@ import itertools import json -import jwt from pathlib import Path from typing import Set import click import click_spinner import ereuse_utils.cli +import jwt import yaml from ereuse_utils.test import ANY from ereuse_devicehub.client import UserClient from ereuse_devicehub.db import db +from ereuse_devicehub.parser.models import SnapshotsLog from ereuse_devicehub.resources.action import models as m from ereuse_devicehub.resources.agent.models import Person from ereuse_devicehub.resources.device.models import Device +from ereuse_devicehub.resources.enums import SessionType from ereuse_devicehub.resources.lot.models import Lot from ereuse_devicehub.resources.tag.model import Tag from ereuse_devicehub.resources.user import User from ereuse_devicehub.resources.user.models import Session -from ereuse_devicehub.resources.enums import SessionType -from ereuse_devicehub.parser.models import SnapshotsLog class Dummy: - TAGS = ( - 'tag1', - 'tag2', - 'tag3' - ) + TAGS = ('tag1', 'tag2', 'tag3') """Tags to create.""" ET = ( ('DT-AAAAA', 'A0000000000001'), ('DT-BBBBB', 'A0000000000002'), ('DT-CCCCC', 'A0000000000003'), ('DT-BRRAB', '04970DA2A15984'), - ('DT-XXXXX', '04e4bc5af95980') + ('DT-XXXXX', '04e4bc5af95980'), ) """eTags to create.""" ORG = 'eReuse.org CAT', '-t', 'G-60437761', '-c', 'ES' @@ -44,27 +40,33 @@ class Dummy: def __init__(self, app) -> None: super().__init__() self.app = app - self.app.cli.command('dummy', short_help='Creates dummy devices and users.')(self.run) + self.app.cli.command('dummy', short_help='Creates dummy devices and users.')( + self.run + ) - @click.option('--tag-url', '-tu', - type=ereuse_utils.cli.URL(scheme=True, host=True, path=False), - default='http://localhost:8081', - help='The base url (scheme and host) of the tag provider.') - @click.option('--tag-token', '-tt', - type=click.UUID, - default='899c794e-1737-4cea-9232-fdc507ab7106', - help='The token provided by the tag provider. It is an UUID.') - @click.confirmation_option(prompt='This command (re)creates the DB from scratch.' - 'Do you want to continue?') + @click.option( + '--tag-url', + '-tu', + type=ereuse_utils.cli.URL(scheme=True, host=True, path=False), + default='http://localhost:8081', + help='The base url (scheme and host) of the tag provider.', + ) + @click.option( + '--tag-token', + '-tt', + type=click.UUID, + default='899c794e-1737-4cea-9232-fdc507ab7106', + help='The token provided by the tag provider. It is an UUID.', + ) + @click.confirmation_option( + prompt='This command (re)creates the DB from scratch.' + 'Do you want to continue?' + ) def run(self, tag_url, tag_token): runner = self.app.test_cli_runner() - self.app.init_db('Dummy', - 'ACME', - 'acme-id', - tag_url, - tag_token, - erase=True, - common=True) + self.app.init_db( + 'Dummy', 'ACME', 'acme-id', tag_url, tag_token, erase=True, common=True + ) print('Creating stuff...'.ljust(30), end='') assert SnapshotsLog.query.filter().all() == [] with click_spinner.spinner(): @@ -79,16 +81,31 @@ class Dummy: for id in self.TAGS: user1.post({'id': id}, res=Tag) for id, sec in self.ET: - runner.invoke('tag', 'add', id, - '-p', 'https://t.devicetag.io', - '-s', sec, - '-u', user1.user["id"], - '-o', org_id) + runner.invoke( + 'tag', + 'add', + id, + '-p', + 'https://t.devicetag.io', + '-s', + sec, + '-u', + user1.user["id"], + '-o', + org_id, + ) # create tag for pc-laudem - runner.invoke('tag', 'add', 'tagA', - '-p', 'https://t.devicetag.io', - '-u', user1.user["id"], - '-s', 'tagA-secondary') + runner.invoke( + 'tag', + 'add', + 'tagA', + '-p', + 'https://t.devicetag.io', + '-u', + user1.user["id"], + '-s', + 'tagA-secondary', + ) files = tuple(Path(__file__).parent.joinpath('files').iterdir()) print('done.') sample_pc = None # We treat this one as a special sample for demonstrations @@ -97,38 +114,49 @@ class Dummy: for path in bar: with path.open() as f: snapshot = yaml.load(f) - # s, _ = user1.post(res=m.Snapshot, data=self.json_encode(snapshot)) - s, _ = user1.post(res=m.Snapshot, data=snapshot) + s, _ = user1.post(res=m.Snapshot, data=self.json_encode(snapshot)) if s.get('uuid', None) == 'ec23c11b-80b6-42cd-ac5c-73ba7acddbc4': sample_pc = s['device']['id'] sample_pc_devicehub_id = s['device']['devicehubID'] else: pcs.add(s['device']['id']) - if s.get('uuid', None) == 'de4f495e-c58b-40e1-a33e-46ab5e84767e': # oreo + if ( + s.get('uuid', None) == 'de4f495e-c58b-40e1-a33e-46ab5e84767e' + ): # oreo # Make one hdd ErasePhysical - hdd = next(hdd for hdd in s['components'] if hdd['type'] == 'HardDrive') - user1.post({'type': 'ErasePhysical', 'method': 'Shred', 'device': hdd['id']}, - res=m.Action) + hdd = next( + hdd for hdd in s['components'] if hdd['type'] == 'HardDrive' + ) + user1.post( + { + 'type': 'ErasePhysical', + 'method': 'Shred', + 'device': hdd['id'], + }, + res=m.Action, + ) assert sample_pc print('PC sample is', sample_pc) # Link tags and eTags - for tag, pc in zip((self.TAGS[1], self.TAGS[2], self.ET[0][0], self.ET[1][1]), pcs): + for tag, pc in zip( + (self.TAGS[1], self.TAGS[2], self.ET[0][0], self.ET[1][1]), pcs + ): user1.put({}, res=Tag, item='{}/device/{}'.format(tag, pc), status=204) # Perform generic actions - for pc, model in zip(pcs, - {m.ToRepair, m.Repair, m.ToPrepare, m.Ready, m.ToPrepare, - m.Prepare}): + for pc, model in zip( + pcs, {m.ToRepair, m.Repair, m.ToPrepare, m.Ready, m.ToPrepare, m.Prepare} + ): user1.post({'type': model.t, 'devices': [pc]}, res=m.Action) # Perform a Sell to several devices # user1.post( - # { - # 'type': m.Sell.t, - # 'to': user1.user['individuals'][0]['id'], - # 'devices': list(itertools.islice(pcs, len(pcs) // 2)) - # }, - # res=m.Action) + # { + # 'type': m.Sell.t, + # 'to': user1.user['individuals'][0]['id'], + # 'devices': list(itertools.islice(pcs, len(pcs) // 2)) + # }, + # res=m.Action) lot_user, _ = user1.post({'name': 'LoteStephan'}, res=Lot) @@ -138,26 +166,34 @@ class Dummy: lot_user4, _ = user4.post({'name': 'LoteJordi'}, res=Lot) - lot, _ = user1.post({}, - res=Lot, - item='{}/devices'.format(lot_user['id']), - query=[('id', pc) for pc in itertools.islice(pcs, 1, 4)]) + lot, _ = user1.post( + {}, + res=Lot, + item='{}/devices'.format(lot_user['id']), + query=[('id', pc) for pc in itertools.islice(pcs, 1, 4)], + ) # assert len(lot['devices']) - lot2, _ = user2.post({}, - res=Lot, - item='{}/devices'.format(lot_user2['id']), - query=[('id', pc) for pc in itertools.islice(pcs, 4, 6)]) + lot2, _ = user2.post( + {}, + res=Lot, + item='{}/devices'.format(lot_user2['id']), + query=[('id', pc) for pc in itertools.islice(pcs, 4, 6)], + ) - lot3, _ = user3.post({}, - res=Lot, - item='{}/devices'.format(lot_user3['id']), - query=[('id', pc) for pc in itertools.islice(pcs, 11, 14)]) + lot3, _ = user3.post( + {}, + res=Lot, + item='{}/devices'.format(lot_user3['id']), + query=[('id', pc) for pc in itertools.islice(pcs, 11, 14)], + ) - lot4, _ = user4.post({}, - res=Lot, - item='{}/devices'.format(lot_user4['id']), - query=[('id', pc) for pc in itertools.islice(pcs, 14, 16)]) + lot4, _ = user4.post( + {}, + res=Lot, + item='{}/devices'.format(lot_user4['id']), + query=[('id', pc) for pc in itertools.islice(pcs, 14, 16)], + ) # Keep this at the bottom inventory, _ = user1.get(res=Device) @@ -174,17 +210,19 @@ class Dummy: user1.post({'type': m.ToPrepare.t, 'devices': [sample_pc]}, res=m.Action) user1.post({'type': m.Prepare.t, 'devices': [sample_pc]}, res=m.Action) user1.post({'type': m.Ready.t, 'devices': [sample_pc]}, res=m.Action) - user1.post({'type': m.Price.t, 'device': sample_pc, 'currency': 'EUR', 'price': 85}, - res=m.Action) + user1.post( + {'type': m.Price.t, 'device': sample_pc, 'currency': 'EUR', 'price': 85}, + res=m.Action, + ) # todo test reserve # user1.post( # Sell device - # { - # 'type': m.Sell.t, - # 'to': user1.user['individuals'][0]['id'], - # 'devices': [sample_pc] - # }, - # res=m.Action) + # { + # 'type': m.Sell.t, + # 'to': user1.user['individuals'][0]['id'], + # 'devices': [sample_pc] + # }, + # res=m.Action) # todo Receive user1.get(res=Device, item=sample_pc_devicehub_id) # Test @@ -206,19 +244,20 @@ class Dummy: db.session.add(session_external) db.session.commit() - client = UserClient(self.app, user.email, password, - response_wrapper=self.app.response_class) + client = UserClient( + self.app, user.email, password, response_wrapper=self.app.response_class + ) client.login() return client def json_encode(self, dev: str) -> dict: """Encode json.""" data = {"type": "Snapshot"} - data['data'] = jwt.encode(dev, - self.app.config['JWT_PASS'], - algorithm="HS256", - json_encoder=ereuse_utils.JSONEncoder + data['data'] = jwt.encode( + dev, + self.app.config['JWT_PASS'], + algorithm="HS256", + json_encoder=ereuse_utils.JSONEncoder, ) return data -