t/unit/frontend/query_test.c

   1 /*
   2  * SysDB - t/unit/frontend/query_test.c
   3  * Copyright (C) 2015 Sebastian 'tokkee' Harl <sh@tokkee.org>
   4  * All rights reserved.
   5  *
   6  * Redistribution and use in source and binary forms, with or without
   7  * modification, are permitted provided that the following conditions
   8  * are met:
   9  * 1. Redistributions of source code must retain the above copyright
  10  *    notice, this list of conditions and the following disclaimer.
  11  * 2. Redistributions in binary form must reproduce the above copyright
  12  *    notice, this list of conditions and the following disclaimer in the
  13  *    documentation and/or other materials provided with the distribution.
  14  *
  15  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  16  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
  17  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  18  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDERS OR
  19  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  20  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  21  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
  22  * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
  23  * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
  24  * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
  25  * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  26  */
  27
  28 #if HAVE_CONFIG_H
  29 #       include "config.h"
  30 #endif
  31
  32 #include "core/plugin.h"
  33 #include "frontend/connection.h"
  34 #include "frontend/connection-private.h"
  35 #include "testutils.h"
  36
  37 #include <check.h>
  38
  39 /*
  40  * private helpers
  41  */
  42
  43 static void
  44 populate(void)
  45 {
  46         sdb_store_t *store;
  47         sdb_data_t datum;
  48
  49         /* the frontend accesses the store via the plugin API */
  50         store = sdb_store_create();
  51         ck_assert(store != NULL);
  52         ck_assert(sdb_plugin_register_writer("test-writer",
  53                                 &sdb_store_writer, SDB_OBJ(store)) == 0);
  54         ck_assert(sdb_plugin_register_reader("test-reader",
  55                                 &sdb_store_reader, SDB_OBJ(store)) == 0);
  56         sdb_object_deref(SDB_OBJ(store));
  57
  58         /* populate the store */
  59         sdb_plugin_store_host("h1", 1 * SDB_INTERVAL_SECOND);
  60         sdb_plugin_store_host("h2", 3 * SDB_INTERVAL_SECOND);
  61
  62         datum.type = SDB_TYPE_STRING;
  63         datum.data.string = "v1";
  64         sdb_plugin_store_attribute("h1", "k1", &datum, 1 * SDB_INTERVAL_SECOND);
  65         datum.data.string = "v2";
  66         sdb_plugin_store_attribute("h1", "k2", &datum, 2 * SDB_INTERVAL_SECOND);
  67         datum.data.string = "v3";
  68         sdb_plugin_store_attribute("h1", "k3", &datum, 2 * SDB_INTERVAL_SECOND);
  69
  70         sdb_plugin_store_metric("h1", "m1", /* store */ NULL, 2 * SDB_INTERVAL_SECOND);
  71         sdb_plugin_store_metric("h1", "m2", /* store */ NULL, 1 * SDB_INTERVAL_SECOND);
  72         sdb_plugin_store_metric("h2", "m1", /* store */ NULL, 1 * SDB_INTERVAL_SECOND);
  73
  74         datum.type = SDB_TYPE_INTEGER;
  75         datum.data.integer = 42;
  76         sdb_plugin_store_metric_attribute("h1", "m1", "k3",
  77                         &datum, 2 * SDB_INTERVAL_SECOND);
  78
  79         sdb_plugin_store_service("h2", "s1", 1 * SDB_INTERVAL_SECOND);
  80         sdb_plugin_store_service("h2", "s2", 2 * SDB_INTERVAL_SECOND);
  81
  82         datum.data.integer = 123;
  83         sdb_plugin_store_service_attribute("h2", "s2", "k1",
  84                         &datum, 2 * SDB_INTERVAL_SECOND);
  85         datum.data.integer = 4711;
  86         sdb_plugin_store_service_attribute("h2", "s2", "k2",
  87                         &datum, 1 * SDB_INTERVAL_SECOND);
  88 } /* populate */
  89
  90 static void
  91 turndown(void)
  92 {
  93         sdb_plugin_unregister_all();
  94 } /* turndown */
  95
  96 #define HOST_H1 \
  97         "{\"name\": \"h1\", \"last_update\": \"1970-01-01 00:00:01 +0000\", " \
  98                         "\"update_interval\": \"0s\", \"backends\": [], " \
  99                 "\"attributes\": [" \
 100                         "{\"name\": \"k1\", \"value\": \"v1\", " \
 101                                 "\"last_update\": \"1970-01-01 00:00:01 +0000\", " \
 102                                 "\"update_interval\": \"0s\", \"backends\": []}," \
 103                         "{\"name\": \"k2\", \"value\": \"v2\", " \
 104                                 "\"last_update\": \"1970-01-01 00:00:02 +0000\", " \
 105                                 "\"update_interval\": \"0s\", \"backends\": []}," \
 106                         "{\"name\": \"k3\", \"value\": \"v3\", " \
 107                                 "\"last_update\": \"1970-01-01 00:00:02 +0000\", " \
 108                                 "\"update_interval\": \"0s\", \"backends\": []}], " \
 109                 "\"metrics\": [" \
 110                         "{\"name\": \"m1\", \"timeseries\": false, " \
 111                                 "\"last_update\": \"1970-01-01 00:00:02 +0000\", " \
 112                                 "\"update_interval\": \"0s\", \"backends\": [], " \
 113                                 "\"attributes\": [" \
 114                                         "{\"name\": \"hostname\", \"value\": \"h1\", " \
 115                                                 "\"last_update\": \"1970-01-01 00:00:02 +0000\", " \
 116                                                 "\"update_interval\": \"0s\", \"backends\": []}," \
 117                                         "{\"name\": \"k3\", \"value\": 42, " \
 118                                                 "\"last_update\": \"1970-01-01 00:00:02 +0000\", " \
 119                                                 "\"update_interval\": \"0s\", \"backends\": []}]}," \
 120                         "{\"name\": \"m2\", \"timeseries\": false, " \
 121                                 "\"last_update\": \"1970-01-01 00:00:01 +0000\", " \
 122                                 "\"update_interval\": \"0s\", \"backends\": [], " \
 123                                 "\"attributes\": [" \
 124                                         "{\"name\": \"hostname\", \"value\": \"h1\", " \
 125                                                 "\"last_update\": \"1970-01-01 00:00:01 +0000\", " \
 126                                                 "\"update_interval\": \"0s\", \"backends\": []}]}]}"
 127 #define HOST_H2 \
 128         "{\"name\": \"h2\", \"last_update\": \"1970-01-01 00:00:03 +0000\", " \
 129                         "\"update_interval\": \"0s\", \"backends\": [], " \
 130                 "\"metrics\": [" \
 131                         "{\"name\": \"m1\", \"timeseries\": false, " \
 132                                 "\"last_update\": \"1970-01-01 00:00:01 +0000\", " \
 133                                 "\"update_interval\": \"0s\", \"backends\": [], " \
 134                                 "\"attributes\": [" \
 135                                         "{\"name\": \"hostname\", \"value\": \"h2\", " \
 136                                                 "\"last_update\": \"1970-01-01 00:00:01 +0000\", " \
 137                                                 "\"update_interval\": \"0s\", \"backends\": []}]}], " \
 138                 "\"services\": [" \
 139                         "{\"name\": \"s1\", \"last_update\": \"1970-01-01 00:00:01 +0000\", " \
 140                                 "\"update_interval\": \"0s\", \"backends\": [], " \
 141                                 "\"attributes\": [" \
 142                                         "{\"name\": \"hostname\", \"value\": \"h2\", " \
 143                                                 "\"last_update\": \"1970-01-01 00:00:01 +0000\", " \
 144                                                 "\"update_interval\": \"0s\", \"backends\": []}]}," \
 145                         "{\"name\": \"s2\", \"last_update\": \"1970-01-01 00:00:02 +0000\", " \
 146                                 "\"update_interval\": \"0s\", \"backends\": [], " \
 147                                 "\"attributes\": [" \
 148                                         "{\"name\": \"hostname\", \"value\": \"h2\", " \
 149                                                 "\"last_update\": \"1970-01-01 00:00:02 +0000\", " \
 150                                                 "\"update_interval\": \"0s\", \"backends\": []}," \
 151                                         "{\"name\": \"k1\", \"value\": 123, " \
 152                                                 "\"last_update\": \"1970-01-01 00:00:02 +0000\", " \
 153                                                 "\"update_interval\": \"0s\", \"backends\": []}," \
 154                                         "{\"name\": \"k2\", \"value\": 4711, " \
 155                                                 "\"last_update\": \"1970-01-01 00:00:01 +0000\", " \
 156                                                 "\"update_interval\": \"0s\", \"backends\": []}]}]}"
 157 #define HOST_H1_ARRAY "["HOST_H1"]"
 158 #define HOST_H12_ARRAY "["HOST_H1","HOST_H2"]"
 159 #define HOST_H1_LISTING \
 160         "{\"name\": \"h1\", \"last_update\": \"1970-01-01 00:00:01 +0000\", " \
 161                         "\"update_interval\": \"0s\", \"backends\": []}"
 162 #define HOST_H2_LISTING \
 163         "{\"name\": \"h2\", \"last_update\": \"1970-01-01 00:00:03 +0000\", " \
 164                         "\"update_interval\": \"0s\", \"backends\": []}"
 165
 166 #define SERVICE_H2_S1 \
 167         "{\"name\": \"h2\", \"last_update\": \"1970-01-01 00:00:03 +0000\", " \
 168                         "\"update_interval\": \"0s\", \"backends\": [], " \
 169                 "\"services\": [" \
 170                         "{\"name\": \"s1\", \"last_update\": \"1970-01-01 00:00:01 +0000\", " \
 171                                 "\"update_interval\": \"0s\", \"backends\": [], " \
 172                                 "\"attributes\": [" \
 173                                         "{\"name\": \"hostname\", \"value\": \"h2\", " \
 174                                                 "\"last_update\": \"1970-01-01 00:00:01 +0000\", " \
 175                                                 "\"update_interval\": \"0s\", \"backends\": []}]}]}"
 176 #define SERVICE_H2_S1_ARRAY "["SERVICE_H2_S1"]"
 177 #define SERVICE_H2_S12_LISTING \
 178         "[{\"name\": \"h2\", \"last_update\": \"1970-01-01 00:00:03 +0000\", " \
 179                         "\"update_interval\": \"0s\", \"backends\": [], " \
 180                 "\"services\": [" \
 181                         "{\"name\": \"s1\", \"last_update\": \"1970-01-01 00:00:01 +0000\", " \
 182                                 "\"update_interval\": \"0s\", \"backends\": []}," \
 183                         "{\"name\": \"s2\", \"last_update\": \"1970-01-01 00:00:02 +0000\", " \
 184                                 "\"update_interval\": \"0s\", \"backends\": []}]}]"
 185
 186 #define METRIC_H1_M1 \
 187         "{\"name\": \"h1\", \"last_update\": \"1970-01-01 00:00:01 +0000\", " \
 188                         "\"update_interval\": \"0s\", \"backends\": [], " \
 189                 "\"metrics\": [" \
 190                         "{\"name\": \"m1\", \"timeseries\": false, " \
 191                                 "\"last_update\": \"1970-01-01 00:00:02 +0000\", " \
 192                                 "\"update_interval\": \"0s\", \"backends\": [], " \
 193                                 "\"attributes\": [" \
 194                                         "{\"name\": \"hostname\", \"value\": \"h1\", " \
 195                                                 "\"last_update\": \"1970-01-01 00:00:02 +0000\", " \
 196                                                 "\"update_interval\": \"0s\", \"backends\": []}," \
 197                                         "{\"name\": \"k3\", \"value\": 42, " \
 198                                                 "\"last_update\": \"1970-01-01 00:00:02 +0000\", " \
 199                                                 "\"update_interval\": \"0s\", \"backends\": []}]}]}"
 200 #define METRIC_H12_M1_ARRAY \
 201         "["METRIC_H1_M1"," \
 202         "{\"name\": \"h2\", \"last_update\": \"1970-01-01 00:00:03 +0000\", " \
 203                         "\"update_interval\": \"0s\", \"backends\": [], " \
 204                 "\"metrics\": [" \
 205                         "{\"name\": \"m1\", \"timeseries\": false, " \
 206                                 "\"last_update\": \"1970-01-01 00:00:01 +0000\", " \
 207                                 "\"update_interval\": \"0s\", \"backends\": [], " \
 208                                 "\"attributes\": [" \
 209                                         "{\"name\": \"hostname\", \"value\": \"h2\", " \
 210                                                 "\"last_update\": \"1970-01-01 00:00:01 +0000\", " \
 211                                                 "\"update_interval\": \"0s\", \"backends\": []}]}]}]"
 212 #define METRIC_H12_M12_LISTING \
 213         "[{\"name\": \"h1\", \"last_update\": \"1970-01-01 00:00:01 +0000\", " \
 214                         "\"update_interval\": \"0s\", \"backends\": [], " \
 215                 "\"metrics\": [" \
 216                         "{\"name\": \"m1\", \"timeseries\": false, " \
 217                                 "\"last_update\": \"1970-01-01 00:00:02 +0000\", " \
 218                                 "\"update_interval\": \"0s\", \"backends\": []}," \
 219                         "{\"name\": \"m2\", \"timeseries\": false, " \
 220                                 "\"last_update\": \"1970-01-01 00:00:01 +0000\", " \
 221                                 "\"update_interval\": \"0s\", \"backends\": []}]}," \
 222         "{\"name\": \"h2\", \"last_update\": \"1970-01-01 00:00:03 +0000\", " \
 223                         "\"update_interval\": \"0s\", \"backends\": [], " \
 224                 "\"metrics\": [" \
 225                         "{\"name\": \"m1\", \"timeseries\": false, " \
 226                                 "\"last_update\": \"1970-01-01 00:00:01 +0000\", " \
 227                                 "\"update_interval\": \"0s\", \"backends\": []}]}]"
 228
 229 typedef struct {
 230         sdb_conn_t conn;
 231         sdb_strbuf_t *write_buf;
 232 } mock_conn_t;
 233 #define MOCK_CONN(obj) ((mock_conn_t *)(obj))
 234 #define CONN(obj) ((sdb_conn_t *)(obj))
 235
 236 static void
 237 mock_conn_destroy(sdb_conn_t *conn)
 238 {
 239         sdb_strbuf_destroy(conn->buf);
 240         sdb_strbuf_destroy(conn->errbuf);
 241         sdb_strbuf_destroy(MOCK_CONN(conn)->write_buf);
 242         free(conn);
 243 } /* mock_conn_destroy */
 244
 245 static ssize_t
 246 mock_conn_read(sdb_conn_t *conn, size_t len)
 247 {
 248         if (! conn)
 249                 return -1;
 250         /* unused so far */
 251         return len;
 252 } /* conn_read */
 253
 254 static ssize_t
 255 mock_conn_write(sdb_conn_t *conn, const void *buf, size_t len)
 256 {
 257         if (! conn)
 258                 return -1;
 259         return sdb_strbuf_memappend(MOCK_CONN(conn)->write_buf, buf, len);
 260 } /* conn_write */
 261
 262 static sdb_conn_t *
 263 mock_conn_create(void)
 264 {
 265         mock_conn_t *conn;
 266
 267         conn = calloc(1, sizeof(*conn));
 268         if (! conn) {
 269                 fail("INTERNAL ERROR: failed to allocate connection object");
 270                 return NULL;
 271         }
 272
 273         SDB_OBJ(conn)->name = "mock_connection";
 274         SDB_OBJ(conn)->ref_cnt = 1;
 275
 276         conn->conn.buf = sdb_strbuf_create(0);
 277         conn->conn.errbuf = sdb_strbuf_create(0);
 278         conn->write_buf = sdb_strbuf_create(64);
 279         if ((! conn->conn.buf) || (! conn->conn.errbuf) || (! conn->write_buf)) {
 280                 mock_conn_destroy(CONN(conn));
 281                 fail("INTERNAL ERROR: failed to allocate connection object");
 282                 return NULL;
 283         }
 284
 285         conn->conn.read = mock_conn_read;
 286         conn->conn.write = mock_conn_write;
 287
 288         conn->conn.username = "mock_user";
 289         conn->conn.cmd = SDB_CONNECTION_IDLE;
 290         conn->conn.cmd_len = 0;
 291         return CONN(conn);
 292 } /* mock_conn_create */
 293
 294 /* TODO: move this into a test helper module */
 295 static void
 296 fail_if_strneq(const char *got, const char *expected, size_t n, const char *fmt, ...)
 297 {
 298         sdb_strbuf_t *buf;
 299         va_list ap;
 300
 301         size_t len1 = strlen(got);
 302         size_t len2 = strlen(expected);
 303
 304         size_t i;
 305         int pos = -1;
 306
 307         if (n) {
 308                 len1 = SDB_MIN(len1, n);
 309                 len2 = SDB_MIN(len2, n);
 310         }
 311
 312         if (len1 != len2)
 313                 pos = (int)SDB_MIN(len1, len2);
 314
 315         for (i = 0; i < SDB_MIN(len1, len2); ++i) {
 316                 if (got[i] != expected[i]) {
 317                         pos = (int)i;
 318                         break;
 319                 }
 320         }
 321
 322         if (pos == -1)
 323                 return;
 324
 325         buf = sdb_strbuf_create(64);
 326         va_start(ap, fmt);
 327         sdb_strbuf_vsprintf(buf, fmt, ap);
 328
 329         fail("%s\n         got: %s\n              %*s\n    expected: %s",
 330                         sdb_strbuf_string(buf), got, pos + 1, "^", expected);
 331 } /* fail_if_strneq */
 332
 333 /*
 334  * tests
 335  */
 336
 337 #define VALUE "\0\0\0\4""v1"
 338 #define VALUE_LEN 7
 339
 340 static struct {
 341         uint32_t cmd;
 342         const char *query;
 343         int query_len;
 344         int expected;
 345         uint32_t code;
 346         size_t len;
 347         uint32_t type;
 348         const char *data;
 349 } query_data[] = {
 350         /* hosts */
 351         {
 352                 SDB_CONNECTION_QUERY, "LIST hosts", -1,
 353                 0, SDB_CONNECTION_DATA, 205, SDB_CONNECTION_LIST,
 354                 "["HOST_H1_LISTING","HOST_H2_LISTING"]",
 355         },
 356         {
 357                 SDB_CONNECTION_LIST, "\0\0\0\1", 4,
 358                 0, SDB_CONNECTION_DATA, 205, SDB_CONNECTION_LIST,
 359                 "["HOST_H1_LISTING","HOST_H2_LISTING"]",
 360         },
 361         {
 362                 SDB_CONNECTION_LIST, "", 0, /* LIST defaults to hosts */
 363                 0, SDB_CONNECTION_DATA, 205, SDB_CONNECTION_LIST,
 364                 "["HOST_H1_LISTING","HOST_H2_LISTING"]",
 365         },
 366         {
 367                 SDB_CONNECTION_QUERY, "LIST hosts; LIST hosts", -1, /* ignore second (and later) commands */
 368                 0, SDB_CONNECTION_DATA, 205, SDB_CONNECTION_LIST,
 369                 "["HOST_H1_LISTING","HOST_H2_LISTING"]",
 370         },
 371         {
 372                 SDB_CONNECTION_QUERY, "LIST hosts FILTER name = 'h1'", -1,
 373                 0, SDB_CONNECTION_DATA, 105, SDB_CONNECTION_LIST, "["HOST_H1_LISTING"]",
 374         },
 375         {
 376                 SDB_CONNECTION_QUERY, "LIST hosts FILTER name = 's1'", -1,
 377                 0, SDB_CONNECTION_DATA, 6, SDB_CONNECTION_LIST, "[]",
 378         },
 379         /* SDB_CONNECTION_LIST doesn't support filters yet */
 380         {
 381                 SDB_CONNECTION_QUERY, "FETCH host 'h1'", -1,
 382                 0, SDB_CONNECTION_DATA, 1110, SDB_CONNECTION_FETCH, HOST_H1,
 383         },
 384         {
 385                 SDB_CONNECTION_FETCH, "\0\0\0\1""h1", 7,
 386                 0, SDB_CONNECTION_DATA, 1110, SDB_CONNECTION_FETCH, HOST_H1,
 387         },
 388         {
 389                 SDB_CONNECTION_QUERY, "LOOKUP hosts MATCHING name = 'h1'", -1,
 390                 0, SDB_CONNECTION_DATA, 1112, SDB_CONNECTION_LOOKUP, HOST_H1_ARRAY,
 391         },
 392         {
 393                 SDB_CONNECTION_LOOKUP, "\0\0\0\1""name = 'h1'", 16,
 394                 0, SDB_CONNECTION_DATA, 1112, SDB_CONNECTION_LOOKUP, HOST_H1_ARRAY,
 395         },
 396         {
 397                 SDB_CONNECTION_QUERY, "FETCH host 'h1' FILTER age >= 0s", -1, /* always matches */
 398                 0, SDB_CONNECTION_DATA, 1110, SDB_CONNECTION_FETCH, HOST_H1,
 399         },
 400         /* SDB_CONNECTION_FETCH doesn't support filters yet */
 401         {
 402                 SDB_CONNECTION_QUERY, "LOOKUP hosts MATCHING name = 'h1' FILTER age >= 0s", -1, /* always matches */
 403                 0, SDB_CONNECTION_DATA, 1112, SDB_CONNECTION_LOOKUP, HOST_H1_ARRAY,
 404         },
 405         {
 406                 SDB_CONNECTION_QUERY, "LOOKUP hosts MATCHING ANY backend = 'b'", -1,
 407                 0, SDB_CONNECTION_DATA, 6, SDB_CONNECTION_LOOKUP, "[]",
 408         },
 409         {
 410                 SDB_CONNECTION_QUERY, "LOOKUP hosts MATCHING ANY backend || 'b' = 'b'", -1,
 411                 0, SDB_CONNECTION_DATA, 2205, SDB_CONNECTION_LOOKUP, HOST_H12_ARRAY,
 412         },
 413         {
 414                 SDB_CONNECTION_QUERY, "FETCH host 'h1' FILTER age < 0s", -1, /* never matches */
 415                 -1, UINT32_MAX, 0, 0, NULL, /* FETCH fails if the object doesn't exist */
 416         },
 417         /* SDB_CONNECTION_FETCH doesn't support filters yet */
 418         {
 419                 SDB_CONNECTION_QUERY, "LOOKUP hosts MATCHING name = 'h1' FILTER age < 0s", -1, /* never matches */
 420                 0, SDB_CONNECTION_DATA, 6, SDB_CONNECTION_LOOKUP, "[]",
 421         },
 422         /* SDB_CONNECTION_LOOKUP doesn't support filters yet */
 423         {
 424                 SDB_CONNECTION_QUERY, "FETCH host 'x1'", -1, /* does not exist */
 425                 -1, UINT32_MAX, 0, 0, NULL,
 426         },
 427         {
 428                 SDB_CONNECTION_FETCH, "\0\0\0\1x1", 7,
 429                 -1, UINT32_MAX, 0, 0, NULL,
 430         },
 431         {
 432                 SDB_CONNECTION_QUERY, "LOOKUP hosts MATCHING name = 'x1'", -1, /* does not exist */
 433                 0, SDB_CONNECTION_DATA, 6, SDB_CONNECTION_LOOKUP, "[]",
 434         },
 435         {
 436                 SDB_CONNECTION_LOOKUP, "\0\0\0\1""name = 'x1'", 16, /* does not exist */
 437                 0, SDB_CONNECTION_DATA, 6, SDB_CONNECTION_LOOKUP, "[]",
 438         },
 439         {
 440                 SDB_CONNECTION_QUERY, "FETCH host 'h1'.'s1'", -1, /* invalid args */
 441                 -1, UINT32_MAX, 0, 0, NULL,
 442         },
 443         {
 444                 SDB_CONNECTION_QUERY, "LOOKUP hosts BY name = 'x1'", -1, /* does not exist */
 445                 -1, UINT32_MAX, 0, 0, NULL,
 446         },
 447         /* services */
 448         {
 449                 SDB_CONNECTION_QUERY, "LIST services", -1,
 450                 0, SDB_CONNECTION_DATA, 320, SDB_CONNECTION_LIST, SERVICE_H2_S12_LISTING,
 451         },
 452         {
 453                 SDB_CONNECTION_LIST, "\0\0\0\2", 4,
 454                 0, SDB_CONNECTION_DATA, 320, SDB_CONNECTION_LIST, SERVICE_H2_S12_LISTING,
 455         },
 456         {
 457                 SDB_CONNECTION_QUERY, "LIST services FILTER host.name = 'h2'", -1,
 458                 0, SDB_CONNECTION_DATA, 320, SDB_CONNECTION_LIST, SERVICE_H2_S12_LISTING,
 459         },
 460         {
 461                 SDB_CONNECTION_QUERY, "LIST services FILTER host.name = 'h1'", -1,
 462                 0, SDB_CONNECTION_DATA, 6, SDB_CONNECTION_LIST, "[]",
 463         },
 464         /* SDB_CONNECTION_LIST doesn't support filters yet */
 465         {
 466                 SDB_CONNECTION_QUERY, "FETCH service 'h2'.'s1'", -1,
 467                 0, SDB_CONNECTION_DATA, 356, SDB_CONNECTION_FETCH, SERVICE_H2_S1,
 468         },
 469         /* SDB_CONNECTION_FETCH doesn't support services yet */
 470         {
 471                 SDB_CONNECTION_QUERY, "LOOKUP services MATCHING name = 's1'", -1,
 472                 0, SDB_CONNECTION_DATA, 358, SDB_CONNECTION_LOOKUP, SERVICE_H2_S1_ARRAY,
 473         },
 474         {
 475                 SDB_CONNECTION_LOOKUP, "\0\0\0\2""name = 's1'", 16,
 476                 0, SDB_CONNECTION_DATA, 358, SDB_CONNECTION_LOOKUP, SERVICE_H2_S1_ARRAY,
 477         },
 478         {
 479                 SDB_CONNECTION_QUERY, "FETCH service 'h2'.'s1' FILTER age >= 0s", -1, /* always matches */
 480                 0, SDB_CONNECTION_DATA, 356, SDB_CONNECTION_FETCH, SERVICE_H2_S1,
 481         },
 482         /* SDB_CONNECTION_FETCH doesn't support services yet */
 483         {
 484                 SDB_CONNECTION_QUERY, "LOOKUP services MATCHING name = 's1' FILTER age >= 0s", -1, /* always matches */
 485                 0, SDB_CONNECTION_DATA, 358, SDB_CONNECTION_LOOKUP, SERVICE_H2_S1_ARRAY,
 486         },
 487         {
 488                 SDB_CONNECTION_QUERY, "FETCH service 'h2'.'s1' FILTER age < 0s", -1, /* never matches */
 489                 -1, UINT32_MAX, 0, 0, NULL,
 490         },
 491         /* SDB_CONNECTION_FETCH doesn't support services yet */
 492         {
 493                 SDB_CONNECTION_QUERY, "LOOKUP services MATCHING name = 's1' FILTER age < 0s", -1, /* never matches */
 494                 0, SDB_CONNECTION_DATA, 6, SDB_CONNECTION_LOOKUP, "[]",
 495         },
 496         /* SDB_CONNECTION_LOOKUP doesn't support filters yet */
 497         {
 498                 SDB_CONNECTION_QUERY, "FETCH service 'h2'.'s1' FILTER name = 'h2'", -1, /* only matches host */
 499                 -1, UINT32_MAX, 0, 0, NULL,
 500         },
 501         /* SDB_CONNECTION_FETCH doesn't support services yet */
 502         {
 503                 SDB_CONNECTION_QUERY, "LOOKUP services MATCHING name = 's1' FILTER name = 'h2'", -1, /* only matches host */
 504                 0, SDB_CONNECTION_DATA, 6, SDB_CONNECTION_LOOKUP, "[]",
 505         },
 506         {
 507                 SDB_CONNECTION_QUERY, "FETCH service 'h2'.'x1'", -1, /* does not exist */
 508                 -1, UINT32_MAX, 0, 0, NULL,
 509         },
 510         /* SDB_CONNECTION_FETCH doesn't support services yet */
 511         {
 512                 SDB_CONNECTION_QUERY, "FETCH service 'x2'.'s1'", -1, /* does not exist */
 513                 -1, UINT32_MAX, 0, 0, NULL,
 514         },
 515         /* SDB_CONNECTION_FETCH doesn't support services yet */
 516         {
 517                 SDB_CONNECTION_QUERY, "FETCH service 'h2'", -1, /* invalid args */
 518                 -1, UINT32_MAX, 0, 0, NULL,
 519         },
 520         /* SDB_CONNECTION_FETCH doesn't support services yet */
 521         /* metrics */
 522         {
 523                 SDB_CONNECTION_QUERY, "LIST metrics", -1,
 524                 0, SDB_CONNECTION_DATA, 596, SDB_CONNECTION_LIST, METRIC_H12_M12_LISTING,
 525         },
 526         {
 527                 SDB_CONNECTION_LIST, "\0\0\0\3", 4,
 528                 0, SDB_CONNECTION_DATA, 596, SDB_CONNECTION_LIST, METRIC_H12_M12_LISTING,
 529         },
 530         {
 531                 SDB_CONNECTION_QUERY, "LIST metrics FILTER age > 0s", -1,
 532                 0, SDB_CONNECTION_DATA, 596, SDB_CONNECTION_LIST, METRIC_H12_M12_LISTING,
 533         },
 534         {
 535                 SDB_CONNECTION_QUERY, "LIST metrics FILTER age < 0s", -1,
 536                 0, SDB_CONNECTION_DATA, 6, SDB_CONNECTION_LIST, "[]",
 537         },
 538         /* SDB_CONNECTION_LIST doesn't support filters yet */
 539         {
 540                 SDB_CONNECTION_QUERY, "FETCH metric 'h1'.'m1'", -1,
 541                 0, SDB_CONNECTION_DATA, 489, SDB_CONNECTION_FETCH, METRIC_H1_M1,
 542         },
 543         /* SDB_CONNECTION_FETCH doesn't support metrics yet */
 544         {
 545                 SDB_CONNECTION_QUERY, "LOOKUP metrics MATCHING name = 'm1'", -1,
 546                 0, SDB_CONNECTION_DATA, 864, SDB_CONNECTION_LOOKUP, METRIC_H12_M1_ARRAY,
 547         },
 548         {
 549                 SDB_CONNECTION_LOOKUP, "\0\0\0\3""name = 'm1'", 16,
 550                 0, SDB_CONNECTION_DATA, 864, SDB_CONNECTION_LOOKUP, METRIC_H12_M1_ARRAY,
 551         },
 552         {
 553                 SDB_CONNECTION_QUERY, "FETCH metric 'h1'.'m1' FILTER age >= 0s", -1, /* always matches */
 554                 0, SDB_CONNECTION_DATA, 489, SDB_CONNECTION_FETCH, METRIC_H1_M1,
 555         },
 556         /* SDB_CONNECTION_FETCH doesn't support metrics yet */
 557         {
 558                 SDB_CONNECTION_QUERY, "LOOKUP metrics MATCHING name = 'm1' FILTER age >= 0s", -1, /* always matches */
 559                 0, SDB_CONNECTION_DATA, 864, SDB_CONNECTION_LOOKUP, METRIC_H12_M1_ARRAY,
 560         },
 561         {
 562                 SDB_CONNECTION_QUERY, "FETCH metric 'h1'.'m1' FILTER age < 0s", -1, /* never matches */
 563                 -1, UINT32_MAX, 0, 0, NULL,
 564         },
 565         /* SDB_CONNECTION_FETCH doesn't support metrics yet */
 566         {
 567                 SDB_CONNECTION_QUERY, "LOOKUP metrics MATCHING name = 'm1' FILTER age < 0s", -1, /* never matches */
 568                 0, SDB_CONNECTION_DATA, 6, SDB_CONNECTION_LOOKUP, "[]",
 569         },
 570         /* SDB_CONEECTION_LOOKUP doesn't support filters yet */
 571         {
 572                 SDB_CONNECTION_QUERY, "FETCH metric 'h1'.'m1' FILTER name = 'h1'", -1, /* only matches host */
 573                 -1, UINT32_MAX, 0, 0, NULL,
 574         },
 575         /* SDB_CONNECTION_FETCH doesn't support metrics yet */
 576         {
 577                 SDB_CONNECTION_QUERY, "LOOKUP metrics MATCHING name = 'm1' FILTER name = 'h1'", -1, /* only matches host */
 578                 0, SDB_CONNECTION_DATA, 6, SDB_CONNECTION_LOOKUP, "[]",
 579         },
 580         {
 581                 SDB_CONNECTION_QUERY, "FETCH metric 'h1'.'x1'", -1, /* does not exist */
 582                 -1, UINT32_MAX, 0, 0, NULL,
 583         },
 584         /* SDB_CONNECTION_FETCH doesn't support metrics yet */
 585         {
 586                 SDB_CONNECTION_QUERY, "FETCH metric 'x1'.'m1'", -1, /* does not exist */
 587                 -1, UINT32_MAX, 0, 0, NULL,
 588         },
 589         /* SDB_CONNECTION_FETCH doesn't support metrics yet */
 590         {
 591                 SDB_CONNECTION_QUERY, "FETCH metric 'x1'", -1, /* invalid args */
 592                 -1, UINT32_MAX, 0, 0, NULL,
 593         },
 594         /* SDB_CONNECTION_FETCH doesn't support metrics yet */
 595         /* timeseries */
 596         {
 597                 SDB_CONNECTION_QUERY, "TIMESERIES 'h1'.'m1'", -1,
 598                 -1, UINT32_MAX, 0, 0, NULL, /* no data-store available */
 599         },
 600         {
 601                 SDB_CONNECTION_QUERY, "TIMESERIES 'h1'.'x1'", -1,
 602                 -1, UINT32_MAX, 0, 0, NULL, /* does not exist */
 603         },
 604         {
 605                 SDB_CONNECTION_QUERY, "TIMESERIES 'x1'.'m1'", -1,
 606                 -1, UINT32_MAX, 0, 0, NULL, /* does not exist */
 607         },
 608         /* store commands */
 609         {
 610                 SDB_CONNECTION_QUERY, "STORE host 'hA' LAST UPDATE 01:00", -1,
 611                 0, SDB_CONNECTION_OK, 27, 0, "Successfully stored host hA",
 612         },
 613         {
 614                 SDB_CONNECTION_STORE, "\0\0\0\1""\0\0\0\0\xd6\x93\xa4\0""hA", 15,
 615                 0, SDB_CONNECTION_OK, 27, 0, "Successfully stored host hA",
 616         },
 617         {
 618                 SDB_CONNECTION_QUERY, "STORE host 'hA'", -1,
 619                 0, SDB_CONNECTION_OK, 27, 0, "Successfully stored host hA",
 620         },
 621         {
 622                 SDB_CONNECTION_QUERY, "STORE host attribute 'h1'.'aA' 'vA'", -1,
 623                 0, SDB_CONNECTION_OK, 40, 0, "Successfully stored host attribute h1.aA",
 624         },
 625         {
 626                 SDB_CONNECTION_STORE, "\0\0\0\x11""\0\0\0\0\xd6\x93\xa4\0""h1\0aA\0"VALUE, 18+VALUE_LEN,
 627                 0, SDB_CONNECTION_OK, 40, 0, "Successfully stored host attribute h1.aA",
 628         },
 629         {
 630                 SDB_CONNECTION_QUERY, "STORE host attribute 'x1'.'aA' 'vA'", -1,
 631                 -1, UINT32_MAX, 0, 0, NULL,
 632         },
 633         {
 634                 SDB_CONNECTION_STORE, "\0\0\0\x11""\0\0\0\0\xd6\x93\xa4\0""x1\0aA\0"VALUE, 18+VALUE_LEN,
 635                 -1, UINT32_MAX, 0, 0, NULL,
 636         },
 637         {
 638                 SDB_CONNECTION_QUERY, "STORE service 'h1'.'sA'", -1,
 639                 0, SDB_CONNECTION_OK, 33, 0, "Successfully stored service h1.sA",
 640         },
 641         {
 642                 SDB_CONNECTION_STORE, "\0\0\0\2""\0\0\0\0\xd6\x93\xa4\0""h1\0sA", 18,
 643                 0, SDB_CONNECTION_OK, 33, 0, "Successfully stored service h1.sA",
 644         },
 645         {
 646                 SDB_CONNECTION_QUERY, "STORE service 'x1'.'sA'", -1,
 647                 -1, UINT32_MAX, 0, 0, NULL,
 648         },
 649         {
 650                 SDB_CONNECTION_STORE, "\0\0\0\x12""\0\0\0\0\xd6\x93\xa4\0""x1\0sA", 18,
 651                 -1, UINT32_MAX, 0, 0, NULL,
 652         },
 653         {
 654                 SDB_CONNECTION_QUERY, "STORE service attribute 'h2'.'s1'.'aA' 'vA'", -1,
 655                 0, SDB_CONNECTION_OK, 46, 0, "Successfully stored service attribute h2.s1.aA",
 656         },
 657         {
 658                 SDB_CONNECTION_STORE, "\0\0\0\x12""\0\0\0\0\xd6\x93\xa4\0""h2\0s1\0aA\0"VALUE,27+VALUE_LEN,
 659                 0, SDB_CONNECTION_OK, 46, 0, "Successfully stored service attribute h2.s1.aA",
 660         },
 661         {
 662                 SDB_CONNECTION_QUERY, "STORE service attribute 'h2'.'x1'.'aA' 'vA'", -1,
 663                 -1, UINT32_MAX, 0, 0, NULL,
 664         },
 665         {
 666                 SDB_CONNECTION_STORE, "\0\0\0\x12""\0\0\0\0\xd6\x93\xa4\0""h2\0x1\0aA\0"VALUE,27+VALUE_LEN,
 667                 -1, UINT32_MAX, 0, 0, NULL,
 668         },
 669         {
 670                 SDB_CONNECTION_QUERY, "STORE metric 'h1'.'mA'", -1,
 671                 0, SDB_CONNECTION_OK, 32, 0, "Successfully stored metric h1.mA",
 672         },
 673         {
 674                 SDB_CONNECTION_STORE, "\0\0\0\3""\0\0\0\0\xd6\x93\xa4\0""h1\0mA", 18,
 675                 0, SDB_CONNECTION_OK, 32, 0, "Successfully stored metric h1.mA",
 676         },
 677         {
 678                 SDB_CONNECTION_QUERY, "STORE metric 'x1'.'mA'", -1,
 679                 -1, UINT32_MAX, 0, 0, NULL,
 680         },
 681         {
 682                 SDB_CONNECTION_STORE, "\0\0\0\3""\0\0\0\0\xd6\x93\xa4\0""x1\0mA", 18,
 683                 -1, UINT32_MAX, 0, 0, NULL,
 684         },
 685         {
 686                 SDB_CONNECTION_QUERY, "STORE metric attribute 'h1'.'m1'.'aA' 'vA'", -1,
 687                 0, SDB_CONNECTION_OK, 45, 0, "Successfully stored metric attribute h1.m1.aA",
 688         },
 689         {
 690                 SDB_CONNECTION_STORE, "\0\0\0\x13""\0\0\0\0\xd6\x93\xa4\0""h1\0m1\0aA\0"VALUE, 27+VALUE_LEN,
 691                 0, SDB_CONNECTION_OK, 45, 0, "Successfully stored metric attribute h1.m1.aA",
 692         },
 693         {
 694                 SDB_CONNECTION_QUERY, "STORE metric attribute 'h1'.'x1'.'aA' 'vA'", -1,
 695                 -1, UINT32_MAX, 0, 0, NULL,
 696         },
 697         {
 698                 SDB_CONNECTION_STORE, "\0\0\0\x13""\0\0\0\0\xd6\x93\xa4\0""h1\0x1\0aA\0"VALUE, 27+VALUE_LEN,
 699                 -1, UINT32_MAX, 0, 0, NULL,
 700         },
 701 };
 702
 703 START_TEST(test_query)
 704 {
 705         sdb_conn_t *conn = mock_conn_create();
 706
 707         uint32_t code = UINT32_MAX, msg_len = UINT32_MAX;
 708         const char *data;
 709         ssize_t tmp;
 710         size_t len;
 711         int check = -1;
 712
 713         conn->cmd = query_data[_i].cmd;
 714         if (query_data[_i].query_len < 0)
 715                 conn->cmd_len = (uint32_t)strlen(query_data[_i].query);
 716         else
 717                 conn->cmd_len = (uint32_t)query_data[_i].query_len;
 718         sdb_strbuf_memcpy(conn->buf, query_data[_i].query, conn->cmd_len);
 719
 720         switch (conn->cmd) {
 721         case SDB_CONNECTION_QUERY:
 722                 check = sdb_conn_query(conn);
 723                 break;
 724         case SDB_CONNECTION_FETCH:
 725                 check = sdb_conn_fetch(conn);
 726                 break;
 727         case SDB_CONNECTION_LIST:
 728                 check = sdb_conn_list(conn);
 729                 break;
 730         case SDB_CONNECTION_LOOKUP:
 731                 check = sdb_conn_lookup(conn);
 732                 break;
 733         /* SDB_CONNECTION_TIMESERIES not supported yet */
 734         case SDB_CONNECTION_STORE:
 735                 check = sdb_conn_store(conn);
 736                 break;
 737         default:
 738                 fail("Invalid command %#x", conn->cmd);
 739         }
 740
 741         fail_unless(check == query_data[_i].expected,
 742                         "sdb_conn_query(%s) = %d; expected: %d (err: %s)",
 743                         query_data[_i].query, check, query_data[_i].expected,
 744                         sdb_strbuf_string(conn->errbuf));
 745
 746         data = sdb_strbuf_string(MOCK_CONN(conn)->write_buf);
 747         len = sdb_strbuf_len(MOCK_CONN(conn)->write_buf);
 748
 749         if (query_data[_i].code == UINT32_MAX) {
 750                 fail_unless(len == 0,
 751                                 "sdb_conn_query(%s) returned data on error: '%s'",
 752                         query_data[_i].query, data);
 753                 mock_conn_destroy(conn);
 754                 return;
 755         }
 756
 757         tmp = sdb_proto_unmarshal_header(data, len, &code, &msg_len);
 758         ck_assert_msg(tmp == (ssize_t)(2 * sizeof(uint32_t)));
 759         data += tmp;
 760         len -= tmp;
 761
 762         fail_unless((code == query_data[_i].code)
 763                                 && ((size_t)msg_len == query_data[_i].len),
 764                         "sdb_conn_query(%s) returned %u, %u; expected: %u, %zu",
 765                         query_data[_i].query, code, msg_len,
 766                         query_data[_i].code, query_data[_i].len);
 767
 768         if (code == SDB_CONNECTION_DATA) {
 769                 tmp = sdb_proto_unmarshal_int32(data, len, &code);
 770                 fail_unless(code == query_data[_i].type,
 771                                 "sdb_conn_query(%s) returned %s object; expected: %s",
 772                                 query_data[_i].query, SDB_CONN_MSGTYPE_TO_STRING((int)code),
 773                                 SDB_CONN_MSGTYPE_TO_STRING((int)query_data[_i].type));
 774                 data += tmp;
 775                 len -= tmp;
 776         }
 777
 778         fail_if_strneq(data, query_data[_i].data, (size_t)msg_len,
 779                         "sdb_conn_query(%s) returned unexpected data",
 780                         query_data[_i].query, data, query_data[_i].data);
 781
 782         mock_conn_destroy(conn);
 783 }
 784 END_TEST
 785
 786 TEST_MAIN("frontend::query")
 787 {
 788         TCase *tc = tcase_create("core");
 789         tcase_add_checked_fixture(tc, populate, turndown);
 790         TC_ADD_LOOP_TEST(tc, query);
 791         ADD_TCASE(tc);
 792 }
 793 TEST_MAIN_END
 794
 795 /* vim: set tw=78 sw=4 ts=4 noexpandtab : */
 796