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 "frontend/connection.h"
  33 #include "frontend/parser.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_data_t datum;
  47
  48         sdb_store_host("h1", 1);
  49         sdb_store_host("h2", 3);
  50
  51         datum.type = SDB_TYPE_STRING;
  52         datum.data.string = "v1";
  53         sdb_store_attribute("h1", "k1", &datum, 1);
  54         datum.data.string = "v2";
  55         sdb_store_attribute("h1", "k2", &datum, 2);
  56         datum.data.string = "v3";
  57         sdb_store_attribute("h1", "k3", &datum, 2);
  58
  59         sdb_store_metric("h1", "m1", /* store */ NULL, 2);
  60         sdb_store_metric("h1", "m2", /* store */ NULL, 1);
  61         sdb_store_metric("h2", "m1", /* store */ NULL, 1);
  62
  63         datum.type = SDB_TYPE_INTEGER;
  64         datum.data.integer = 42;
  65         sdb_store_metric_attr("h1", "m1", "k3", &datum, 2);
  66
  67         sdb_store_service("h2", "s1", 1);
  68         sdb_store_service("h2", "s2", 2);
  69
  70         datum.data.integer = 123;
  71         sdb_store_service_attr("h2", "s2", "k1", &datum, 2);
  72         datum.data.integer = 4711;
  73         sdb_store_service_attr("h2", "s2", "k2", &datum, 1);
  74 } /* populate */
  75
  76 #define HOST_H1 \
  77         "{\"name\": \"h1\", \"last_update\": \"1970-01-01 00:00:00 +0000\", " \
  78                         "\"update_interval\": \"0s\", \"backends\": [], " \
  79                 "\"attributes\": [" \
  80                         "{\"name\": \"k1\", \"value\": \"v1\", " \
  81                                 "\"last_update\": \"1970-01-01 00:00:00 +0000\", " \
  82                                 "\"update_interval\": \"0s\", \"backends\": []}," \
  83                         "{\"name\": \"k2\", \"value\": \"v2\", " \
  84                                 "\"last_update\": \"1970-01-01 00:00:00 +0000\", " \
  85                                 "\"update_interval\": \"0s\", \"backends\": []}," \
  86                         "{\"name\": \"k3\", \"value\": \"v3\", " \
  87                                 "\"last_update\": \"1970-01-01 00:00:00 +0000\", " \
  88                                 "\"update_interval\": \"0s\", \"backends\": []}], " \
  89                 "\"metrics\": [" \
  90                         "{\"name\": \"m1\", \"timeseries\": false, " \
  91                                 "\"last_update\": \"1970-01-01 00:00:00 +0000\", " \
  92                                 "\"update_interval\": \"0s\", \"backends\": [], " \
  93                                 "\"attributes\": [" \
  94                                         "{\"name\": \"hostname\", \"value\": \"h1\", " \
  95                                                 "\"last_update\": \"1970-01-01 00:00:00 +0000\", " \
  96                                                 "\"update_interval\": \"0s\", \"backends\": []}," \
  97                                         "{\"name\": \"k3\", \"value\": 42, " \
  98                                                 "\"last_update\": \"1970-01-01 00:00:00 +0000\", " \
  99                                                 "\"update_interval\": \"0s\", \"backends\": []}]}," \
 100                         "{\"name\": \"m2\", \"timeseries\": false, " \
 101                                 "\"last_update\": \"1970-01-01 00:00:00 +0000\", " \
 102                                 "\"update_interval\": \"0s\", \"backends\": [], " \
 103                                 "\"attributes\": [" \
 104                                         "{\"name\": \"hostname\", \"value\": \"h1\", " \
 105                                                 "\"last_update\": \"1970-01-01 00:00:00 +0000\", " \
 106                                                 "\"update_interval\": \"0s\", \"backends\": []}]}]}"
 107 #define HOST_H1_ARRAY "["HOST_H1"]"
 108 #define HOST_H1_LISTING \
 109         "{\"name\": \"h1\", \"last_update\": \"1970-01-01 00:00:00 +0000\", " \
 110                         "\"update_interval\": \"0s\", \"backends\": []}"
 111 #define HOST_H2_LISTING \
 112         "{\"name\": \"h2\", \"last_update\": \"1970-01-01 00:00:00 +0000\", " \
 113                         "\"update_interval\": \"0s\", \"backends\": []}"
 114
 115 #define SERVICE_H2_S1 \
 116         "{\"name\": \"h2\", \"last_update\": \"1970-01-01 00:00:00 +0000\", " \
 117                         "\"update_interval\": \"0s\", \"backends\": [], " \
 118                 "\"services\": [" \
 119                         "{\"name\": \"s1\", \"last_update\": \"1970-01-01 00:00:00 +0000\", " \
 120                                 "\"update_interval\": \"0s\", \"backends\": [], " \
 121                                 "\"attributes\": [" \
 122                                         "{\"name\": \"hostname\", \"value\": \"h2\", " \
 123                                                 "\"last_update\": \"1970-01-01 00:00:00 +0000\", " \
 124                                                 "\"update_interval\": \"0s\", \"backends\": []}]}]}"
 125 #define SERVICE_H2_S1_ARRAY "["SERVICE_H2_S1"]"
 126 #define SERVICE_H2_S12_LISTING \
 127         "[{\"name\": \"h2\", \"last_update\": \"1970-01-01 00:00:00 +0000\", " \
 128                         "\"update_interval\": \"0s\", \"backends\": [], " \
 129                 "\"services\": [" \
 130                         "{\"name\": \"s1\", \"last_update\": \"1970-01-01 00:00:00 +0000\", " \
 131                                 "\"update_interval\": \"0s\", \"backends\": []}," \
 132                         "{\"name\": \"s2\", \"last_update\": \"1970-01-01 00:00:00 +0000\", " \
 133                                 "\"update_interval\": \"0s\", \"backends\": []}]}]"
 134
 135 #define METRIC_H1_M1 \
 136         "{\"name\": \"h1\", \"last_update\": \"1970-01-01 00:00:00 +0000\", " \
 137                         "\"update_interval\": \"0s\", \"backends\": [], " \
 138                 "\"metrics\": [" \
 139                         "{\"name\": \"m1\", \"timeseries\": false, " \
 140                                 "\"last_update\": \"1970-01-01 00:00:00 +0000\", " \
 141                                 "\"update_interval\": \"0s\", \"backends\": [], " \
 142                                 "\"attributes\": [" \
 143                                         "{\"name\": \"hostname\", \"value\": \"h1\", " \
 144                                                 "\"last_update\": \"1970-01-01 00:00:00 +0000\", " \
 145                                                 "\"update_interval\": \"0s\", \"backends\": []}," \
 146                                         "{\"name\": \"k3\", \"value\": 42, " \
 147                                                 "\"last_update\": \"1970-01-01 00:00:00 +0000\", " \
 148                                                 "\"update_interval\": \"0s\", \"backends\": []}]}]}"
 149 #define METRIC_H12_M1_ARRAY \
 150         "["METRIC_H1_M1"," \
 151         "{\"name\": \"h2\", \"last_update\": \"1970-01-01 00:00:00 +0000\", " \
 152                         "\"update_interval\": \"0s\", \"backends\": [], " \
 153                 "\"metrics\": [" \
 154                         "{\"name\": \"m1\", \"timeseries\": false, " \
 155                                 "\"last_update\": \"1970-01-01 00:00:00 +0000\", " \
 156                                 "\"update_interval\": \"0s\", \"backends\": [], " \
 157                                 "\"attributes\": [" \
 158                                         "{\"name\": \"hostname\", \"value\": \"h2\", " \
 159                                                 "\"last_update\": \"1970-01-01 00:00:00 +0000\", " \
 160                                                 "\"update_interval\": \"0s\", \"backends\": []}]}]}]"
 161 #define METRIC_H12_M12_LISTING \
 162         "[{\"name\": \"h1\", \"last_update\": \"1970-01-01 00:00:00 +0000\", " \
 163                         "\"update_interval\": \"0s\", \"backends\": [], " \
 164                 "\"metrics\": [" \
 165                         "{\"name\": \"m1\", \"timeseries\": false, " \
 166                                 "\"last_update\": \"1970-01-01 00:00:00 +0000\", " \
 167                                 "\"update_interval\": \"0s\", \"backends\": []}," \
 168                         "{\"name\": \"m2\", \"timeseries\": false, " \
 169                                 "\"last_update\": \"1970-01-01 00:00:00 +0000\", " \
 170                                 "\"update_interval\": \"0s\", \"backends\": []}]}," \
 171         "{\"name\": \"h2\", \"last_update\": \"1970-01-01 00:00:00 +0000\", " \
 172                         "\"update_interval\": \"0s\", \"backends\": [], " \
 173                 "\"metrics\": [" \
 174                         "{\"name\": \"m1\", \"timeseries\": false, " \
 175                                 "\"last_update\": \"1970-01-01 00:00:00 +0000\", " \
 176                                 "\"update_interval\": \"0s\", \"backends\": []}]}]"
 177
 178 typedef struct {
 179         sdb_conn_t conn;
 180         sdb_strbuf_t *write_buf;
 181 } mock_conn_t;
 182 #define MOCK_CONN(obj) ((mock_conn_t *)(obj))
 183 #define CONN(obj) ((sdb_conn_t *)(obj))
 184
 185 static void
 186 mock_conn_destroy(sdb_conn_t *conn)
 187 {
 188         sdb_strbuf_destroy(conn->buf);
 189         sdb_strbuf_destroy(conn->errbuf);
 190         sdb_strbuf_destroy(MOCK_CONN(conn)->write_buf);
 191         free(conn);
 192 } /* mock_conn_destroy */
 193
 194 static ssize_t
 195 mock_conn_read(sdb_conn_t *conn, size_t len)
 196 {
 197         if (! conn)
 198                 return -1;
 199         /* unused so far */
 200         return len;
 201 } /* conn_read */
 202
 203 static ssize_t
 204 mock_conn_write(sdb_conn_t *conn, const void *buf, size_t len)
 205 {
 206         if (! conn)
 207                 return -1;
 208         return sdb_strbuf_memappend(MOCK_CONN(conn)->write_buf, buf, len);
 209 } /* conn_write */
 210
 211 static sdb_conn_t *
 212 mock_conn_create(void)
 213 {
 214         mock_conn_t *conn;
 215
 216         conn = calloc(1, sizeof(*conn));
 217         if (! conn) {
 218                 fail("INTERNAL ERROR: failed to allocate connection object");
 219                 return NULL;
 220         }
 221
 222         SDB_OBJ(conn)->name = "mock_connection";
 223         SDB_OBJ(conn)->ref_cnt = 1;
 224
 225         conn->conn.buf = sdb_strbuf_create(0);
 226         conn->conn.errbuf = sdb_strbuf_create(0);
 227         conn->write_buf = sdb_strbuf_create(64);
 228         if ((! conn->conn.buf) || (! conn->conn.errbuf) || (! conn->write_buf)) {
 229                 mock_conn_destroy(CONN(conn));
 230                 fail("INTERNAL ERROR: failed to allocate connection object");
 231                 return NULL;
 232         }
 233
 234         conn->conn.read = mock_conn_read;
 235         conn->conn.write = mock_conn_write;
 236
 237         conn->conn.username = "mock_user";
 238         conn->conn.cmd = SDB_CONNECTION_IDLE;
 239         conn->conn.cmd_len = 0;
 240         return CONN(conn);
 241 } /* mock_conn_create */
 242
 243 /* TODO: move this into a test helper module */
 244 static void
 245 fail_if_strneq(const char *got, const char *expected, size_t n, const char *fmt, ...)
 246 {
 247         sdb_strbuf_t *buf;
 248         va_list ap;
 249
 250         size_t len1 = strlen(got);
 251         size_t len2 = strlen(expected);
 252
 253         size_t i;
 254         int pos = -1;
 255
 256         if (n) {
 257                 len1 = SDB_MIN(len1, n);
 258                 len2 = SDB_MIN(len2, n);
 259         }
 260
 261         if (len1 != len2)
 262                 pos = (int)SDB_MIN(len1, len2);
 263
 264         for (i = 0; i < SDB_MIN(len1, len2); ++i) {
 265                 if (got[i] != expected[i]) {
 266                         pos = (int)i;
 267                         break;
 268                 }
 269         }
 270
 271         if (pos == -1)
 272                 return;
 273
 274         buf = sdb_strbuf_create(64);
 275         va_start(ap, fmt);
 276         sdb_strbuf_vsprintf(buf, fmt, ap);
 277
 278         fail("%s\n         got: %s\n              %*s\n    expected: %s",
 279                         sdb_strbuf_string(buf), got, pos + 1, "^", expected);
 280 } /* fail_if_strneq */
 281
 282 /*
 283  * tests
 284  */
 285
 286 static struct {
 287         const char *query;
 288         int expected;
 289         uint32_t code;
 290         size_t len;
 291         uint32_t type;
 292         const char *data;
 293 } query_data[] = {
 294         /* hosts */
 295         {
 296                 "LIST hosts",
 297                 0, SDB_CONNECTION_DATA, 205, SDB_CONNECTION_LIST,
 298                 "["HOST_H1_LISTING","HOST_H2_LISTING"]",
 299         },
 300         {
 301                 "LIST hosts; LIST hosts", /* ignore second (and later) commands */
 302                 0, SDB_CONNECTION_DATA, 205, SDB_CONNECTION_LIST,
 303                 "["HOST_H1_LISTING","HOST_H2_LISTING"]",
 304         },
 305         {
 306                 "LIST hosts FILTER name = 'h1'",
 307                 0, SDB_CONNECTION_DATA, 105, SDB_CONNECTION_LIST, "["HOST_H1_LISTING"]",
 308         },
 309         {
 310                 "LIST hosts FILTER name = 's1'",
 311                 0, SDB_CONNECTION_DATA, 6, SDB_CONNECTION_LIST, "[]",
 312         },
 313         {
 314                 "FETCH host 'h1'",
 315                 0, SDB_CONNECTION_DATA, 1110, SDB_CONNECTION_FETCH, HOST_H1,
 316         },
 317         {
 318                 "LOOKUP hosts MATCHING name = 'h1'",
 319                 0, SDB_CONNECTION_DATA, 1112, SDB_CONNECTION_LOOKUP, HOST_H1_ARRAY,
 320         },
 321         {
 322                 "FETCH host 'h1' FILTER age >= 0s", /* always matches */
 323                 0, SDB_CONNECTION_DATA, 1110, SDB_CONNECTION_FETCH, HOST_H1,
 324         },
 325         {
 326                 "LOOKUP hosts MATCHING name = 'h1' FILTER age >= 0s", /* always matches */
 327                 0, SDB_CONNECTION_DATA, 1112, SDB_CONNECTION_LOOKUP, HOST_H1_ARRAY,
 328         },
 329         {
 330                 "FETCH host 'h1' FILTER age < 0s", /* never matches */
 331                 -1, UINT32_MAX, 0, 0, NULL, /* FETCH fails if the object doesn't exist */
 332         },
 333         {
 334                 "LOOKUP hosts MATCHING name = 'h1' FILTER age < 0s", /* never matches */
 335                 0, SDB_CONNECTION_DATA, 6, SDB_CONNECTION_LOOKUP, "[]",
 336         },
 337         {
 338                 "FETCH host 'x1'", /* does not exist */
 339                 -1, UINT32_MAX, 0, 0, NULL,
 340         },
 341         {
 342                 "LOOKUP hosts MATCHING name = 'x1'", /* does not exist */
 343                 0, SDB_CONNECTION_DATA, 6, SDB_CONNECTION_LOOKUP, "[]",
 344         },
 345         {
 346                 "FETCH host 'h1'.'s1'", /* invalid args */
 347                 -1, UINT32_MAX, 0, 0, NULL,
 348         },
 349         {
 350                 "LOOKUP hosts BY name = 'x1'", /* does not exist */
 351                 -1, UINT32_MAX, 0, 0, NULL,
 352         },
 353         /* services */
 354         {
 355                 "LIST services",
 356                 0, SDB_CONNECTION_DATA, 320, SDB_CONNECTION_LIST, SERVICE_H2_S12_LISTING,
 357         },
 358         {
 359                 "LIST services FILTER host.name = 'h2'",
 360                 0, SDB_CONNECTION_DATA, 320, SDB_CONNECTION_LIST, SERVICE_H2_S12_LISTING,
 361         },
 362         {
 363                 "LIST services FILTER host.name = 'h1'",
 364                 0, SDB_CONNECTION_DATA, 6, SDB_CONNECTION_LIST, "[]",
 365         },
 366         {
 367                 "FETCH service 'h2'.'s1'",
 368                 0, SDB_CONNECTION_DATA, 356, SDB_CONNECTION_FETCH, SERVICE_H2_S1,
 369         },
 370         {
 371                 "LOOKUP services MATCHING name = 's1'",
 372                 0, SDB_CONNECTION_DATA, 358, SDB_CONNECTION_LOOKUP, SERVICE_H2_S1_ARRAY,
 373         },
 374         {
 375                 "FETCH service 'h2'.'s1' FILTER age >= 0s", /* always matches */
 376                 0, SDB_CONNECTION_DATA, 356, SDB_CONNECTION_FETCH, SERVICE_H2_S1,
 377         },
 378         {
 379                 "LOOKUP services MATCHING name = 's1' FILTER age >= 0s", /* always matches */
 380                 0, SDB_CONNECTION_DATA, 358, SDB_CONNECTION_LOOKUP, SERVICE_H2_S1_ARRAY,
 381         },
 382         {
 383                 "FETCH service 'h2'.'s1' FILTER age < 0s", /* never matches */
 384                 -1, UINT32_MAX, 0, 0, NULL,
 385         },
 386         {
 387                 "LOOKUP services MATCHING name = 's1' FILTER age < 0s", /* never matches */
 388                 0, SDB_CONNECTION_DATA, 6, SDB_CONNECTION_LOOKUP, "[]",
 389         },
 390         {
 391                 "FETCH service 'h2'.'s1' FILTER name = 'h2'", /* only matches host */
 392                 -1, UINT32_MAX, 0, 0, NULL,
 393         },
 394         {
 395                 "LOOKUP services MATCHING name = 's1' FILTER name = 'h2'", /* only matches host */
 396                 0, SDB_CONNECTION_DATA, 6, SDB_CONNECTION_LOOKUP, "[]",
 397         },
 398         {
 399                 "FETCH service 'h2'.'x1'", /* does not exist */
 400                 -1, UINT32_MAX, 0, 0, NULL,
 401         },
 402         {
 403                 "FETCH service 'x2'.'s1'", /* does not exist */
 404                 -1, UINT32_MAX, 0, 0, NULL,
 405         },
 406         {
 407                 "FETCH service 'h2'", /* invalid args */
 408                 -1, UINT32_MAX, 0, 0, NULL,
 409         },
 410         /* metrics */
 411         {
 412                 "LIST metrics",
 413                 0, SDB_CONNECTION_DATA, 596, SDB_CONNECTION_LIST, METRIC_H12_M12_LISTING,
 414         },
 415         {
 416                 "LIST metrics FILTER age > 0s",
 417                 0, SDB_CONNECTION_DATA, 596, SDB_CONNECTION_LIST, METRIC_H12_M12_LISTING,
 418         },
 419         {
 420                 "LIST metrics FILTER age < 0s",
 421                 0, SDB_CONNECTION_DATA, 6, SDB_CONNECTION_LIST, "[]",
 422         },
 423         {
 424                 "FETCH metric 'h1'.'m1'",
 425                 0, SDB_CONNECTION_DATA, 489, SDB_CONNECTION_FETCH, METRIC_H1_M1,
 426         },
 427         {
 428                 "LOOKUP metrics MATCHING name = 'm1'",
 429                 0, SDB_CONNECTION_DATA, 864, SDB_CONNECTION_LOOKUP, METRIC_H12_M1_ARRAY,
 430         },
 431         {
 432                 "FETCH metric 'h1'.'m1' FILTER age >= 0s", /* always matches */
 433                 0, SDB_CONNECTION_DATA, 489, SDB_CONNECTION_FETCH, METRIC_H1_M1,
 434         },
 435         {
 436                 "LOOKUP metrics MATCHING name = 'm1' FILTER age >= 0s", /* always matches */
 437                 0, SDB_CONNECTION_DATA, 864, SDB_CONNECTION_LOOKUP, METRIC_H12_M1_ARRAY,
 438         },
 439         {
 440                 "FETCH metric 'h1'.'m1' FILTER age < 0s", /* never matches */
 441                 -1, UINT32_MAX, 0, 0, NULL,
 442         },
 443         {
 444                 "LOOKUP metrics MATCHING name = 'm1' FILTER age < 0s", /* never matches */
 445                 0, SDB_CONNECTION_DATA, 6, SDB_CONNECTION_LOOKUP, "[]",
 446         },
 447         {
 448                 "FETCH metric 'h1'.'m1' FILTER name = 'h1'", /* only matches host */
 449                 -1, UINT32_MAX, 0, 0, NULL,
 450         },
 451         {
 452                 "LOOKUP metrics MATCHING name = 'm1' FILTER name = 'h1'", /* only matches host */
 453                 0, SDB_CONNECTION_DATA, 6, SDB_CONNECTION_LOOKUP, "[]",
 454         },
 455         {
 456                 "FETCH metric 'h1'.'x1'", /* does not exist */
 457                 -1, UINT32_MAX, 0, 0, NULL,
 458         },
 459         {
 460                 "FETCH metric 'x1'.'m1'", /* does not exist */
 461                 -1, UINT32_MAX, 0, 0, NULL,
 462         },
 463         {
 464                 "FETCH metric 'x1'", /* invalid args */
 465                 -1, UINT32_MAX, 0, 0, NULL,
 466         },
 467         /* timeseries */
 468         {
 469                 "TIMESERIES 'h1'.'m1'",
 470                 -1, UINT32_MAX, 0, 0, NULL, /* no data-store available */
 471         },
 472         {
 473                 "TIMESERIES 'h1'.'x1'",
 474                 -1, UINT32_MAX, 0, 0, NULL, /* does not exist */
 475         },
 476         {
 477                 "TIMESERIES 'x1'.'m1'",
 478                 -1, UINT32_MAX, 0, 0, NULL, /* does not exist */
 479         },
 480         /* store commands */
 481         {
 482                 "STORE host 'hA' LAST UPDATE 01:00",
 483                 0, SDB_CONNECTION_OK, 27, 0, "Successfully stored host hA",
 484         },
 485         {
 486                 "STORE host 'hA'",
 487                 0, SDB_CONNECTION_OK, 27, 0, "Successfully stored host hA",
 488         },
 489         {
 490                 "STORE host attribute 'h1'.'aA' 'vA'",
 491                 0, SDB_CONNECTION_OK, 40, 0, "Successfully stored host attribute h1.aA",
 492         },
 493         {
 494                 "STORE host attribute 'x1'.'aA' 'vA'",
 495                 -1, UINT32_MAX, 0, 0, NULL,
 496         },
 497         {
 498                 "STORE service 'h1'.'sA'",
 499                 0, SDB_CONNECTION_OK, 33, 0, "Successfully stored service h1.sA",
 500         },
 501         {
 502                 "STORE service 'x1'.'sA'",
 503                 -1, UINT32_MAX, 0, 0, NULL,
 504         },
 505         {
 506                 "STORE service attribute 'h2'.'s1'.'aA' 'vA'",
 507                 0, SDB_CONNECTION_OK, 43, 0, "Successfully stored service attribute s1.aA",
 508         },
 509         {
 510                 "STORE service attribute 'h2'.'x1'.'aA' 'vA'",
 511                 -1, UINT32_MAX, 0, 0, NULL,
 512         },
 513         {
 514                 "STORE metric 'h1'.'mA'",
 515                 0, SDB_CONNECTION_OK, 32, 0, "Successfully stored metric h1.mA",
 516         },
 517         {
 518                 "STORE metric 'x1'.'mA'",
 519                 -1, UINT32_MAX, 0, 0, NULL,
 520         },
 521         {
 522                 "STORE metric attribute 'h1'.'m1'.'aA' 'vA'",
 523                 0, SDB_CONNECTION_OK, 42, 0, "Successfully stored metric attribute m1.aA",
 524         },
 525         {
 526                 "STORE metric attribute 'h1'.'x1'.'aA' 'vA'",
 527                 -1, UINT32_MAX, 0, 0, NULL,
 528         },
 529 };
 530
 531 START_TEST(test_query)
 532 {
 533         sdb_conn_t *conn = mock_conn_create();
 534
 535         uint32_t code = UINT32_MAX, msg_len = UINT32_MAX;
 536         const char *data;
 537         ssize_t tmp;
 538         size_t len;
 539         int check;
 540
 541         conn->cmd = SDB_CONNECTION_QUERY;
 542         conn->cmd_len = (uint32_t)strlen(query_data[_i].query);
 543         sdb_strbuf_memcpy(conn->buf, query_data[_i].query, conn->cmd_len);
 544
 545         check = sdb_fe_query(conn);
 546         fail_unless(check == query_data[_i].expected,
 547                         "sdb_fe_query(%s) = %d; expected: %d",
 548                         query_data[_i].query, check, query_data[_i].expected);
 549
 550         data = sdb_strbuf_string(MOCK_CONN(conn)->write_buf);
 551         len = sdb_strbuf_len(MOCK_CONN(conn)->write_buf);
 552
 553         if (query_data[_i].code == UINT32_MAX) {
 554                 fail_unless(len == 0,
 555                                 "sdb_fe_query(%s) returned data on error: '%s'",
 556                         query_data[_i].query, data);
 557                 mock_conn_destroy(conn);
 558                 return;
 559         }
 560
 561         tmp = sdb_proto_unmarshal_header(data, len, &code, &msg_len);
 562         ck_assert_msg(tmp == (ssize_t)(2 * sizeof(uint32_t)));
 563         data += tmp;
 564         len -= tmp;
 565
 566         fail_unless((code == query_data[_i].code)
 567                                 && ((size_t)msg_len == query_data[_i].len),
 568                         "sdb_fe_query(%s) returned %u, %u; expected: %u, %zu",
 569                         query_data[_i].query, code, msg_len,
 570                         query_data[_i].code, query_data[_i].len);
 571
 572         if (code == SDB_CONNECTION_DATA) {
 573                 tmp = sdb_proto_unmarshal_int32(data, len, &code);
 574                 fail_unless(code == query_data[_i].type,
 575                                 "sdb_fe_query(%s) returned %s object; expected: %s",
 576                                 query_data[_i].query, SDB_CONN_MSGTYPE_TO_STRING((int)code),
 577                                 SDB_CONN_MSGTYPE_TO_STRING((int)query_data[_i].type));
 578                 data += tmp;
 579                 len -= tmp;
 580         }
 581
 582         fail_if_strneq(data, query_data[_i].data, (size_t)msg_len,
 583                         "sdb_fe_query(%s) returned unexpected data",
 584                         query_data[_i].query, data, query_data[_i].data);
 585
 586         mock_conn_destroy(conn);
 587 }
 588 END_TEST
 589
 590 TEST_MAIN("frontend::query")
 591 {
 592         TCase *tc = tcase_create("core");
 593         tcase_add_checked_fixture(tc, populate, sdb_store_clear);
 594         TC_ADD_LOOP_TEST(tc, query);
 595         ADD_TCASE(tc);
 596 }
 597 TEST_MAIN_END
 598
 599 /* vim: set tw=78 sw=4 ts=4 noexpandtab : */
 600