def setUp(self):
self.db = "graflux_test"
self.config = {
"influxdb": {"host": "localhost", "port": 8086, "user": "******", "password": "******", "db": self.db},
"index": {"build_interval": 10, "load_interval": 5, "storage": "file", "path": "/tmp"},
}
self.client = InfluxDBClient(database=self.db)
self.metric_lookup = MetricLookup(self.config)
self.metric_lookup.delete_index()
self.metric_lookup.storage.release_update_lock()
self.clean_db()
class QueryEngineTest(unittest.TestCase):
def setUp(self):
self.db = "graflux_test"
self.config = {
"influxdb": {"host": "localhost", "port": 8086, "user": "******", "password": "******", "db": self.db},
"index": {"build_interval": 10, "load_interval": 5, "storage": "file", "path": "/tmp"},
}
self.client = InfluxDBClient(database=self.db)
self.metric_lookup = MetricLookup(self.config)
self.metric_lookup.delete_index()
self.metric_lookup.storage.release_update_lock()
self.clean_db()
def tearDown(self):
self.metric_lookup.stop_background_refresh()
def clean_db(self):
try:
self.client.drop_database(self.db)
except influxdb.exceptions.InfluxDBClientError:
pass
self.client.create_database(self.db)
def create_test_data(self, metrics):
data = [{"measurement": metric, "tags": {}, "fields": {"value": 1}} for metric in metrics]
self.assertTrue(self.client.write_points(data))
def test_background_update(self):
# awkwardly large number here but Influx has done a few odd things with pagination of
# SHOW SERIES in the past and we want to ensure we aren't being silently limited in the result set
metrics = ["test.series.test{0}".format(x) for x in range(0, 100000)]
self.create_test_data(metrics)
self.metric_lookup.start_background_refresh()
time.sleep(8) # allow background threads time to process
self.assertEqual(len(self.metric_lookup.query("test.series.*")), 100000)
