80 lines
2.5 KiB
Python
80 lines
2.5 KiB
Python
"""
|
|
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()
|