def test_cassandra_accessor(self):
settings = {"BG_DRIVER": "cassandra"}
settings["BG_CASSANDRA_CONTACT_POINTS"] = "localhost"
settings["BG_CASSANDRA_COMPRESSION"] = True
settings["BG_CASSANDRA_TIMEOUT"] = 5
settings = bg_utils.settings_from_confattr(settings)
accessor = bg_utils.accessor_from_settings(settings)
self.assertNotEquals(accessor, None)
settings["BG_CASSANDRA_COMPRESSION"] = False
settings = bg_utils.settings_from_confattr(settings)
accessor = bg_utils.accessor_from_settings(settings)
self.assertNotEquals(accessor, None)
def test_cassandra_accessor(self):
settings = {"BG_DRIVER": "cassandra"}
settings["BG_CASSANDRA_CONTACT_POINTS"] = "localhost"
settings["BG_CASSANDRA_COMPRESSION"] = True
settings["BG_CASSANDRA_TIMEOUT"] = 5
settings = bg_utils.settings_from_confattr(settings)
accessor = bg_utils.accessor_from_settings(settings)
self.assertNotEquals(accessor, None)
settings["BG_CASSANDRA_COMPRESSION"] = False
settings = bg_utils.settings_from_confattr(settings)
accessor = bg_utils.accessor_from_settings(settings)
self.assertNotEquals(accessor, None)
def __init__(self, opts):
settings = bg_utils.settings_from_args(opts)
bg_utils.set_log_level(settings)
self._accessor = bg_utils.accessor_from_settings(settings)
self._opts = opts
self.time_start = time.mktime(self._opts.time_start.timetuple())
self.time_end = time.mktime(self._opts.time_end.timetuple())
def main(args=None):
"""Entry point for the module."""
if not args:
args = sys.argv[1:]
opts = _parse_opts(args)
settings = bg_utils.settings_from_args(opts)
bg_utils.set_log_level(settings)
accessor = bg_utils.accessor_from_settings(settings)
opts.func(accessor, opts)
def main(args=None, accessor=None):
"""Entry point for the module."""
if not args:
args = sys.argv[1:]
opts = _parse_opts(args)
settings = bg_utils.settings_from_args(opts)
bg_utils.set_log_level(settings)
accessor = accessor or bg_utils.accessor_from_settings(settings)
opts.func(accessor, opts)
accessor.flush()
def accessor_from_settings(settings):
"""Get Accessor from Graphite-related configuration object.
Args:
settings: either carbon_conf.Settings or a Django-like settings object.
Returns:
Accessor (not connected).
"""
settings = bg_utils.settings_from_confattr(settings)
bg_utils.set_log_level(settings)
return bg_utils.accessor_from_settings(settings)
def accessor_from_settings(settings):
"""Get Accessor from configuration.
Args:
settings: either carbon_conf.Settings or a Django-like settings object
Returns:
Accessor (not connected)
"""
settings = bg_utils.settings_from_confattr(settings)
bg_utils.set_log_level(settings)
return bg_utils.accessor_from_settings(settings)
def main(args=None, accessor=None):
"""Entry point for the module."""
if not args:
args = sys.argv[1:]
opts = _parse_opts(args)
if not getattr(opts, "func", None):
opts.func = command_shell.CommandShell().run
settings = bg_utils.settings_from_args(opts)
bg_utils.set_log_level(settings)
accessor = accessor or bg_utils.accessor_from_settings(settings)
opts.func(accessor, opts)
accessor.flush()
accessor.shutdown()
def setUpClass(cls):
"""Create the test Accessor."""
# TODO (t.chataigner) Handle hybrid accessor here.
driver_name = cls.ACCESSOR_SETTINGS.get('driver',
bg_utils.DEFAULT_DRIVER)
if "cassandra" in driver_name:
cls.cassandra_helper = CassandraHelper()
cls.cassandra_helper.setUpClass()
cls.ACCESSOR_SETTINGS.update(
cls.cassandra_helper.get_accessor_settings())
if "elasticsearch" in driver_name:
cls.elasticsearch_helper = ElasticsearchHelper()
cls.elasticsearch_helper.setUpClass()
cls.ACCESSOR_SETTINGS.update(
cls.elasticsearch_helper.get_accessor_settings())
cls.accessor = bg_utils.accessor_from_settings(cls.ACCESSOR_SETTINGS)
cls.accessor.syncdb()
cls.accessor.connect()
def test_memory_accessor(self):
settings = {"BG_DRIVER": "memory"}
settings = bg_utils.settings_from_confattr(settings)
accessor = bg_utils.accessor_from_settings(settings)
self.assertNotEquals(accessor, None)
def test_memory_accessor(self):
settings = {"BG_DRIVER": "memory"}
settings = bg_utils.settings_from_confattr(settings)
accessor = bg_utils.accessor_from_settings(settings)
self.assertNotEquals(accessor, None)