testsettings.py - master - githubmirror/commcare-hq

testsettings.py
143 строки4 KB
Amit Phulera
update ES_MULTIPLEX_TO_VERSION in testsettings so that tests run fine
21 май 2025, 15:26
21 май 2025, 15:262630215
import os
from warnings import filterwarnings

import settingshelper as helper

assert helper.is_testing(), 'test mode is required before importing settings'
from settings import *  # noqa: E402, F403

# Commenting out temporarily for tests
# if os.environ.get('ELASTICSEARCH_MAJOR_VERSION'):
#     ELASTICSEARCH_MAJOR_VERSION = int(os.environ.get('ELASTICSEARCH_MAJOR_VERSION'))

ELASTICSEARCH_MAJOR_VERSION = 6

# timeout faster in tests
ES_SEARCH_TIMEOUT = 5

# Default multiplexed value for_test adapter
ES_FOR_TEST_INDEX_MULTIPLEXED = False
ES_FOR_TEST_INDEX_SWAPPED = False


# Set multiplexed settings to false for test runs

ES_APPS_INDEX_MULTIPLEXED = False
ES_CASE_SEARCH_INDEX_MULTIPLEXED = False
ES_CASES_INDEX_MULTIPLEXED = False
ES_DOMAINS_INDEX_MULTIPLEXED = False
ES_FORMS_INDEX_MULTIPLEXED = False
ES_GROUPS_INDEX_MULTIPLEXED = False
ES_SMS_INDEX_MULTIPLEXED = False
ES_USERS_INDEX_MULTIPLEXED = False


ES_APPS_INDEX_SWAPPED = False
ES_CASE_SEARCH_INDEX_SWAPPED = False
ES_CASES_INDEX_SWAPPED = False
ES_DOMAINS_INDEX_SWAPPED = False
ES_FORMS_INDEX_SWAPPED = False
ES_GROUPS_INDEX_SWAPPED = False
ES_SMS_INDEX_SWAPPED = False
ES_USERS_INDEX_SWAPPED = False

# This should be updated when a new value is added to ES_REINDEX_LOG else test will fail
ES_MULTIPLEX_TO_VERSION = '7'
ES_SETTINGS = {
    'default': {
        'number_of_replicas': 0,
        'number_of_shards': 1,
    },
}

# note: the only reason these are prepended to INSTALLED_APPS is because of
# a weird travis issue with kafka. if for any reason this order causes problems
# it can be reverted whenever that's figured out.
# https://github.com/dimagi/commcare-hq/pull/10034#issuecomment-174868270
INSTALLED_APPS = (
    'testapps.test_elasticsearch',
    'testapps.test_pillowtop',
) + tuple(INSTALLED_APPS)  # noqa: F405

# these settings can be overridden with environment variables
for key, value in {
    'DD_DOGSTATSD_DISABLE': 'true',
    'DD_TRACE_ENABLED': 'false',
}.items():
    os.environ.setdefault(key, value)
del key, value

if "SKIP_TESTS_REQUIRING_EXTRA_SETUP" not in globals():
    SKIP_TESTS_REQUIRING_EXTRA_SETUP = False

CELERY_TASK_ALWAYS_EAGER = True
# keep a copy of the original PILLOWTOPS setting around in case other tests want it.
_PILLOWTOPS = PILLOWTOPS # noqa F405
PILLOWTOPS = {}

PHONE_TIMEZONES_HAVE_BEEN_PROCESSED = True
PHONE_TIMEZONES_SHOULD_BE_PROCESSED = True

ENABLE_PRELOGIN_SITE = True

# override dev_settings
CACHE_REPORTS = True

# Hide couchdb 'unclosed socket' warnings
filterwarnings("ignore", r"unclosed.*socket.*raddr=\([^) ]* 5984\)", ResourceWarning)


def _set_logging_levels(levels):
    import logging
    for path, level in levels.items():
        logging.getLogger(path).setLevel(level)


_set_logging_levels({
    # Quiet down noisy loggers. Selective removal can be handy for debugging.
    'alembic': 'WARNING',
    'corehq.apps.auditcare': 'INFO',
    'boto3': 'WARNING',
    'botocore': 'INFO',
    'couchdbkit.request': 'INFO',
    'couchdbkit.designer': 'WARNING',
    'datadog': 'WARNING',
    'elasticsearch': 'ERROR',
    'kafka.conn': 'WARNING',
    'kafka.client': 'WARNING',
    'kafka.consumer.kafka': 'WARNING',
    'kafka.metrics': 'WARNING',
    'kafka.protocol.parser': 'WARNING',
    'kafka.producer': 'WARNING',
    'quickcache': 'INFO',
    'requests.packages.urllib3': 'WARNING',
    's3transfer': 'INFO',
    'urllib3': 'WARNING',
})

# TODO empty logging config (and fix revealed deprecation warnings)
LOGGING = {
    'disable_existing_loggers': False,
    'version': 1,
    'loggers': {},
}

helper.assign_test_db_names(DATABASES)  # noqa: F405
helper.update_redis_location_for_tests(CACHES)  # noqa: F405

# See comment under settings.SMS_QUEUE_ENABLED
SMS_QUEUE_ENABLED = False

# use all providers in tests
METRICS_PROVIDERS = [
    'corehq.util.metrics.datadog.DatadogMetrics',
    'corehq.util.metrics.prometheus.PrometheusMetrics',
]

FORMPLAYER_INTERNAL_AUTH_KEY = "abc123"

# A workaround to test the messaging framework. See: https://stackoverflow.com/a/60218100
MESSAGE_STORAGE = 'django.contrib.messages.storage.cookie.CookieStorage'

if os.environ.get("STRIPE_PRIVATE_KEY"):
    STRIPE_PRIVATE_KEY = os.environ.get("STRIPE_PRIVATE_KEY")