t/integration: Add a simple time-series fetcher test.

[sysdb.git] / t / integration / mock_plugin.c

diff --git a/t/integration/mock_plugin.c b/t/integration/mock_plugin.c
index 7e2b4a67dae3fc5e843a1eaf33cd307becc5762c..8041c0dfab9a4e1af2b5132d886528642ec5c27d 100644 (file)
--- a/t/integration/mock_plugin.c
+++ b/t/integration/mock_plugin.c
@@ -51,6 +51,29 @@ static const char *hostnames[] = {
        "localhost",
 };
 
+static struct {
+       const char *hostname;
+       const char *metric;
+       sdb_metric_store_t store;
+} metrics[] = {
+       { "some.host.name", "foo/bar/qux",
+               { "mock", "/var/lib/collectd/rrd/foo/bar/qux.rrd" } },
+       { "some.host.name", "foo/bar/baz",
+               { "mock", "/var/lib/collectd/rrd/foo/bar/baz.rrd" } },
+       { "some.host.name", "foo2/bar/qux",
+               { "mock", "/var/lib/collectd/rrd/foo2/bar/qux.rrd" } },
+       { "some.host.name", "foo2/bar/baz",
+               { "mock", "/var/lib/collectd/rrd/foo2/bar/baz.rrd" } },
+       { "other.host.name", "foo/bar/qux",
+               { "mock", "/var/lib/collectd/rrd/foo/bar/qux.rrd" } },
+       { "other.host.name", "foo/bar/baz",
+               { "mock", "/var/lib/collectd/rrd/foo/bar/baz.rrd" } },
+       { "other.host.name", "foo2/bar/qux",
+               { "mock", "/var/lib/collectd/rrd/foo2/bar/qux.rrd" } },
+       { "other.host.name", "foo2/bar/baz",
+               { "mock", "/var/lib/collectd/rrd/foo2/bar/baz.rrd" } },
+};
+
 static struct {
        const char *hostname;
        const char *service;
@@ -136,6 +159,15 @@ mock_collect(sdb_object_t *user_data)
                        exit(1);
                }
        }
+       for (i = 0; i < SDB_STATIC_ARRAY_LEN(metrics); ++i) {
+               if ((check = sdb_store_metric(metrics[i].hostname,
+                                               metrics[i].metric, &metrics[i].store,
+                                               sdb_gettime()))) {
+                       sdb_log(SDB_LOG_ERR, "mock::plugin: Failed to store metric: "
+                                       "status %d", check);
+                       exit(1);
+               }
+       }
        for (i = 0; i < SDB_STATIC_ARRAY_LEN(services); ++i) {
                if ((check = sdb_store_service(services[i].hostname,
                                                services[i].service, sdb_gettime()))) {
@@ -160,6 +192,42 @@ mock_collect(sdb_object_t *user_data)
        return 0;
 } /* mock_collect */
 
+static sdb_timeseries_t *
+mock_fetch_ts(const char *id, sdb_timeseries_opts_t *opts,
+               sdb_object_t *user_data)
+{
+       sdb_timeseries_t *ts;
+       const char *names[] = { "nameA", "nameB" };
+       size_t i, j;
+
+       if (*id != '/') {
+               sdb_log(SDB_LOG_ERR, "mock::plugin: Invalid time-series %s", id);
+               exit(1);
+       }
+
+       if (SDB_OBJ_WRAPPER(user_data)->data != MAGIC_DATA) {
+               sdb_log(SDB_LOG_ERR, "mock::plugin: Invalid user data %p "
+                               "passed to collect", SDB_OBJ_WRAPPER(user_data)->data);
+               exit(1);
+       }
+
+       ts = sdb_timeseries_create(SDB_STATIC_ARRAY_LEN(names), names, 10);
+       if (! ts)
+               return NULL;
+
+       ts->start = opts->start;
+       ts->end = opts->end;
+
+       for (i = 0; i < 10; ++i) {
+               for (j = 0; j < SDB_STATIC_ARRAY_LEN(names); ++j) {
+                       ts->data[j][i].timestamp = ts->start
+                               + i * (ts->end - ts->start) / 10;
+                       ts->data[j][i].value = (double)(i + j);
+               }
+       }
+       return ts;
+} /* mock_fetch_ts */
+
 static int
 mock_config(oconfig_item_t *ci)
 {
@@ -183,11 +251,13 @@ mock_config(oconfig_item_t *ci)
                exit(1);
        }
 
-       sdb_plugin_register_init("mock::init", mock_init, user_data);
-       sdb_plugin_register_shutdown("mock::shutdown", mock_shutdown, user_data);
-       sdb_plugin_register_collector("mock::collect", mock_collect,
+       sdb_plugin_register_init("main", mock_init, user_data);
+       sdb_plugin_register_shutdown("main", mock_shutdown, user_data);
+       sdb_plugin_register_collector("main", mock_collect,
                        /* interval = */ NULL, user_data);
 
+       sdb_plugin_register_ts_fetcher("mock", mock_fetch_ts, user_data);
+
        sdb_object_deref(user_data);
        return 0;
 } /* mock_config */
@@ -195,7 +265,6 @@ mock_config(oconfig_item_t *ci)
 int
 sdb_module_init(sdb_plugin_info_t *info)
 {
-       sdb_plugin_set_info(info, SDB_PLUGIN_INFO_NAME, "test::integration::mock");
        sdb_plugin_set_info(info, SDB_PLUGIN_INFO_DESC, "a mock plugin");
        sdb_plugin_set_info(info, SDB_PLUGIN_INFO_COPYRIGHT,
                        "Copyright (C) 2012 Sebastian 'tokkee' Harl <sh@tokkee.org>");