# coding: utf-8

"""
    Agents API

     ## Overview Manage Cloud and Enterprise Agents available to your account in ThousandEyes.

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

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


import json
import unittest
import thousandeyes_sdk.agents.models

from thousandeyes_sdk.core.exceptions import ApiException
from thousandeyes_sdk.agents.api.tests_assignment_on_agents_api import TestsAssignmentOnAgentsApi
from .integration_test_utils import IntegrationTestBase
from .test_utils import assert_constructed_model_matches_example_json


class TestTestsAssignmentOnAgentsApiIntegration(IntegrationTestBase):
    """TestsAssignmentOnAgentsApi integration test stubs"""

    def setUp(self) -> None:
        super().setUp()
        self.api = TestsAssignmentOnAgentsApi(self.api_client)


    def test_assign_tests_happy_path(self) -> None:
        """Integration test for assign_tests success path"""
        request_body_json = """
        
                {
                  "testIds" : [ "281474976710706" ]
                }
        
                """
        agent_tests_assign_request = thousandeyes_sdk.agents.models.AgentTestsAssignRequest.from_json(request_body_json)
        agent_id = '281474976710706'
        aid = '1234'
        response_body_json = """
                {
                  "agentId" : "281474976710706",
                  "agentType" : "cloud",
                  "_links" : {
                    "self" : {
                      "hreflang" : "hreflang",
                      "templated" : true,
                      "profile" : "profile",
                      "name" : "name",
                      "href" : "https://api.thousandeyes.com/v7/link/to/resource/id",
                      "type" : "type",
                      "deprecation" : "deprecation",
                      "title" : "title"
                    }
                  },
                  "prefix" : "99.128.0.0/11",
                  "coordinates" : {
                    "latitude" : 37.77493,
                    "longitude" : -122.41942
                  },
                  "agentName" : "thousandeyes-stg-va-254",
                  "networkProviderInfo" : {
                    "asn" : 7018,
                    "name" : "AT&T Services, Inc.",
                    "type" : "isp"
                  },
                  "countryId" : "US",
                  "enabled" : true,
                  "network" : "AT&T Services, Inc. (AS 7018)",
                  "labels" : [ {
                    "labelId" : "11",
                    "name" : "Label name"
                  }, {
                    "labelId" : "11",
                    "name" : "Label name"
                  } ],
                  "tags" : [ {
                    "id" : "5aeab5d5-0d34-4d44-a7ac-fb440185295c",
                    "value" : "San Francisco",
                    "key" : "Location"
                  }, {
                    "id" : "5aeab5d5-0d34-4d44-a7ac-fb440185295c",
                    "value" : "San Francisco",
                    "key" : "Location"
                  } ],
                  "tests" : [ {
                    "_links" : {
                      "testResults" : [ {
                        "href" : "https://api.thousandeyes.com/v7/test-results/281474976710706/network"
                      }, {
                        "href" : "https://api.thousandeyes.com/v7/test-results/281474976710706/path-vis"
                      } ],
                      "self" : {
                        "hreflang" : "hreflang",
                        "templated" : true,
                        "profile" : "profile",
                        "name" : "name",
                        "href" : "https://api.thousandeyes.com/v7/link/to/resource/id",
                        "type" : "type",
                        "deprecation" : "deprecation",
                        "title" : "title"
                      }
                    },
                    "liveShare" : false,
                    "savedEvent" : true,
                    "description" : "ThousandEyes Test",
                    "type" : "agent-to-server",
                    "enabled" : true,
                    "createdDate" : "2022-07-17T22:00:54Z",
                    "createdBy" : "user@user.com",
                    "modifiedDate" : "2022-07-17T22:00:54Z",
                    "interval" : 60,
                    "modifiedBy" : "user@user.com",
                    "testId" : "281474976710706",
                    "alertsEnabled" : true,
                    "testName" : "ThousandEyes Test"
                  }, {
                    "_links" : {
                      "testResults" : [ {
                        "href" : "https://api.thousandeyes.com/v7/test-results/281474976710706/network"
                      }, {
                        "href" : "https://api.thousandeyes.com/v7/test-results/281474976710706/path-vis"
                      } ],
                      "self" : {
                        "hreflang" : "hreflang",
                        "templated" : true,
                        "profile" : "profile",
                        "name" : "name",
                        "href" : "https://api.thousandeyes.com/v7/link/to/resource/id",
                        "type" : "type",
                        "deprecation" : "deprecation",
                        "title" : "title"
                      }
                    },
                    "liveShare" : false,
                    "savedEvent" : true,
                    "description" : "ThousandEyes Test",
                    "type" : "agent-to-server",
                    "enabled" : true,
                    "createdDate" : "2022-07-17T22:00:54Z",
                    "createdBy" : "user@user.com",
                    "modifiedDate" : "2022-07-17T22:00:54Z",
                    "interval" : 60,
                    "modifiedBy" : "user@user.com",
                    "testId" : "281474976710706",
                    "alertsEnabled" : true,
                    "testName" : "ThousandEyes Test"
                  } ],
                  "publicIpAddresses" : [ "192.168.1.78", "f9b2:3a21:f25c:d300:03f4:586d:f8d6:4e1c" ],
                  "ipAddresses" : [ "99.139.65.220", "9bbd:8a0a:a257:5876:288b:6cb2:3f36:64ce" ],
                  "location" : "San Francisco Bay Area",
                  "verifySslCertificates" : true
                }
                """
        expected_response = json.loads(response_body_json)
        response = self.api.assign_tests(

            agent_id=agent_id,

            agent_tests_assign_request=agent_tests_assign_request,

            aid=aid,

            _headers=self.te_headers("assign_tests"),
        )
        assert_constructed_model_matches_example_json(response, expected_response)


    def test_assign_tests_error_400(self) -> None:
        """Integration test for assign_tests error path (HTTP 400)"""
        request_body_json = """
        
                {
                  "testIds" : [ "281474976710706" ]
                }
        
                """
        agent_tests_assign_request = thousandeyes_sdk.agents.models.AgentTestsAssignRequest.from_json(request_body_json)
        agent_id = '281474976710706'
        aid = '1234'
        error_body_json = """
                {
                  "instance" : "instance",
                  "detail" : "detail",
                  "type" : "type",
                  "title" : "title",
                  "errors" : [ {
                    "code" : "code",
                    "field" : "field",
                    "message" : "message"
                  }, {
                    "code" : "code",
                    "field" : "field",
                    "message" : "message"
                  } ],
                  "status" : 0
                }
                """
        expected_error = json.loads(error_body_json)
        with self.assertRaises(
            ApiException.exception_class_for_http_status(400)
        ) as context:
            self.api.assign_tests(

                agent_id=agent_id,

                agent_tests_assign_request=agent_tests_assign_request,

                aid=aid,

                _headers=self.te_headers("assign_tests", error_status="400"),
            )
        assert_constructed_model_matches_example_json(context.exception.data, expected_error)


    def test_assign_tests_error_401(self) -> None:
        """Integration test for assign_tests error path (HTTP 401)"""
        request_body_json = """
        
                {
                  "testIds" : [ "281474976710706" ]
                }
        
                """
        agent_tests_assign_request = thousandeyes_sdk.agents.models.AgentTestsAssignRequest.from_json(request_body_json)
        agent_id = '281474976710706'
        aid = '1234'
        error_body_json = """
                {
                  "error_description" : "Invalid access token",
                  "error" : "invalid_token"
                }
                """
        expected_error = json.loads(error_body_json)
        with self.assertRaises(
            ApiException.exception_class_for_http_status(401)
        ) as context:
            self.api.assign_tests(

                agent_id=agent_id,

                agent_tests_assign_request=agent_tests_assign_request,

                aid=aid,

                _headers=self.te_headers("assign_tests", error_status="401"),
            )
        assert_constructed_model_matches_example_json(context.exception.data, expected_error)


    def test_assign_tests_error_403(self) -> None:
        """Integration test for assign_tests error path (HTTP 403)"""
        request_body_json = """
        
                {
                  "testIds" : [ "281474976710706" ]
                }
        
                """
        agent_tests_assign_request = thousandeyes_sdk.agents.models.AgentTestsAssignRequest.from_json(request_body_json)
        agent_id = '281474976710706'
        aid = '1234'
        error_body_json = """
                {
                  "instance" : "instance",
                  "detail" : "detail",
                  "type" : "type",
                  "title" : "title",
                  "status" : 0
                }
                """
        expected_error = json.loads(error_body_json)
        with self.assertRaises(
            ApiException.exception_class_for_http_status(403)
        ) as context:
            self.api.assign_tests(

                agent_id=agent_id,

                agent_tests_assign_request=agent_tests_assign_request,

                aid=aid,

                _headers=self.te_headers("assign_tests", error_status="403"),
            )
        assert_constructed_model_matches_example_json(context.exception.data, expected_error)


    def test_assign_tests_error_404(self) -> None:
        """Integration test for assign_tests error path (HTTP 404)"""
        request_body_json = """
        
                {
                  "testIds" : [ "281474976710706" ]
                }
        
                """
        agent_tests_assign_request = thousandeyes_sdk.agents.models.AgentTestsAssignRequest.from_json(request_body_json)
        agent_id = '281474976710706'
        aid = '1234'
        error_body_json = """
                {
                  "instance" : "instance",
                  "detail" : "detail",
                  "type" : "type",
                  "title" : "title",
                  "status" : 0
                }
                """
        expected_error = json.loads(error_body_json)
        with self.assertRaises(
            ApiException.exception_class_for_http_status(404)
        ) as context:
            self.api.assign_tests(

                agent_id=agent_id,

                agent_tests_assign_request=agent_tests_assign_request,

                aid=aid,

                _headers=self.te_headers("assign_tests", error_status="404"),
            )
        assert_constructed_model_matches_example_json(context.exception.data, expected_error)


    def test_assign_tests_error_429(self) -> None:
        """Integration test for assign_tests error path (HTTP 429)"""
        request_body_json = """
        
                {
                  "testIds" : [ "281474976710706" ]
                }
        
                """
        agent_tests_assign_request = thousandeyes_sdk.agents.models.AgentTestsAssignRequest.from_json(request_body_json)
        agent_id = '281474976710706'
        aid = '1234'
        error_body_json = """
                {
                  "instance" : "instance",
                  "detail" : "detail",
                  "type" : "type",
                  "title" : "title",
                  "status" : 0
                }
                """
        expected_error = json.loads(error_body_json)
        with self.assertRaises(
            ApiException.exception_class_for_http_status(429)
        ) as context:
            self.api.assign_tests(

                agent_id=agent_id,

                agent_tests_assign_request=agent_tests_assign_request,

                aid=aid,

                _headers=self.te_headers("assign_tests", error_status="429"),
            )
        assert_constructed_model_matches_example_json(context.exception.data, expected_error)


    def test_assign_tests_error_500(self) -> None:
        """Integration test for assign_tests error path (HTTP 500)"""
        request_body_json = """
        
                {
                  "testIds" : [ "281474976710706" ]
                }
        
                """
        agent_tests_assign_request = thousandeyes_sdk.agents.models.AgentTestsAssignRequest.from_json(request_body_json)
        agent_id = '281474976710706'
        aid = '1234'
        error_body_json = """
                {
                  "instance" : "instance",
                  "detail" : "detail",
                  "type" : "type",
                  "title" : "title",
                  "status" : 0
                }
                """
        expected_error = json.loads(error_body_json)
        with self.assertRaises(
            ApiException.exception_class_for_http_status(500)
        ) as context:
            self.api.assign_tests(

                agent_id=agent_id,

                agent_tests_assign_request=agent_tests_assign_request,

                aid=aid,

                _headers=self.te_headers("assign_tests", error_status="500"),
            )
        assert_constructed_model_matches_example_json(context.exception.data, expected_error)


    def test_assign_tests_error_502(self) -> None:
        """Integration test for assign_tests error path (HTTP 502)"""
        request_body_json = """
        
                {
                  "testIds" : [ "281474976710706" ]
                }
        
                """
        agent_tests_assign_request = thousandeyes_sdk.agents.models.AgentTestsAssignRequest.from_json(request_body_json)
        agent_id = '281474976710706'
        aid = '1234'
        error_body_json = """
                {
                  "instance" : "instance",
                  "detail" : "detail",
                  "type" : "type",
                  "title" : "title",
                  "status" : 0
                }
                """
        expected_error = json.loads(error_body_json)
        with self.assertRaises(
            ApiException.exception_class_for_http_status(502)
        ) as context:
            self.api.assign_tests(

                agent_id=agent_id,

                agent_tests_assign_request=agent_tests_assign_request,

                aid=aid,

                _headers=self.te_headers("assign_tests", error_status="502"),
            )
        assert_constructed_model_matches_example_json(context.exception.data, expected_error)




    def test_overwrite_tests_happy_path(self) -> None:
        """Integration test for overwrite_tests success path"""
        request_body_json = """
        
                {
                  "testIds" : [ "281474976710706" ]
                }
        
                """
        agent_tests_assign_request = thousandeyes_sdk.agents.models.AgentTestsAssignRequest.from_json(request_body_json)
        agent_id = '281474976710706'
        aid = '1234'
        response_body_json = """
                {
                  "agentId" : "281474976710706",
                  "agentType" : "cloud",
                  "_links" : {
                    "self" : {
                      "hreflang" : "hreflang",
                      "templated" : true,
                      "profile" : "profile",
                      "name" : "name",
                      "href" : "https://api.thousandeyes.com/v7/link/to/resource/id",
                      "type" : "type",
                      "deprecation" : "deprecation",
                      "title" : "title"
                    }
                  },
                  "prefix" : "99.128.0.0/11",
                  "coordinates" : {
                    "latitude" : 37.77493,
                    "longitude" : -122.41942
                  },
                  "agentName" : "thousandeyes-stg-va-254",
                  "networkProviderInfo" : {
                    "asn" : 7018,
                    "name" : "AT&T Services, Inc.",
                    "type" : "isp"
                  },
                  "countryId" : "US",
                  "enabled" : true,
                  "network" : "AT&T Services, Inc. (AS 7018)",
                  "labels" : [ {
                    "labelId" : "11",
                    "name" : "Label name"
                  }, {
                    "labelId" : "11",
                    "name" : "Label name"
                  } ],
                  "tags" : [ {
                    "id" : "5aeab5d5-0d34-4d44-a7ac-fb440185295c",
                    "value" : "San Francisco",
                    "key" : "Location"
                  }, {
                    "id" : "5aeab5d5-0d34-4d44-a7ac-fb440185295c",
                    "value" : "San Francisco",
                    "key" : "Location"
                  } ],
                  "tests" : [ {
                    "_links" : {
                      "testResults" : [ {
                        "href" : "https://api.thousandeyes.com/v7/test-results/281474976710706/network"
                      }, {
                        "href" : "https://api.thousandeyes.com/v7/test-results/281474976710706/path-vis"
                      } ],
                      "self" : {
                        "hreflang" : "hreflang",
                        "templated" : true,
                        "profile" : "profile",
                        "name" : "name",
                        "href" : "https://api.thousandeyes.com/v7/link/to/resource/id",
                        "type" : "type",
                        "deprecation" : "deprecation",
                        "title" : "title"
                      }
                    },
                    "liveShare" : false,
                    "savedEvent" : true,
                    "description" : "ThousandEyes Test",
                    "type" : "agent-to-server",
                    "enabled" : true,
                    "createdDate" : "2022-07-17T22:00:54Z",
                    "createdBy" : "user@user.com",
                    "modifiedDate" : "2022-07-17T22:00:54Z",
                    "interval" : 60,
                    "modifiedBy" : "user@user.com",
                    "testId" : "281474976710706",
                    "alertsEnabled" : true,
                    "testName" : "ThousandEyes Test"
                  }, {
                    "_links" : {
                      "testResults" : [ {
                        "href" : "https://api.thousandeyes.com/v7/test-results/281474976710706/network"
                      }, {
                        "href" : "https://api.thousandeyes.com/v7/test-results/281474976710706/path-vis"
                      } ],
                      "self" : {
                        "hreflang" : "hreflang",
                        "templated" : true,
                        "profile" : "profile",
                        "name" : "name",
                        "href" : "https://api.thousandeyes.com/v7/link/to/resource/id",
                        "type" : "type",
                        "deprecation" : "deprecation",
                        "title" : "title"
                      }
                    },
                    "liveShare" : false,
                    "savedEvent" : true,
                    "description" : "ThousandEyes Test",
                    "type" : "agent-to-server",
                    "enabled" : true,
                    "createdDate" : "2022-07-17T22:00:54Z",
                    "createdBy" : "user@user.com",
                    "modifiedDate" : "2022-07-17T22:00:54Z",
                    "interval" : 60,
                    "modifiedBy" : "user@user.com",
                    "testId" : "281474976710706",
                    "alertsEnabled" : true,
                    "testName" : "ThousandEyes Test"
                  } ],
                  "publicIpAddresses" : [ "192.168.1.78", "f9b2:3a21:f25c:d300:03f4:586d:f8d6:4e1c" ],
                  "ipAddresses" : [ "99.139.65.220", "9bbd:8a0a:a257:5876:288b:6cb2:3f36:64ce" ],
                  "location" : "San Francisco Bay Area",
                  "verifySslCertificates" : true
                }
                """
        expected_response = json.loads(response_body_json)
        response = self.api.overwrite_tests(

            agent_id=agent_id,

            agent_tests_assign_request=agent_tests_assign_request,

            aid=aid,

            _headers=self.te_headers("overwrite_tests"),
        )
        assert_constructed_model_matches_example_json(response, expected_response)


    def test_overwrite_tests_error_400(self) -> None:
        """Integration test for overwrite_tests error path (HTTP 400)"""
        request_body_json = """
        
                {
                  "testIds" : [ "281474976710706" ]
                }
        
                """
        agent_tests_assign_request = thousandeyes_sdk.agents.models.AgentTestsAssignRequest.from_json(request_body_json)
        agent_id = '281474976710706'
        aid = '1234'
        error_body_json = """
                {
                  "instance" : "instance",
                  "detail" : "detail",
                  "type" : "type",
                  "title" : "title",
                  "errors" : [ {
                    "code" : "code",
                    "field" : "field",
                    "message" : "message"
                  }, {
                    "code" : "code",
                    "field" : "field",
                    "message" : "message"
                  } ],
                  "status" : 0
                }
                """
        expected_error = json.loads(error_body_json)
        with self.assertRaises(
            ApiException.exception_class_for_http_status(400)
        ) as context:
            self.api.overwrite_tests(

                agent_id=agent_id,

                agent_tests_assign_request=agent_tests_assign_request,

                aid=aid,

                _headers=self.te_headers("overwrite_tests", error_status="400"),
            )
        assert_constructed_model_matches_example_json(context.exception.data, expected_error)


    def test_overwrite_tests_error_401(self) -> None:
        """Integration test for overwrite_tests error path (HTTP 401)"""
        request_body_json = """
        
                {
                  "testIds" : [ "281474976710706" ]
                }
        
                """
        agent_tests_assign_request = thousandeyes_sdk.agents.models.AgentTestsAssignRequest.from_json(request_body_json)
        agent_id = '281474976710706'
        aid = '1234'
        error_body_json = """
                {
                  "error_description" : "Invalid access token",
                  "error" : "invalid_token"
                }
                """
        expected_error = json.loads(error_body_json)
        with self.assertRaises(
            ApiException.exception_class_for_http_status(401)
        ) as context:
            self.api.overwrite_tests(

                agent_id=agent_id,

                agent_tests_assign_request=agent_tests_assign_request,

                aid=aid,

                _headers=self.te_headers("overwrite_tests", error_status="401"),
            )
        assert_constructed_model_matches_example_json(context.exception.data, expected_error)


    def test_overwrite_tests_error_403(self) -> None:
        """Integration test for overwrite_tests error path (HTTP 403)"""
        request_body_json = """
        
                {
                  "testIds" : [ "281474976710706" ]
                }
        
                """
        agent_tests_assign_request = thousandeyes_sdk.agents.models.AgentTestsAssignRequest.from_json(request_body_json)
        agent_id = '281474976710706'
        aid = '1234'
        error_body_json = """
                {
                  "instance" : "instance",
                  "detail" : "detail",
                  "type" : "type",
                  "title" : "title",
                  "status" : 0
                }
                """
        expected_error = json.loads(error_body_json)
        with self.assertRaises(
            ApiException.exception_class_for_http_status(403)
        ) as context:
            self.api.overwrite_tests(

                agent_id=agent_id,

                agent_tests_assign_request=agent_tests_assign_request,

                aid=aid,

                _headers=self.te_headers("overwrite_tests", error_status="403"),
            )
        assert_constructed_model_matches_example_json(context.exception.data, expected_error)


    def test_overwrite_tests_error_404(self) -> None:
        """Integration test for overwrite_tests error path (HTTP 404)"""
        request_body_json = """
        
                {
                  "testIds" : [ "281474976710706" ]
                }
        
                """
        agent_tests_assign_request = thousandeyes_sdk.agents.models.AgentTestsAssignRequest.from_json(request_body_json)
        agent_id = '281474976710706'
        aid = '1234'
        error_body_json = """
                {
                  "instance" : "instance",
                  "detail" : "detail",
                  "type" : "type",
                  "title" : "title",
                  "status" : 0
                }
                """
        expected_error = json.loads(error_body_json)
        with self.assertRaises(
            ApiException.exception_class_for_http_status(404)
        ) as context:
            self.api.overwrite_tests(

                agent_id=agent_id,

                agent_tests_assign_request=agent_tests_assign_request,

                aid=aid,

                _headers=self.te_headers("overwrite_tests", error_status="404"),
            )
        assert_constructed_model_matches_example_json(context.exception.data, expected_error)


    def test_overwrite_tests_error_429(self) -> None:
        """Integration test for overwrite_tests error path (HTTP 429)"""
        request_body_json = """
        
                {
                  "testIds" : [ "281474976710706" ]
                }
        
                """
        agent_tests_assign_request = thousandeyes_sdk.agents.models.AgentTestsAssignRequest.from_json(request_body_json)
        agent_id = '281474976710706'
        aid = '1234'
        error_body_json = """
                {
                  "instance" : "instance",
                  "detail" : "detail",
                  "type" : "type",
                  "title" : "title",
                  "status" : 0
                }
                """
        expected_error = json.loads(error_body_json)
        with self.assertRaises(
            ApiException.exception_class_for_http_status(429)
        ) as context:
            self.api.overwrite_tests(

                agent_id=agent_id,

                agent_tests_assign_request=agent_tests_assign_request,

                aid=aid,

                _headers=self.te_headers("overwrite_tests", error_status="429"),
            )
        assert_constructed_model_matches_example_json(context.exception.data, expected_error)


    def test_overwrite_tests_error_500(self) -> None:
        """Integration test for overwrite_tests error path (HTTP 500)"""
        request_body_json = """
        
                {
                  "testIds" : [ "281474976710706" ]
                }
        
                """
        agent_tests_assign_request = thousandeyes_sdk.agents.models.AgentTestsAssignRequest.from_json(request_body_json)
        agent_id = '281474976710706'
        aid = '1234'
        error_body_json = """
                {
                  "instance" : "instance",
                  "detail" : "detail",
                  "type" : "type",
                  "title" : "title",
                  "status" : 0
                }
                """
        expected_error = json.loads(error_body_json)
        with self.assertRaises(
            ApiException.exception_class_for_http_status(500)
        ) as context:
            self.api.overwrite_tests(

                agent_id=agent_id,

                agent_tests_assign_request=agent_tests_assign_request,

                aid=aid,

                _headers=self.te_headers("overwrite_tests", error_status="500"),
            )
        assert_constructed_model_matches_example_json(context.exception.data, expected_error)


    def test_overwrite_tests_error_502(self) -> None:
        """Integration test for overwrite_tests error path (HTTP 502)"""
        request_body_json = """
        
                {
                  "testIds" : [ "281474976710706" ]
                }
        
                """
        agent_tests_assign_request = thousandeyes_sdk.agents.models.AgentTestsAssignRequest.from_json(request_body_json)
        agent_id = '281474976710706'
        aid = '1234'
        error_body_json = """
                {
                  "instance" : "instance",
                  "detail" : "detail",
                  "type" : "type",
                  "title" : "title",
                  "status" : 0
                }
                """
        expected_error = json.loads(error_body_json)
        with self.assertRaises(
            ApiException.exception_class_for_http_status(502)
        ) as context:
            self.api.overwrite_tests(

                agent_id=agent_id,

                agent_tests_assign_request=agent_tests_assign_request,

                aid=aid,

                _headers=self.te_headers("overwrite_tests", error_status="502"),
            )
        assert_constructed_model_matches_example_json(context.exception.data, expected_error)




    def test_unassign_tests_happy_path(self) -> None:
        """Integration test for unassign_tests success path"""
        request_body_json = """
        
                {
                  "testIds" : [ "281474976710706" ]
                }
        
                """
        agent_tests_assign_request = thousandeyes_sdk.agents.models.AgentTestsAssignRequest.from_json(request_body_json)
        agent_id = '281474976710706'
        aid = '1234'
        response_body_json = """
                {
                  "agentId" : "281474976710706",
                  "agentType" : "cloud",
                  "_links" : {
                    "self" : {
                      "hreflang" : "hreflang",
                      "templated" : true,
                      "profile" : "profile",
                      "name" : "name",
                      "href" : "https://api.thousandeyes.com/v7/link/to/resource/id",
                      "type" : "type",
                      "deprecation" : "deprecation",
                      "title" : "title"
                    }
                  },
                  "prefix" : "99.128.0.0/11",
                  "coordinates" : {
                    "latitude" : 37.77493,
                    "longitude" : -122.41942
                  },
                  "agentName" : "thousandeyes-stg-va-254",
                  "networkProviderInfo" : {
                    "asn" : 7018,
                    "name" : "AT&T Services, Inc.",
                    "type" : "isp"
                  },
                  "countryId" : "US",
                  "enabled" : true,
                  "network" : "AT&T Services, Inc. (AS 7018)",
                  "labels" : [ {
                    "labelId" : "11",
                    "name" : "Label name"
                  }, {
                    "labelId" : "11",
                    "name" : "Label name"
                  } ],
                  "tags" : [ {
                    "id" : "5aeab5d5-0d34-4d44-a7ac-fb440185295c",
                    "value" : "San Francisco",
                    "key" : "Location"
                  }, {
                    "id" : "5aeab5d5-0d34-4d44-a7ac-fb440185295c",
                    "value" : "San Francisco",
                    "key" : "Location"
                  } ],
                  "tests" : [ {
                    "_links" : {
                      "testResults" : [ {
                        "href" : "https://api.thousandeyes.com/v7/test-results/281474976710706/network"
                      }, {
                        "href" : "https://api.thousandeyes.com/v7/test-results/281474976710706/path-vis"
                      } ],
                      "self" : {
                        "hreflang" : "hreflang",
                        "templated" : true,
                        "profile" : "profile",
                        "name" : "name",
                        "href" : "https://api.thousandeyes.com/v7/link/to/resource/id",
                        "type" : "type",
                        "deprecation" : "deprecation",
                        "title" : "title"
                      }
                    },
                    "liveShare" : false,
                    "savedEvent" : true,
                    "description" : "ThousandEyes Test",
                    "type" : "agent-to-server",
                    "enabled" : true,
                    "createdDate" : "2022-07-17T22:00:54Z",
                    "createdBy" : "user@user.com",
                    "modifiedDate" : "2022-07-17T22:00:54Z",
                    "interval" : 60,
                    "modifiedBy" : "user@user.com",
                    "testId" : "281474976710706",
                    "alertsEnabled" : true,
                    "testName" : "ThousandEyes Test"
                  }, {
                    "_links" : {
                      "testResults" : [ {
                        "href" : "https://api.thousandeyes.com/v7/test-results/281474976710706/network"
                      }, {
                        "href" : "https://api.thousandeyes.com/v7/test-results/281474976710706/path-vis"
                      } ],
                      "self" : {
                        "hreflang" : "hreflang",
                        "templated" : true,
                        "profile" : "profile",
                        "name" : "name",
                        "href" : "https://api.thousandeyes.com/v7/link/to/resource/id",
                        "type" : "type",
                        "deprecation" : "deprecation",
                        "title" : "title"
                      }
                    },
                    "liveShare" : false,
                    "savedEvent" : true,
                    "description" : "ThousandEyes Test",
                    "type" : "agent-to-server",
                    "enabled" : true,
                    "createdDate" : "2022-07-17T22:00:54Z",
                    "createdBy" : "user@user.com",
                    "modifiedDate" : "2022-07-17T22:00:54Z",
                    "interval" : 60,
                    "modifiedBy" : "user@user.com",
                    "testId" : "281474976710706",
                    "alertsEnabled" : true,
                    "testName" : "ThousandEyes Test"
                  } ],
                  "publicIpAddresses" : [ "192.168.1.78", "f9b2:3a21:f25c:d300:03f4:586d:f8d6:4e1c" ],
                  "ipAddresses" : [ "99.139.65.220", "9bbd:8a0a:a257:5876:288b:6cb2:3f36:64ce" ],
                  "location" : "San Francisco Bay Area",
                  "verifySslCertificates" : true
                }
                """
        expected_response = json.loads(response_body_json)
        response = self.api.unassign_tests(

            agent_id=agent_id,

            agent_tests_assign_request=agent_tests_assign_request,

            aid=aid,

            _headers=self.te_headers("unassign_tests"),
        )
        assert_constructed_model_matches_example_json(response, expected_response)


    def test_unassign_tests_error_400(self) -> None:
        """Integration test for unassign_tests error path (HTTP 400)"""
        request_body_json = """
        
                {
                  "testIds" : [ "281474976710706" ]
                }
        
                """
        agent_tests_assign_request = thousandeyes_sdk.agents.models.AgentTestsAssignRequest.from_json(request_body_json)
        agent_id = '281474976710706'
        aid = '1234'
        error_body_json = """
                {
                  "instance" : "instance",
                  "detail" : "detail",
                  "type" : "type",
                  "title" : "title",
                  "errors" : [ {
                    "code" : "code",
                    "field" : "field",
                    "message" : "message"
                  }, {
                    "code" : "code",
                    "field" : "field",
                    "message" : "message"
                  } ],
                  "status" : 0
                }
                """
        expected_error = json.loads(error_body_json)
        with self.assertRaises(
            ApiException.exception_class_for_http_status(400)
        ) as context:
            self.api.unassign_tests(

                agent_id=agent_id,

                agent_tests_assign_request=agent_tests_assign_request,

                aid=aid,

                _headers=self.te_headers("unassign_tests", error_status="400"),
            )
        assert_constructed_model_matches_example_json(context.exception.data, expected_error)


    def test_unassign_tests_error_401(self) -> None:
        """Integration test for unassign_tests error path (HTTP 401)"""
        request_body_json = """
        
                {
                  "testIds" : [ "281474976710706" ]
                }
        
                """
        agent_tests_assign_request = thousandeyes_sdk.agents.models.AgentTestsAssignRequest.from_json(request_body_json)
        agent_id = '281474976710706'
        aid = '1234'
        error_body_json = """
                {
                  "error_description" : "Invalid access token",
                  "error" : "invalid_token"
                }
                """
        expected_error = json.loads(error_body_json)
        with self.assertRaises(
            ApiException.exception_class_for_http_status(401)
        ) as context:
            self.api.unassign_tests(

                agent_id=agent_id,

                agent_tests_assign_request=agent_tests_assign_request,

                aid=aid,

                _headers=self.te_headers("unassign_tests", error_status="401"),
            )
        assert_constructed_model_matches_example_json(context.exception.data, expected_error)


    def test_unassign_tests_error_403(self) -> None:
        """Integration test for unassign_tests error path (HTTP 403)"""
        request_body_json = """
        
                {
                  "testIds" : [ "281474976710706" ]
                }
        
                """
        agent_tests_assign_request = thousandeyes_sdk.agents.models.AgentTestsAssignRequest.from_json(request_body_json)
        agent_id = '281474976710706'
        aid = '1234'
        error_body_json = """
                {
                  "instance" : "instance",
                  "detail" : "detail",
                  "type" : "type",
                  "title" : "title",
                  "status" : 0
                }
                """
        expected_error = json.loads(error_body_json)
        with self.assertRaises(
            ApiException.exception_class_for_http_status(403)
        ) as context:
            self.api.unassign_tests(

                agent_id=agent_id,

                agent_tests_assign_request=agent_tests_assign_request,

                aid=aid,

                _headers=self.te_headers("unassign_tests", error_status="403"),
            )
        assert_constructed_model_matches_example_json(context.exception.data, expected_error)


    def test_unassign_tests_error_404(self) -> None:
        """Integration test for unassign_tests error path (HTTP 404)"""
        request_body_json = """
        
                {
                  "testIds" : [ "281474976710706" ]
                }
        
                """
        agent_tests_assign_request = thousandeyes_sdk.agents.models.AgentTestsAssignRequest.from_json(request_body_json)
        agent_id = '281474976710706'
        aid = '1234'
        error_body_json = """
                {
                  "instance" : "instance",
                  "detail" : "detail",
                  "type" : "type",
                  "title" : "title",
                  "status" : 0
                }
                """
        expected_error = json.loads(error_body_json)
        with self.assertRaises(
            ApiException.exception_class_for_http_status(404)
        ) as context:
            self.api.unassign_tests(

                agent_id=agent_id,

                agent_tests_assign_request=agent_tests_assign_request,

                aid=aid,

                _headers=self.te_headers("unassign_tests", error_status="404"),
            )
        assert_constructed_model_matches_example_json(context.exception.data, expected_error)


    def test_unassign_tests_error_429(self) -> None:
        """Integration test for unassign_tests error path (HTTP 429)"""
        request_body_json = """
        
                {
                  "testIds" : [ "281474976710706" ]
                }
        
                """
        agent_tests_assign_request = thousandeyes_sdk.agents.models.AgentTestsAssignRequest.from_json(request_body_json)
        agent_id = '281474976710706'
        aid = '1234'
        error_body_json = """
                {
                  "instance" : "instance",
                  "detail" : "detail",
                  "type" : "type",
                  "title" : "title",
                  "status" : 0
                }
                """
        expected_error = json.loads(error_body_json)
        with self.assertRaises(
            ApiException.exception_class_for_http_status(429)
        ) as context:
            self.api.unassign_tests(

                agent_id=agent_id,

                agent_tests_assign_request=agent_tests_assign_request,

                aid=aid,

                _headers=self.te_headers("unassign_tests", error_status="429"),
            )
        assert_constructed_model_matches_example_json(context.exception.data, expected_error)


    def test_unassign_tests_error_500(self) -> None:
        """Integration test for unassign_tests error path (HTTP 500)"""
        request_body_json = """
        
                {
                  "testIds" : [ "281474976710706" ]
                }
        
                """
        agent_tests_assign_request = thousandeyes_sdk.agents.models.AgentTestsAssignRequest.from_json(request_body_json)
        agent_id = '281474976710706'
        aid = '1234'
        error_body_json = """
                {
                  "instance" : "instance",
                  "detail" : "detail",
                  "type" : "type",
                  "title" : "title",
                  "status" : 0
                }
                """
        expected_error = json.loads(error_body_json)
        with self.assertRaises(
            ApiException.exception_class_for_http_status(500)
        ) as context:
            self.api.unassign_tests(

                agent_id=agent_id,

                agent_tests_assign_request=agent_tests_assign_request,

                aid=aid,

                _headers=self.te_headers("unassign_tests", error_status="500"),
            )
        assert_constructed_model_matches_example_json(context.exception.data, expected_error)


    def test_unassign_tests_error_502(self) -> None:
        """Integration test for unassign_tests error path (HTTP 502)"""
        request_body_json = """
        
                {
                  "testIds" : [ "281474976710706" ]
                }
        
                """
        agent_tests_assign_request = thousandeyes_sdk.agents.models.AgentTestsAssignRequest.from_json(request_body_json)
        agent_id = '281474976710706'
        aid = '1234'
        error_body_json = """
                {
                  "instance" : "instance",
                  "detail" : "detail",
                  "type" : "type",
                  "title" : "title",
                  "status" : 0
                }
                """
        expected_error = json.loads(error_body_json)
        with self.assertRaises(
            ApiException.exception_class_for_http_status(502)
        ) as context:
            self.api.unassign_tests(

                agent_id=agent_id,

                agent_tests_assign_request=agent_tests_assign_request,

                aid=aid,

                _headers=self.te_headers("unassign_tests", error_status="502"),
            )
        assert_constructed_model_matches_example_json(context.exception.data, expected_error)






if __name__ == '__main__':
    unittest.main()
