示例#1
0
#!/usr/bin/env python

import gevent.monkey; gevent.monkey.patch_all()

import functools
import logging
import os
import subprocess
import sys
import unittest

from argparse import ArgumentParser

from kronos.conf import settings
from tests.conf import default_settings
settings.update(default_settings)

# Disable log handlers.
import kronos.app  # noqa
logging.root.handlers[:] = []
logging.root.addHandler(logging.NullHandler())


# Mapping from test name => (<pathname of directory with test file>,
# <backends to run test against>)
TESTS = {
  'cassandra': ('tests/storage/cassandra', ('cassandra', )),
  'common': ('tests/common', ('memory',
                              'cassandra',
                              'elasticsearch',)),
  'conf': ('tests/conf', ('memory', )),
示例#2
0
#!/usr/bin/python

import gevent.monkey; gevent.monkey.patch_all()

import functools
import logging
import os
import subprocess
import sys
import unittest

from argparse import ArgumentParser

from kronos.conf import settings
from tests.conf import default_settings
settings.update(default_settings)

# Disable log handlers.
import kronos.app  # noqa
logging.root.handlers[:] = []
logging.root.addHandler(logging.NullHandler())


# Mapping from test name => (<pathname of directory with test file>,
# <backends to run test against>)
TESTS = {
  'cassandra': ('tests/storage/cassandra', ('cassandra', )),
  'common': ('tests/common', ('memory',
                              'cassandra',
                              'elasticsearch',)),
  'conf': ('tests/conf', ('memory', )),
示例#3
0
  parser.add_argument('--port', action='store', default='8150',
                      help='port to listen for incoming requests')
  parser.add_argument('--serving-mode',
                      choices=[ServingMode.ALL, ServingMode.COLLECTOR,
                               ServingMode.READONLY],
                      help='which serving mode to run in')
  parser.add_argument('--config', action='store',
                      help='path of config file to use')
  parser.add_argument('--profile', action='store_true',
                      help='Profile each request using cProfile?')
  args = parser.parse_args()

  settings.clear()
  if args.config:
    # If a config file path is given, import that as the `settings` module.  
    settings.update(imp.load_source('kronos.conf.run_settings', args.config))
  else:
    # Otherwise use default settings. This is to ensure we never try to read
    # the settings for the configured kronos service when using this runner
    # script.
    from kronos.conf import default_settings
    settings.update(default_settings)

  # Override the `debug` in the settings module and `debug` for
  # `args`.
  settings.debug = args.debug or settings.debug
  settings.serving_mode = args.serving_mode or settings.serving_mode
  settings.profile = args.profile or settings.profile

  # Only load the application after we've overwritten settings.serving_mode, or
  # else the endpoint access control logic will kick in too early.
示例#4
0
  parser.add_argument('--port', action='store', default='8150',
                      help='port to listen for incoming requests')
  parser.add_argument('--serving-mode',
                      choices=[ServingMode.ALL, ServingMode.COLLECTOR,
                               ServingMode.READONLY],
                      help='which serving mode to run in')
  parser.add_argument('--config', action='store',
                      help='path of config file to use')
  parser.add_argument('--profile', action='store_true',
                      help='Profile each request using cProfile?')
  args = parser.parse_args()

  settings.clear()
  if args.config:
    # If a config file path is given, import that as the `settings` module.
    settings.update(imp.load_source('kronos.conf.run_settings', args.config))
  else:
    # Otherwise use default settings. This is to ensure we never try to read
    # the settings for the configured kronos service when using this runner
    # script.
    from kronos.conf import default_settings
    settings.update(default_settings)

  # Override the `debug` in the settings module and `debug` for
  # `args`.
  settings.debug = args.debug or settings.debug
  settings.serving_mode = args.serving_mode or settings.serving_mode
  settings.profile = args.profile or settings.profile

  # Only load the application after we've overwritten settings.serving_mode, or
  # else the endpoint access control logic will kick in too early.