def test_token_time_limits_reflect_server_limits(app):
"""Test that if a user's token time limits are bounded to those current on the server."""
with app.app_context():
user = User(username="******", password="******")
user_group = Group(name="TEST_USER", user_group=True)
user.groups.append(user_group)
server = Server(
name="TEST_SERVER",
longest_token_life=2880,
latest_token_expiry=datetime.datetime(2020, 1, 1),
)
token_limits = GroupServerTokenLimits(
group=user_group,
longest_life=2880,
latest_end=datetime.datetime(2020, 1, 1),
server=server,
)
db.session.add(user)
db.session.add(user_group)
db.session.add(server)
db.session.add(token_limits)
db.session.commit()
limits = user.token_limits(server)
assert 2880 == limits["longest_life"]
assert datetime.datetime(2020, 1, 1) == limits["latest_end"]
server.longest_token_life = 10
limits = user.token_limits(server)
assert 10 == limits["longest_life"]
server.latest_token_expiry = datetime.datetime(2019, 1, 1)
limits = user.token_limits(server)
assert datetime.datetime(2019, 1, 1) == limits["latest_end"]
def test_add_admin_promotes(app):
"""
Test that we can promote an existing user to admin.
"""
with app.app_context():
user = User(username="******")
user.password = "******"
original_password_hash = user.password
db.session.add(user)
db.session.commit()
runner = app.test_cli_runner()
result = runner.invoke(
add_admin_command, ["DUMMY_ADMINISTRATOR", "DUMMY_ADMINISTRATOR_PASSWORD"]
)
user = User.query.filter(User.username == "DUMMY_ADMINISTRATOR").first()
assert user.is_admin
assert user.password != original_password_hash
def test_user(app):
with app.app_context():
user = User(username="******", password="******")
ug = Group(name="TEST_USER", user_group=True, members=[user])
db.session.add(user)
db.session.add(ug)
db.session.commit()
return TestUser(user.id, user.username, "TEST_USER_PASSWORD")
def test_disallow_right_on_server_disallows_for_group(app):
"""Test that if a claim is disallowed on a server, it will be disallowed even if previously granted to groups."""
with app.app_context():
session = db.session
user = User(username="******", password="******")
user_group = Group(name="TEST_USER", user_group=True)
user.groups.append(user_group)
session.add(user)
session.add(user_group)
server = Server(
name="TEST_SERVER",
longest_token_life=2880,
latest_token_expiry=datetime.datetime(2020, 1, 1),
)
session.add(server)
server_capability = ServerCapability(
server=server,
capability="get_result&DUMMY_ROUTE",
enabled=True,
capability_hash=md5(b"get_result&DUMMY_ROUTE").hexdigest(),
)
session.add(server_capability)
gsp = GroupServerPermission(group=user_group,
server_capability=server_capability)
session.add(gsp)
token_limits = GroupServerTokenLimits(
group=user_group,
longest_life=1440,
latest_end=datetime.datetime(2019, 1, 1),
server=server,
)
session.add(token_limits)
session.commit()
claims = user.allowed_claims(server)
assert "get_result&DUMMY_ROUTE" in claims
session.delete(gsp)
session.commit()
claims = user.allowed_claims(server)
assert [] == claims
def test_two_factor_auth_user(app):
with app.app_context():
user = User(username="******", password="******")
ug = Group(name="TEST_FACTOR_USER", user_group=True, members=[user])
secret = pyotp.random_base32()
auth = TwoFactorAuth(user=user, enabled=True)
auth.secret_key = secret
otp_generator = partial(get_two_factor_code, secret)
db.session.add(user)
db.session.add(auth)
db.session.add(ug)
db.session.commit()
backup_codes = generate_backup_codes()
for code in backup_codes:
backup = TwoFactorBackup(auth_id=auth.user_id)
backup.backup_code = code
db.session.add(backup)
db.session.commit()
return TestTwoFactorUser(user.id, user.username, "TEST_USER_PASSWORD",
otp_generator, backup_codes)