78a9b1f0ae
The read_cached_programs switch was used to control the release of changes for reading programs exclusively from the cache. With those changes stable, the switch is no longer necessary. LEARNER-382
55 lines
1.7 KiB
Python
55 lines
1.7 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(object):
|
|
"""
|
|
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 = '{base}.{uuid}'.format(base=key, uuid=uuid)
|
|
|
|
requests.put(
|
|
'{}/set_config'.format(CATALOG_STUB_URL),
|
|
data={program_key: json.dumps(program)},
|
|
)
|
|
|
|
# Stub list endpoint as if the uuids_only query param had been passed.
|
|
requests.put(
|
|
'{}/set_config'.format(CATALOG_STUB_URL),
|
|
data={key: json.dumps(uuids)},
|
|
)
|
|
|
|
|
|
class CatalogIntegrationMixin(object):
|
|
"""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': '{}/api/v1/'.format(CATALOG_STUB_URL),
|
|
'cache_ttl': 0,
|
|
'service_username': service_username,
|
|
}).install()
|