"""
Tools for creating certificates config fixture data.
"""


import json

from common.test.acceptance.fixtures import STUDIO_BASE_URL
from common.test.acceptance.fixtures.base import StudioApiFixture


class CertificateConfigFixtureError(Exception):
    """
    Error occurred while installing certificate config fixture.
    """
    pass  # lint-amnesty, pylint: disable=unnecessary-pass


class CertificateConfigUpdateFixtureError(Exception):
    """
    Error occurred while updating certificate config fixture.
    """
    pass  # lint-amnesty, pylint: disable=unnecessary-pass


class CertificateConfigFixture(StudioApiFixture):
    """
    Fixture to create certificates configuration for a course
    """
    certificates = []

    def __init__(self, course_id, certificates_data):
        self.course_id = course_id
        self.certificates = certificates_data
        super().__init__()

    def install(self):
        """
        Push the certificates config data to certificate endpoint.
        """
        response = self.session.post(
            f'{STUDIO_BASE_URL}/certificates/{self.course_id}',
            data=json.dumps(self.certificates),
            headers=self.headers
        )

        if not response.ok:
            raise CertificateConfigFixtureError(
                "Could not create certificate {}.  Status was {}".format(
                    json.dumps(self.certificates), response.status_code
                )
            )

        return self

    def update_certificate(self, certificate_id):
        """
        Update the certificates config data to certificate endpoint.
        """
        response = self.session.put(
            f'{STUDIO_BASE_URL}/certificates/{self.course_id}/{certificate_id}',
            data=json.dumps(self.certificates),
            headers=self.headers
        )

        if not response.ok:
            raise CertificateConfigUpdateFixtureError(
                "Could not update certificate {}.  Status was {}".format(
                    json.dumps(self.certificates), response.status_code
                )
            )

        return self