mirror of
https://github.com/aljazceru/CTFd.git
synced 2025-12-17 05:54:19 +01:00
49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
from CTFd.models import Users
|
|
from CTFd.utils import set_config
|
|
from tests.helpers import create_ctfd, destroy_ctfd, login_with_mlc, register_user
|
|
|
|
|
|
def test_num_users_oauth_limit():
|
|
"""Only num_users users can be created even via MLC"""
|
|
app = create_ctfd()
|
|
app.config.update(
|
|
{
|
|
"OAUTH_CLIENT_ID": "ctfd_testing_client_id",
|
|
"OAUTH_CLIENT_SECRET": "ctfd_testing_client_secret",
|
|
"OAUTH_AUTHORIZATION_ENDPOINT": "http://auth.localhost/oauth/authorize",
|
|
"OAUTH_TOKEN_ENDPOINT": "http://auth.localhost/oauth/token",
|
|
"OAUTH_API_ENDPOINT": "http://api.localhost/user",
|
|
}
|
|
)
|
|
with app.app_context():
|
|
register_user(app)
|
|
# There should be the admin and our registered user
|
|
assert Users.query.count() == 2
|
|
set_config("num_users", 1)
|
|
|
|
# This registration should fail and we should still have 2 users
|
|
login_with_mlc(
|
|
app,
|
|
name="foobarbaz",
|
|
email="foobarbaz@a.com",
|
|
oauth_id=111,
|
|
scope="profile",
|
|
raise_for_error=False,
|
|
)
|
|
assert Users.query.count() == 2
|
|
|
|
# We increment num_users to 2 and then login again
|
|
set_config("num_users", 2)
|
|
login_with_mlc(
|
|
app,
|
|
name="foobarbaz",
|
|
email="foobarbaz@a.com",
|
|
oauth_id=111,
|
|
scope="profile",
|
|
)
|
|
# The above login should have succeeded
|
|
assert Users.query.count() == 3
|
|
destroy_ctfd(app)
|