"""
Tools to create catalog-related data for use in bok choy tests.
"""


import json

import requests

from common.test.acceptance.fixtures import CATALOG_STUB_URL
from common.test.acceptance.fixtures.config import ConfigModelFixture


class CatalogFixture:
    """
    Interface to set up mock responses from the Catalog stub server.
    """
    def install_programs(self, programs):
        """
        Stub the discovery service's program list and detail API endpoints.

        Arguments:
            programs (list): A list of programs. Both list and detail endpoints
                will be stubbed using data from this list.
        """
        key = 'catalog.programs'

        uuids = []
        for program in programs:
            uuid = program['uuid']
            uuids.append(uuid)

            program_key = f'{key}.{uuid}'
            requests.put(
                f'{CATALOG_STUB_URL}/set_config',
                data={program_key: json.dumps(program)},
            )

        # Stub list endpoint as if the uuids_only query param had been passed.
        requests.put(
            f'{CATALOG_STUB_URL}/set_config',
            data={key: json.dumps(uuids)},
        )

    def install_pathways(self, pathways):
        """
        Stub the discovery service's credit pathways API endpoint

        Arguments:
             pathways (list): A list of credit pathways. List endpoint will be stubbed using data from this list.
        """
        requests.put(
            f'{CATALOG_STUB_URL}/set_config',
            data={'catalog.pathways': json.dumps({'results': pathways, 'next': None})}
        )

    def install_program_types(self, program_types):
        """
        Stub the discovery service's program type list API endpoints.

        Arguments:
            program_types (list): A list of program types. List endpoint will be stubbed using data from this list.
        """
        requests.put(
            f'{CATALOG_STUB_URL}/set_config',
            data={'catalog.programs_types': json.dumps(program_types)},
        )


class CatalogIntegrationMixin:
    """Mixin providing a method used to configure the catalog integration."""
    def set_catalog_integration(self, is_enabled=False, service_username=None):
        """Use this to change the catalog integration config model during tests."""
        ConfigModelFixture('/config/catalog', {
            'enabled': is_enabled,
            'internal_api_url': f'{CATALOG_STUB_URL}/api/v1/',
            'cache_ttl': 0,
            'service_username': service_username,
        }).install()