# coding: utf-8

"""
    Credentials API

    Manage credentials for transaction tests using the Credentials API.  The following permissions are required to access Credentials API operations:  * `Settings Tests Read` for read operations. * `Settings Tests Update` for write operations. * `View sensitive data in web transaction scripts` to view the encrypted value property of credentials. * `Settings Tests Create Transaction (Tx) Tests` to create credentials.  For more information about credentials, see [Working With Secure Credentials](https://docs.thousandeyes.com/product-documentation/browser-synthetics/transaction-tests/getting-started/working-with-secure-credentials). 

    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
