vishalbhat07/ctfd

1#!/usr/bin/env python2# -*- coding: utf-8 -*-3 4from CTFd.utils import get_config5from CTFd.utils.security.csrf import generate_nonce6from CTFd.utils.security.signing import serialize7from tests.helpers import create_ctfd, destroy_ctfd, login_as_user, register_user8 9 10def test_setup_integrations():11    app = create_ctfd()12    with app.app_context():13        register_user(app)14        user = login_as_user(app)15        r = user.get("/setup/integrations")16        assert r.status_code == 40317 18        admin = login_as_user(app, "admin")19        r = admin.get("/setup/integrations")20        assert r.status_code == 40321 22        admin = login_as_user(app, "admin")23 24        url = "/setup/integrations?state={state}&mlc_client_id=client_id&mlc_client_secret=client_secret&name=mlc".format(25            state=serialize(generate_nonce())26        )27        r = admin.get(url)28        assert r.status_code == 20029        assert get_config("oauth_client_id") == "client_id"30        assert get_config("oauth_client_secret") == "client_secret"31    destroy_ctfd(app)32