mirror of
https://github.com/dirkjanm/ROADtools
synced 2026-06-08 13:48:32 +00:00
Dirk-jan Mollema
62 lines
1.9 KiB
Python
62 lines
1.9 KiB
Python
from roadtools.roadrecon.server import create_app_test, db
|
|
from sqlalchemy import func
|
|
from roadtools.roadlib.metadef.database import Policy
|
|
from flask_sqlalchemy import SQLAlchemy
|
|
|
|
import pytest
|
|
|
|
@pytest.fixture(scope='module')
|
|
def app():
|
|
app = create_app_test()
|
|
return app
|
|
|
|
def test_db_has_users(client):
|
|
"""Test if there are users in the db"""
|
|
|
|
rv = client.get('/api/users')
|
|
assert len(rv.json) > 5
|
|
for user in rv.json:
|
|
assert user['userPrincipalName'] is not None
|
|
assert user['objectId'] is not None
|
|
|
|
def test_db_has_devices(client):
|
|
"""Test if there are devices in the db"""
|
|
|
|
rv = client.get('/api/devices')
|
|
assert len(rv.json) > 5
|
|
for device in rv.json:
|
|
assert device['deviceId'] is not None
|
|
assert device['objectId'] is not None
|
|
|
|
def test_db_has_devices_owners(client):
|
|
"""Test if there are devices with an owner in the db"""
|
|
|
|
rv = client.get('/api/devices')
|
|
assert len(rv.json) > 5
|
|
owners = 0
|
|
for device in rv.json:
|
|
ddata = client.get('/api/devices/' + device['objectId'])
|
|
ddata_json = ddata.json
|
|
owners += len(ddata_json['owner'])
|
|
for owner in ddata_json['owner']:
|
|
assert 'displayName' in owner
|
|
|
|
assert owners > 0
|
|
|
|
def test_sp_links_work(client):
|
|
"""Test if the ServicePrincipal links work"""
|
|
|
|
rv = client.get('/api/serviceprincipals')
|
|
assert len(rv.json) > 5
|
|
outlinks = 0
|
|
for sp in rv.json:
|
|
ddata = client.get('/api/serviceprincipals/' + sp['objectId'])
|
|
ddata_json = ddata.json
|
|
outlinks += len(ddata_json['oauth2PermissionGrants'])
|
|
for owner in ddata_json['oauth2PermissionGrants']:
|
|
assert 'resourceId' in owner
|
|
outlinks += len(ddata_json['appRolesAssigned'])
|
|
for owner in ddata_json['appRolesAssigned']:
|
|
assert 'principalId' in owner
|
|
assert outlinks > 0
|