#!/usr/bin/env python3
import argparse
import os
import sys

os.environ["RUNNING_OPENAPI_CURL_TEST"] = "1"

# check for the venv
from lib import sanity_check

sanity_check.check_venv(__file__)

ZULIP_PATH = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.insert(0, ZULIP_PATH)
os.chdir(ZULIP_PATH)

from zulip import Client

from tools.lib.test_script import assert_provisioning_status_ok
from tools.lib.test_server import test_server_running

usage = """test-api [options]"""
parser = argparse.ArgumentParser(usage)
parser.add_argument('--force',
                    action="store_true",
                    help='Run tests despite possible provisioning problems.')
options = parser.parse_args()

assert_provisioning_status_ok(options.force)

with test_server_running(force=options.force, external_host='zulipdev.com:9981'):
    # Zerver imports should happen after `django.setup()` is run
    # by the test_server_running decorator.
    from zerver.lib.actions import do_create_user
    from zerver.lib.test_helpers import reset_emails_in_zulip_realm
    from zerver.lib.users import get_api_key
    from zerver.models import get_realm, get_user
    from zerver.openapi.javascript_examples import test_js_bindings
    from zerver.openapi.python_examples import test_invalid_api_key, test_the_api
    from zerver.openapi.test_curl_examples import test_generated_curl_examples_for_success

    print("Running API tests...")

    reset_emails_in_zulip_realm()

    # Prepare the admin client
    email = 'iago@zulip.com'  # Iago is an admin
    realm = get_realm('zulip')
    user = get_user(email, realm)
    # Required to test can_create_users endpoints.
    user.can_create_users = True
    user.save(update_fields=["can_create_users"])

    api_key = get_api_key(user)
    site = 'http://zulip.zulipdev.com:9981'
    client = Client(
        email=email,
        api_key=api_key,
        site=site,
    )

    # Prepare a generic bot client for curl testing
    email = 'default-bot@zulip.com'
    realm = get_realm("zulip")
    bot_user = get_user(email, realm)
    api_key = get_api_key(bot_user)
    bot_client = Client(
        email=email,
        api_key=api_key,
        site=site,
    )

    # Prepare the non-admin client
    email = 'guest@zulip.com'  # guest is not an admin
    guest_user = do_create_user(
        'guest@zulip.com',
        'secret',
        get_realm('zulip'),
        'Mr. Guest',
    )
    api_key = get_api_key(guest_user)
    nonadmin_client = Client(
        email=email,
        api_key=api_key,
        site=site,
    )

    test_the_api(client, nonadmin_client)
    test_generated_curl_examples_for_success(client)
    test_js_bindings(client)

    # Test error payloads
    client = Client(
        email=email,
        api_key='X'*32,
        site=site,
    )
    test_invalid_api_key(client)


print("API tests passed!")