# coding: utf-8

"""
    Tags API

    The ThousandEyes Tags API provides a tagging system with key/value pairs. It allows you to tag assets within the ThousandEyes platform (such as agents, tests, or dashboards) with meaningful metadata. For example: `branch:sfo`, `branch:nyc`, and `team:netops`.  This feature provides:  * Support for automation. * Powerful and flexible reports/dashboards. * Support for third-party integrations.  Things to note with the ThousandEyes Tags API:  * Tags are backwards-compatible with existing labels. * Tags are separated by Tests (CEA), Agents (CEA), Endpoint Agents, Scheduled Endpoint Tests, and Reports. A single tag can only apply to one type of target object, so each tag must specify the target type of object via a `type` field. * Tags are defined in a single table so that they can be represented using a single model - `Tag`. 

    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
