# coding: utf-8

"""
    Alerts API

    **Note:** API operations for the creation or retrieval of API, Page Load, or Web-Transaction alert rules are not available for ThousandEyes for Government instance.  You can manage the following alert functionalities on the ThousandEyes platform using the Alerts API:  * **Alerts**: Retrieve alert details. Alerts are assigned to tests through alert rules.  * **Alert Rules**: Conditions that you configure in order to highlight or be notified of events of interest in your ThousandEyes tests. When an alert rule’s conditions are met, the associated alert is triggered and the alert becomes active. It remains active until the alert is cleared. Alert rules are reusable across multiple tests..  * **Alert Suppression Windows**: Suppress alerts for tests during periods such as planned maintenance. Windows can be one-time events or recurring events to handle periodic occurrences such as monthly downtime for maintenance.  For more information about the alerts, see [Alerts](https://docs.thousandeyes.com/product-documentation/alerts). 

    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
