def create_client(self, email='*****@*****.**', password='******'): client = UserClient(self, email, password, response_wrapper=self.response_class) client.login() return client
def test_live(): """Tests inserting a Live into the database and GETting it.""" db_live = models.Live(ip=ipaddress.ip_address('79.147.10.10'), subdivision_confidence=84, subdivision=Subdivision['ES-CA'], city='barcelona', city_confidence=20, isp='acme', device=Desktop(serial_number='sn1', model='ml1', manufacturer='mr1', chassis=ComputerChassis.Docking), organization='acme1', organization_type='acme1bis') db.session.add(db_live) db.session.commit() client = UserClient(app, '*****@*****.**', 'foo', response_wrapper=app.response_class) client.login() live, _ = client.get(res=models.Event, item=str(db_live.id)) assert live['ip'] == '79.147.10.10' assert live['subdivision'] == 'ES-CA' assert live['country'] == 'ES' device, _ = client.get(res=Device, item=live['device']['id']) assert device['physical'] == states.Physical.InUse.name
def user(app: Devicehub) -> UserClient: """Gets a client with a logged-in dummy user.""" with app.app_context(): password = '******' user = create_user(password=password) client = UserClient(app, user.email, password, response_wrapper=app.response_class) client.login() return client
def user_client(self, email: str, password: str, name: str): user = User(email=email, password=password) user.individuals.add(Person(name=name)) db.session.add(user) session_external = Session(user=user, type=SessionType.External) session_internal = Session(user=user, type=SessionType.Internal) db.session.add(session_internal) db.session.add(session_external) db.session.commit() client = UserClient(self.app, user.email, password, response_wrapper=self.app.response_class) client.login() return client
def user(app: Devicehub) -> UserClient: """Gets a client with a logged-in dummy user.""" with app.app_context(): user = create_user() client = UserClient(application=app, response_wrapper=app.response_class, email=user.email, password='******') client.user, _ = client.login(client.email, client.password) return client
def test_price_custom(): computer = Desktop(serial_number='sn1', model='ml1', manufacturer='mr1', chassis=ComputerChassis.Docking) price = models.Price(price=Decimal(25.25), currency=Currency.EUR) price.device = computer assert computer.price == price db.session.add(computer) db.session.commit() client = UserClient(app, '*****@*****.**', 'foo', response_wrapper=app.response_class) client.login() p, _ = client.get(res=models.Event, item=str(price.id)) assert p['device']['id'] == price.device.id == computer.id assert p['price'] == 25.25 assert p['currency'] == Currency.EUR.name == 'EUR' c, _ = client.get(res=Device, item=computer.id) assert c['price']['id'] == p['id']