from django.test import TestCase

from access_core.cms_integration.models import CMSApiRequest, CMSApiResponse


class CMSApiRequestTest(TestCase):
    def test_create_request(self):
        req = CMSApiRequest.objects.create(
            api_category="eligibility",
            endpoint="https://api.cms.gov/v1/eligibility",
            method="POST",
            request_payload={"mbi": "SYNTH123456"},
            request_headers={"Content-Type": "application/json"},
            status_code=200,
        )
        self.assertEqual(req.api_category, "eligibility")
        self.assertEqual(req.status_code, 200)


class CMSApiResponseTest(TestCase):
    def test_create_response(self):
        req = CMSApiRequest.objects.create(
            api_category="eligibility",
            endpoint="https://api.cms.gov/v1/eligibility",
            method="POST",
            request_payload={},
            request_headers={},
        )
        resp = CMSApiResponse.objects.create(
            request=req,
            response_payload={"eligible": True},
            response_headers={"Content-Type": "application/json"},
            status_code=200,
        )
        self.assertEqual(resp.status_code, 200)
        self.assertEqual(resp.response_payload["eligible"], True)