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_settings.settings_from_confattr(settings)
accessor = bg_accessor_factory.accessor_from_settings(settings)
self.assertNotEqual(accessor, None)
settings["BG_CASSANDRA_COMPRESSION"] = False
settings = bg_settings.settings_from_confattr(settings)
accessor = bg_accessor_factory.accessor_from_settings(settings)
self.assertNotEqual(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_settings.settings_from_confattr(settings)
accessor = bg_accessor_factory.accessor_from_settings(settings)
self.assertNotEqual(accessor, None)
settings["BG_CASSANDRA_COMPRESSION"] = False
settings = bg_settings.settings_from_confattr(settings)
accessor = bg_accessor_factory.accessor_from_settings(settings)
self.assertNotEqual(accessor, None)
def __init__(self, opts):
settings = bg_settings.settings_from_args(opts)
bg_utils.set_log_level(settings)
self._accessor = bg_accessor_factory.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 __init__(self, opts):
settings = bg_settings.settings_from_args(opts)
bg_utils.set_log_level(settings)
self._accessor = bg_accessor_factory.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 setUpClass(cls):
"""Create the test Accessor."""
# TODO (t.chataigner) Handle hybrid accessor here.
driver_name = cls.ACCESSOR_SETTINGS.get(
"driver", bg_accessor_factory.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()
)
if "hybrid" in driver_name:
if cls.ACCESSOR_SETTINGS['data_driver'] == 'cassandra':
cls.cassandra_helper = CassandraHelper()
cls.cassandra_helper.setUpClass()
cls.ACCESSOR_SETTINGS.update(
cls.cassandra_helper.get_accessor_settings()
)
# force read_on to be updated everytime during the tests
cls.ACCESSOR_SETTINGS['read_on_sampling_rate'] = 1.0
cls.accessor = bg_accessor_factory.accessor_from_settings(cls.ACCESSOR_SETTINGS)
cls.accessor.syncdb()
cls.accessor.connect()
def test_hybrid_accessor(self, cassandra_accessor_mock):
settings = {
"BG_DATA_DRIVER": "cassandra",
"BG_METADATA_DRIVER": "memory"
}
settings = bg_settings.settings_from_confattr(settings)
accessor = bg_accessor_factory.accessor_from_settings(settings)
self.assertNotEqual(accessor, None)
cassandra_accessor_mock.assert_called_with(enable_metadata=False)
def test_hybrid_accessor(self, cassandra_accessor_mock):
settings = {
"BG_DATA_DRIVER": "cassandra",
"BG_METADATA_DRIVER": "memory"
}
settings = bg_settings.settings_from_confattr(settings)
accessor = bg_accessor_factory.accessor_from_settings(settings)
self.assertNotEqual(accessor, None)
cassandra_accessor_mock.assert_called_with(enable_metadata=False)
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_settings.settings_from_confattr(settings)
bg_utils.set_log_level(settings)
return bg_accessor_factory.accessor_from_settings(settings)
def main(args=None, accessor=None):
"""Entry point for the module."""
if not args:
args = sys.argv[1:]
opts = commands.parse_opts(args)
if not getattr(opts, "func", None):
opts.func = command_shell.CommandShell().run
settings = bg_settings.settings_from_args(opts)
bg_utils.set_log_level(settings)
accessor = accessor or bg_accessor_factory.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_accessor_factory.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_accessor_factory.accessor_from_settings(cls.ACCESSOR_SETTINGS)
cls.accessor.syncdb()
cls.accessor.connect()
def test_memory_accessor(self):
settings = {"BG_DRIVER": "memory"}
settings = bg_settings.settings_from_confattr(settings)
accessor = bg_accessor_factory.accessor_from_settings(settings)
self.assertNotEqual(accessor, None)
def test_memory_accessor(self):
settings = {"BG_DRIVER": "memory"}
settings = bg_settings.settings_from_confattr(settings)
accessor = bg_accessor_factory.accessor_from_settings(settings)
self.assertNotEqual(accessor, None)
