# coding: utf-8

"""
    Administrative API

    Manage users, accounts, and account groups in the ThousandEyes platform using the Administrative API. This API provides the following operations to manage your organization:     * `/account-groups`: Account groups are used to divide an organization into different sections. These operations can be used to create, retrieve, update and delete account groups.   * `/users`: Create, retrieve, update and delete users within an organization.    * `/roles`: Create, retrieve and update roles for the current user.    * `/permissions`: Retrieve all assignable permissions. Used in the context of modifying roles.    * `/audit-user-events`: Retrieve all activity log events.    For more information about the administrative models, see [Account Management](https://docs.thousandeyes.com/product-documentation/user-management).

    Generated by OpenAPI Generator (https://openapi-generator.tech)

    Do not edit the class manually.
"""  # noqa: E501


import unittest

from thousandeyes_sdk.core.api_client import ApiClient
from thousandeyes_sdk.core.configuration import Configuration
from sdk_test_support.mock_server import ERROR_STATUS_HEADER, OPERATION_ID_HEADER, MockApiServer

from .mock_manifest import OPERATION_MANIFEST


class IntegrationTestBase(unittest.TestCase):
    def setUp(self) -> None:
        self.server = MockApiServer(OPERATION_MANIFEST)
        self.server.start()
        configuration = Configuration(host=self.server.base_url, access_token="test-token")
        self.api_client = ApiClient(configuration=configuration)

    def tearDown(self) -> None:
        self.server.stop()

    def te_headers(self, operation_id: str, error_status=None):
        headers = {OPERATION_ID_HEADER: operation_id}
        if error_status is not None:
            headers[ERROR_STATUS_HEADER] = error_status
        return headers
