tsoding.frozen/unit_test.c

/*
 * Copyright (c) 2004-2013 Sergey Lyubka <[email protected]>
 * Copyright (c) 2013 Cesanta Software Limited
 * All rights reserved
 *
 * This library is dual-licensed: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation. For the terms of this
 * license, see <http: *www.gnu.org/licenses/>.
 *
 * You are free to use this library under the terms of the GNU General
 * Public License, but WITHOUT ANY WARRANTY; without even the implied
 * warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU General Public License for more details.
 *
 * Alternatively, you can license this library under a commercial
 * license, as set out in <http://cesanta.com/products.html>.
 */

/*
 * To unit test on Mac system, do
 *
 * g++ unit_test.c -o unit_test -W -Wall -g -O0 -fprofile-arcs -ftest-coverage
 * clang unit_test.c -o unit_test -W -Wall -g -O0 -fprofile-arcs -ftest-coverage
 * ./unit_test
 * gcov -a unit_test.c
 */

#include "frozen.c"

#include <float.h>
#include <math.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

const char *tok_type_names[] = {
    "INVALID", "STRING",       "NUMBER",     "TRUE",        "FALSE",
    "NULL",    "OBJECT_START", "OBJECT_END", "ARRAY_START", "ARRAY_END",
};

#define FAIL(str, line)                                    \
  do {                                                     \
    fprintf(stderr, "Fail on line %d: [%s]\n", line, str); \
    return str;                                            \
  } while (0)

#define ASSERT(expr)                    \
  do {                                  \
    static_num_tests++;                 \
    if (!(expr)) FAIL(#expr, __LINE__); \
  } while (0)

#define ARRAY_SIZE(x) (sizeof(x) / sizeof((x)[0]))
#define RUN_TEST(test)        \
  do {                        \
    const char *msg = test(); \
    if (msg) return msg;      \
  } while (0)

static int static_num_tests = 0;

static const char *test_errors(void) {
  /* clang-format off */
  static const char *invalid_tests[] = {
      "p",        "a:3",           "\x01",         "{:",
      " { 1",     "{a:\"\n\"}",    "{a:1x}",       "{a:1e}",
      "{a:.1}",   "{a:0.}",        "{a:0.e}",      "{a:0.e1}",
      "{a:0.1e}", "{a:\"\\u\" } ", "{a:\"\\yx\"}", "{a:\"\\u111r\"}",
      NULL};
  static const char *incomplete_tests[] = {"",
                                           " \r\n\t",
                                           "{",
                                           " { a",
                                           "{a:",
                                           "{a:\"",
                                           " { a : \"xx",
                                           "{a:12",
                                           "{a:\"\\uf",
                                           "{a:\"\\uff",
                                           "{a:\"\\ufff",
                                           "{a:\"\\uffff",
                                           "{a:\"\\uffff\"",
                                           "{a:\"\\uffff\" ,",
                                           "{a:n",
                                           "{a:nu",
                                           "{a:nul",
                                           "{a:null",
                                           NULL};
  /* clang-format on */
  static const struct {
    const char *str;
    int expected_len;
  } success_tests[] = {{"{}", 2},
                       /* 2, 3, 4 byte utf-8 chars */
                       {"{a:\"\xd0\xb1\xe3\x81\xaf\xf0\xa2\xb3\x82\"}", 15},
                       {"{a:\"\\u0006\"}", 12},
                       {" { } ", 4},
                       {"{a:1}", 5},
                       {"{a:1.23}", 8},
                       {"{a:1e23}", 8},
                       {"{a:1.23e2}", 10},
                       {"{a:-123}", 8},
                       {"{a:-1.3}", 8},
                       {"{a:-1.3e-2}", 11},
                       {"{a:\"\"}", 6},
                       {"{a:\" \\n\\t\\r\"}", 13},
                       {" {a:[1]} 123456", 8},
                       {" {a:[]} 123456", 7},
                       {" {a:[1,2]} 123456", 10},
                       {"{a:1,b:2} xxxx", 9},
                       {"{a:1,b:{},c:[{}]} xxxx", 17},
                       {"{a:true,b:[false,null]} xxxx", 23},
                       {"[1.23, 3, 5]", 12},
                       {"[13, {\"a\":\"hi there\"}, 5]", 25},
                       {NULL, 0}};
  const char *s1 =
      " { a: 1, b: \"hi there\", c: true, d: false, "
      " e : null, f: [ 1, -2, 3], g: { \"1\": [], h: [ 7 ] } } ";
  int i;

  ASSERT(json_walk(NULL, 0, NULL, 0) == JSON_STRING_INVALID);
  for (i = 0; invalid_tests[i] != NULL; i++) {
    ASSERT(json_walk(invalid_tests[i], strlen(invalid_tests[i]), NULL, NULL) ==
           JSON_STRING_INVALID);
  }

  for (i = 0; incomplete_tests[i] != NULL; i++) {
    ASSERT(json_walk(incomplete_tests[i], strlen(incomplete_tests[i]), NULL,
                     NULL) == JSON_STRING_INCOMPLETE);
  }

  for (i = 0; success_tests[i].str != NULL; i++) {
    ASSERT(json_walk(success_tests[i].str, strlen(success_tests[i].str), NULL,
                     NULL) == success_tests[i].expected_len);
  }

  ASSERT(json_walk("{}", 2, NULL, NULL) == 2);
  ASSERT(json_walk(s1, strlen(s1), NULL, 0) > 0);

  return NULL;
}

struct my_struct {
  int a, b;
};

static int print_my_struct(struct json_out *out, va_list *ap) {
  struct my_struct *p = va_arg(*ap, struct my_struct *);
  return json_printf(out, "{a: %d, b: %d}", p->a, p->b);
}

static const char *test_json_printf(void) {
  char buf[200] = "";

  {
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    const char *result = "42 42";
    json_printf(&out, "%ld %d", 42, 42);
    ASSERT(strcmp(buf, result) == 0);
  }

  /* platform specific compatibility where it matters */
  {
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    const char *result = "16045690985373621933 42";
    json_printf(&out, "%" UINT64_FMT " %d", 0xdeadbeeffee1deadUL, 42);
    ASSERT(strcmp(buf, result) == 0);
  }
  {
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    const char *result = "12 42";
    size_t foo = 12;
    json_printf(&out, "%lu %d", foo, 42);
    ASSERT(strcmp(buf, result) == 0);
  }

  /* people live in the future today, %llu works even on recent windozes */
  {
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    const char *result = "16045690985373621933 42";
    json_printf(&out, "%llu %d", 0xdeadbeeffee1deadUL, 42);
    ASSERT(strcmp(buf, result) == 0);
  }

  {
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    const char *result = "12 42";
    size_t foo = 12;
    json_printf(&out, "%zu %d", foo, 42);
    ASSERT(strcmp(buf, result) == 0);
  }

  {
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    const char *result = "{\"foo\": 123, \"x\": [false, true], \"y\": \"hi\"}";
    json_printf(&out, "{%Q: %d, x: [%B, %B], y: %Q}", "foo", 123, 0, -1, "hi");
    ASSERT(strcmp(buf, result) == 0);
  }

  {
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    int arr[] = {-2387, 943478};
    json_printf(&out, "%M", json_printf_array, arr, sizeof(arr), sizeof(arr[0]),
                "%d");
    ASSERT(strcmp(buf, "[-2387, 943478]") == 0);
  }

  {
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    double arr[] = {9.32156, 3.1415926};
    json_printf(&out, "%M", json_printf_array, arr, sizeof(arr), sizeof(arr[0]),
                "%.2lf");
    ASSERT(strcmp(buf, "[9.32, 3.14]") == 0);
  }

  {
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    unsigned short arr[] = {65535, 777};
    const char *result = "{\"a\": [-1, 777], \"b\": 37}";
    json_printf(&out, "{a: %M, b: %d}", json_printf_array, arr, sizeof(arr),
                sizeof(arr[0]), "%hd", 37);
    ASSERT(strcmp(buf, result) == 0);
  }

  {
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    const char *arr[] = {"hi", "there", NULL};
    const char *result = "[\"hi\", \"there\", null]";
    json_printf(&out, "%M", json_printf_array, arr, sizeof(arr), sizeof(arr[0]),
                "%Q");
    ASSERT(strcmp(buf, result) == 0);
  }

  {
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    const char *result = "{\"a\": \"\\\"\\\\\\r\\nя\\t\\u0002\"}";
    json_printf(&out, "{a: %Q}", "\"\\\r\nя\t\x02");
    ASSERT(strcmp(buf, result) == 0);
  }

  {
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    struct my_struct mys = {1, 2};
    const char *result = "{\"foo\": {\"a\": 1, \"b\": 2}, \"bar\": 3}";
    json_printf(&out, "{foo: %M, bar: %d}", print_my_struct, &mys, 3);
    ASSERT(strcmp(buf, result) == 0);
  }

  {
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    out.u.buf.size = 3;
    memset(buf, 0, sizeof(buf));
    ASSERT(json_printf(&out, "{%d}", 123) == 5);
    ASSERT(memcmp(buf, "{1\x00\x00\x00", 5) == 0);
  }

  {
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    const char *result = "\"foo\"";
    out.u.buf.size = 6;
    memset(buf, 0, sizeof(buf));
    ASSERT(json_printf(&out, "%.*Q", 3, "foobar") == 5);
    ASSERT(memcmp(buf, result, 5) == 0);
  }

  {
    /*
     * Check long string (which forces frozen to allocate a temporary buffer
     * from heap)
     */
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    const char *result =
        "{\"foo\": "
        "\"12345678901234567890123456789012345678901234567890123456789012345678"
        "90123456789012345678901234567890\"}";
    const char *s =
        "\"123456789012345678901234567890123456789012345678901234567890"
        "1234567890123456789012345678901234567890\"";
    json_printf(&out, "{foo: %s}", s);
    ASSERT(strcmp(buf, result) == 0);
  }

  {
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    const char *fmt = "{a: \"%s\"}", *result = "{\"a\": \"b\"}";
    memset(buf, 0, sizeof(buf));
    ASSERT(json_printf(&out, fmt, "b") > 0);
    ASSERT(strcmp(buf, result) == 0);
  }

  {
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    memset(buf, 0, sizeof(buf));
    ASSERT(json_printf(&out, "%.*s %d", 2, "abc", 5) > 0);
    ASSERT(strcmp(buf, "ab 5") == 0);
  }

  {
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    const char *result = "\"a_b0\": 1";
    memset(buf, 0, sizeof(buf));
    ASSERT(json_printf(&out, "a_b0: %d", 1) > 0);
    ASSERT(strcmp(buf, result) == 0);
  }
#if JSON_ENABLE_BASE64
  {
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    const char *result = "\"YTI=\"";
    memset(buf, 0, sizeof(buf));
    ASSERT(json_printf(&out, "%V", "a2", 2) > 0);
    ASSERT(strcmp(buf, result) == 0);
  }

  {
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    const char *result = "\"ACABIAIgYWJj\"";
    memset(buf, 0, sizeof(buf));
    ASSERT(json_printf(&out, "%V", "\x00 \x01 \x02 abc", 9) > 0);
    ASSERT(strcmp(buf, result) == 0);
  }
#endif /* JSON_ENABLE_BASE64 */
#if JSON_ENABLE_HEX
  {
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    const char *result = "\"002001200220616263\"";
    memset(buf, 0, sizeof(buf));
    ASSERT(json_printf(&out, "%H", 9, "\x00 \x01 \x02 abc") > 0);
    ASSERT(strcmp(buf, result) == 0);
  }
#endif /* JSON_ENABLE_HEX */
  {
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    memset(buf, 0, sizeof(buf));
    ASSERT(json_printf(&out, "%c", 0x53) > 0);
    ASSERT(strcmp(buf, "S") == 0);
  }

  {
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    const char *result = "<\"array\">0f";
    memset(buf, 0, sizeof(buf));
    ASSERT(json_printf(&out, "<array>%02x", 15) > 0);
    ASSERT(strcmp(buf, result) == 0);
  }

  {
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    double arr[] = {9.32156, 3.1415926};
#if !defined(_MSC_VER) || _MSC_VER >= 1700
    const char *result = "[9.32e+00, 3.14e+00]";  // Modern compilers
#else
    const char *result = "[9.32e+000, 3.14e+000]";  // Old VC98 compiler
#endif
    json_printf(&out, "%M", json_printf_array, arr, sizeof(arr), sizeof(arr[0]),
                "%.2e");
    ASSERT(strcmp(buf, result) == 0);
  }
  {
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    double arr[] = {9.32156, 3.1415926};
    json_printf(&out, "%M", json_printf_array, arr, sizeof(arr), sizeof(arr[0]),
                "%.4g");
    ASSERT(strcmp(buf, "[9.322, 3.142]") == 0);
  }

  return NULL;
}

static const char *test_system(void) {
  char buf[2020];
  uint64_t u = (uint64_t) 0xdeadbeeffee1dead;
  int64_t d = (int64_t) u;
  int res;

  snprintf(buf, sizeof(buf), "%" UINT64_FMT, u);
  ASSERT(strcmp(buf, "16045690985373621933") == 0);

  snprintf(buf, sizeof(buf), "%" INT64_FMT, d);
  ASSERT(strcmp(buf, "-2401053088335929683") == 0);

  res = snprintf(buf, 3, "foo");
  ASSERT(res == 3);
  ASSERT(buf[0] == 'f');
  ASSERT(buf[1] == 'o');
  ASSERT(buf[2] == '\0');

  return NULL;
}

static void cb(void *data, const char *name, size_t name_len, const char *path,
               const struct json_token *token) {
  char *buf = (char *) data;

  const char *snull = "<null>";

  sprintf(buf + strlen(buf), "name:'%.*s', path:'%s', type:%s, val:'%.*s'\n",
          (int) (name != NULL ? name_len : strlen(snull)),
          name != NULL ? name : snull, path, tok_type_names[token->type],
          (int) (token->ptr != NULL ? token->len : (int) strlen(snull)),
          token->ptr != NULL ? token->ptr : snull);
}

static const char *test_callback_api(void) {
  const char *s =
      "{\"c\":[\"foo\", \"bar\", {\"a\":9, \"b\": \"x\"}], "
      "\"mynull\": null, \"mytrue\": true, \"myfalse\": false}";

  const char *result =
      "name:'<null>', path:'', type:OBJECT_START, val:'<null>'\n"
      "name:'c', path:'.c', type:ARRAY_START, val:'<null>'\n"
      "name:'0', path:'.c[0]', type:STRING, val:'foo'\n"
      "name:'1', path:'.c[1]', type:STRING, val:'bar'\n"
      "name:'2', path:'.c[2]', type:OBJECT_START, val:'<null>'\n"
      "name:'a', path:'.c[2].a', type:NUMBER, val:'9'\n"
      "name:'b', path:'.c[2].b', type:STRING, val:'x'\n"
      "name:'<null>', path:'.c[2]', type:OBJECT_END, val:'{\"a\":9, \"b\": "
      "\"x\"}'\n"
      "name:'<null>', path:'.c', type:ARRAY_END, val:'[\"foo\", \"bar\", "
      "{\"a\":9, \"b\": \"x\"}]'\n"
      "name:'mynull', path:'.mynull', type:NULL, val:'null'\n"
      "name:'mytrue', path:'.mytrue', type:TRUE, val:'true'\n"
      "name:'myfalse', path:'.myfalse', type:FALSE, val:'false'\n"
      "name:'<null>', path:'', type:OBJECT_END, val:'{\"c\":[\"foo\", \"bar\", "
      "{\"a\":9, \"b\": \"x\"}], \"mynull\": null, \"mytrue\": true, "
      "\"myfalse\": false}'\n";

  char buf[4096] = "";
  ASSERT(json_walk(s, strlen(s), cb, buf) == (int) strlen(s));
  ASSERT(strcmp(buf, result) == 0);
  return NULL;
}

/*
 * Tests with the path which is longer than JSON_MAX_PATH_LEN (at the moment,
 * 60)
 */
static const char *test_callback_api_long_path(void) {
  const char *s =
      "{\"MyWZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZ"
      "ZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZ"
      "ZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZ"
      "ZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZ"
      "ZZZZZZZZZZZZZZZZZvf\": {}, \"jYP-27917287424p\": {}}";

  const char *result =
      "name:'<null>', path:'', type:OBJECT_START, val:'<null>'\n"
      "name:'"
      "MyWZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZ"
      "ZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZ"
      "ZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZ"
      "ZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZvf', "
      "path:'."
      "MyWZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZ"
      "ZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZ"
      "ZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZ"
      "ZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZ', "
      "type:OBJECT_START, val:'<null>'\n"
      "name:'<null>', "
      "path:'."
      "MyWZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZ"
      "ZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZ"
      "ZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZ"
      "ZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZ', "
      "type:OBJECT_END, val:'{}'\n"
      "name:'jYP-27917287424p', path:'.jYP-27917287424p', type:OBJECT_START, "
      "val:'<null>'\n"
      "name:'<null>', path:'.jYP-27917287424p', type:OBJECT_END, val:'{}'\n"
      "name:'<null>', path:'', type:OBJECT_END, "
      "val:'{"
      "\"MyWZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZ"
      "ZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZ"
      "ZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZ"
      "ZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZ"
      "ZZZZZZZZZZZZZZZZvf\": {}, \"jYP-27917287424p\": {}}'\n";

  char buf[4096] = "";
  ASSERT(json_walk(s, strlen(s), cb, buf) == (int) strlen(s));
  ASSERT(strcmp(buf, result) == 0);
  return NULL;
}

static void scan_array(const char *str, int len, void *user_data) {
  struct json_token t;
  int i;
  char *buf = (char *) user_data;
  for (i = 0; json_scanf_array_elem(str, len, ".x", i, &t) > 0; i++) {
    sprintf(buf + strlen(buf), "%d[%.*s] ", i, t.len, t.ptr);
  }
}

static const char *test_scanf(void) {
  char buf[100] = "";
  int a = 0, b = 0;
  char *d = NULL;
  const char *str =
      "{ a: 1234, b : true, \"c\": {x: [17, 78, -20]}, d: \"hi%20there\" }";

  ASSERT(json_scanf(str, strlen(str), "{a: %d, b: %B, c: [%M], d: %Q}", &a, &b,
                    &scan_array, buf, &d) == 4);
  ASSERT(a == 1234);
  ASSERT(b == 1);
  ASSERT(strcmp(buf, "0[17] 1[78] 2[-20] ") == 0);
  ASSERT(d != NULL);
  ASSERT(strcmp(d, "hi%20there") == 0);
  free(d);

  {
    /* Test errors */
    const char *str = "{foo:1, bar:[2,3,4]}";
    size_t i;
    ASSERT(json_walk(str, strlen(str), NULL, NULL) == (int) strlen(str));
    for (i = 1; i < strlen(str); i++) {
      ASSERT(json_walk(str, i, NULL, NULL) == JSON_STRING_INCOMPLETE);
    }
  }

  {
    /* Test that paths are utf8 */
    const char *str = "{\"ы\": 123}";
    int x = 0;
    ASSERT(json_scanf(str, strlen(str), "{ы: %d}", &x) == 1);
    ASSERT(x == 123);
  }

  {
    /* Test that paths are utf8 */
    const char *str = "{a: 123, b: [1,2,3]}";
    struct json_token t;
    memset(&t, 0, sizeof(t));
    ASSERT(json_scanf(str, strlen(str), "{b: %T}", &t) == 1);
    ASSERT(t.type == JSON_TYPE_ARRAY_END);
    ASSERT(t.len == 7);
    ASSERT(strncmp(t.ptr, "[1,2,3]", t.len) == 0);
  }

  {
    /* Test zero termination */
    char *s = NULL;
    const char *str = "{a: \"foo\", b:123}";
    ASSERT(json_scanf(str, strlen(str), "{a: %Q}", &s) == 1);
    ASSERT(s != NULL);
    ASSERT(s[3] == '\0');
    free(s);
  }

  {
    /* Test for scalar value being a root element */
    int n = 0;
    const char *str = " true ";
    ASSERT(json_scanf(str, strlen(str), " %B ", &n) == 1);
    ASSERT(n == 1);
  }

  {
    /* Test array of objects */
    const char *str = " { \"a\": [ {\"b\": 123}, {\"b\": 345} ]} ";
    int i, value, len = strlen(str), values[] = {123, 345};
    struct json_token t;

    /* Scan each array element into a token */
    for (i = 0; json_scanf_array_elem(str, len, ".a", i, &t) > 0; i++) {
      /* Now scan each token */
      ASSERT(t.type == JSON_TYPE_OBJECT_END);
      ASSERT(json_scanf(t.ptr, t.len, "{b: %d}", &value) == 1);
      ASSERT((size_t) i < sizeof(values) / sizeof(values[0]));
      ASSERT(values[i] == value);
    }
    ASSERT(i == 2);
  }

  {
    const char *str = "{a : [\"foo\", \"\", \"a\"] }";
    struct json_token t;
    ASSERT(json_scanf_array_elem(str, strlen(str), ".a", 0, &t) == 3);
    ASSERT(json_scanf_array_elem(str, strlen(str), ".a", 1, &t) == 0);
    ASSERT(json_scanf_array_elem(str, strlen(str), ".a", 2, &t) == 1);
    ASSERT(json_scanf_array_elem(str, strlen(str), ".a", 3, &t) == -1);
  }

  {
    const char *str = "{a : \"foo\\b\\f\\n\\r\\t\\\\\" }";
    char *result;
    ASSERT(json_scanf(str, strlen(str), "{a: %Q}", &result) == 1);
    ASSERT(strcmp(result, "foo\b\f\n\r\t\\") == 0);
    free(result);

    ASSERT(json_scanf(str, 9, "{a: %Q}", &result) == 0);
  }

  {
    const char *str = "{a : \"привет\" }";
    char *result;
    ASSERT(json_scanf(str, strlen(str), "{a: %Q}", &result) == 1);
    ASSERT(strcmp(result, "привет") == 0);
    free(result);
  }
#if JSON_ENABLE_BASE64
  {
    const char *str = "{a : \"YTI=\" }";
    int len;
    char *result;
    ASSERT(json_scanf(str, strlen(str), "{a: %V}", &result, &len) == 1);
    ASSERT(len == 2);
    ASSERT(strcmp(result, "a2") == 0);
    free(result);
  }

  {
    const char *str = "{a : \"0L/RgNC40LLQtdGC0Ys=\" }";
    int len;
    char *result;
    ASSERT(json_scanf(str, strlen(str), "{a: %V}", &result, &len) == 1);
    ASSERT(len == 14);
    ASSERT(strcmp(result, "приветы") == 0);
    free(result);
  }
#endif /* JSON_ENABLE_BASE64 */
#if JSON_ENABLE_HEX
  {
    const char *str = "{a : \"61626320\" }";
    int len = 0;
    char *result = NULL;
    ASSERT(json_scanf(str, strlen(str), "{a: %H}", &len, &result) == 1);
    ASSERT(len == 4);
    ASSERT(strcmp(result, "abc ") == 0);
    free(result);
  }
#endif /* JSON_ENABLE_HEX */
  {
    const char *str = "{a : null }";
    char *result = (char *) 123;
    ASSERT(json_scanf(str, strlen(str), "{a: %Q}", &result) == 0);
    ASSERT(result == NULL);
    free(result);
  }

  {
    int a = 0;
    bool b = false;
    int c = 0xFFFFFFFF;
    const char *str = "{\"b\":true,\"c\":false,\"a\":2}";
    ASSERT(json_scanf(str, strlen(str), "{a:%d, b:%B, c:%B}", &a, &b, &c) == 3);
    ASSERT(a == 2);
    ASSERT(b == true);
    if (sizeof(bool) == 1) {
      ASSERT((char) c == false);
    } else {
      ASSERT(c == false);
    }
  }

  {
    const char *str = "{ fa: 1, fb: 2.34, fc: 5.67 }";
    const char *fmt = "{fa: %f, fb: %f, fc: %lf}";
    float fa = 0.0, fb = 0.0;
    double fc = 0.0;
#if !JSON_MINIMAL
    float a = 1.0f, b = 2.34f;
    double c = 5.67;
    ASSERT(json_scanf(str, strlen(str), fmt, &fa, &fb, &fc) == 3);
    ASSERT(fa == a);
    ASSERT(fb == b);
    ASSERT(fabs(fc - c) < FLT_EPSILON);
#else
    ASSERT(json_scanf(str, strlen(str), fmt, &fa, &fb, &fc) == 0);
#endif
  }

  {
    int v = -1;
    long lv = -1;
    const char *s = "{\"v\": 0x12, \"lv\": 0x34}";
    ASSERT(json_scanf("0x", 2, "%i", &v) == 0);  // Incomplete string
    ASSERT(json_scanf("0xe", 3, "%i", &v) == 1);
    ASSERT(v == 0xe);
    ASSERT(json_scanf("12", 2, "%i", &v) == 1);
    ASSERT(v == 12);
    // %d and %ld accept hex
    ASSERT(json_scanf(s, strlen(s), "{lv:%ld, v:%d}", &lv, &v) == 2);
    ASSERT(v == 0x12);
    ASSERT(lv == 0x34);
  }

  {
    unsigned int v = 0;
    unsigned long lv = 0;
    const char *s = "{\"v\": 0x12, \"lv\": 0x34}";
    // %u and %lu accept hex
    ASSERT(json_scanf(s, strlen(s), "{lv:%lu, v:%u}", &lv, &v) == 2);
    ASSERT(v == 0x12);
    ASSERT(lv == 0x34);
  }

  return NULL;
}

static const char *test_json_unescape(void) {
  char buf[1];
  ASSERT(json_unescape("foo", 3, NULL, 0) == 3);
  ASSERT(json_unescape("foo\\", 4, NULL, 0) == JSON_STRING_INCOMPLETE);
  ASSERT(json_unescape("foo\\x", 5, NULL, 0) == JSON_STRING_INVALID);
  ASSERT(json_unescape("\\ueeee", 5, NULL, 0) == JSON_STRING_INCOMPLETE);
  ASSERT(json_unescape("\\ueeee", 6, NULL, 0) == JSON_STRING_INVALID);
  // Simple one-byte escapes should work
  ASSERT(json_unescape("\\u0026", 2, NULL, 0) == JSON_STRING_INCOMPLETE);
  ASSERT(json_unescape("\\u0026", 3, NULL, 0) == JSON_STRING_INCOMPLETE);
  ASSERT(json_unescape("\\u0026", 4, NULL, 0) == JSON_STRING_INCOMPLETE);
  ASSERT(json_unescape("\\u0026", 5, NULL, 0) == JSON_STRING_INCOMPLETE);
  ASSERT(json_unescape("\\u0026", 6, buf, sizeof(buf)) == 1);
  ASSERT(buf[0] == '&');
  return NULL;
}

static void cb2(void *data, const char *name, size_t name_len, const char *path,
                const struct json_token *token) {
  struct json_token *pt = (struct json_token *) data;
  pt->ptr = token->ptr;
  pt->len = token->len;
  (void) path;
  (void) name_len;
  (void) name;
}

static const char *test_parse_string(void) {
  const char *str = " \" foo\\bar\"";
  const int str_len = strlen(str);
  struct json_token t;
  struct frozen f;
  memset(&f, 0, sizeof(f));
  f.end = str + str_len;
  f.cur = str;
  f.callback_data = (void *) &t;
  f.callback = cb2;

  ASSERT(json_parse_string(&f) == 0);
  ASSERT(strncmp(t.ptr, " foo\\bar", t.len) == 0);

  return NULL;
}

static const char *test_eos(void) {
  const char *s = "{\"a\": 12345}";
  size_t n = 999;
  char *buf = (char *) malloc(n);
  int s_len = strlen(s), a = 0;
  ASSERT(buf != NULL);
  memset(buf, 'x', n);
  memcpy(buf, s, s_len);
  ASSERT(json_scanf(buf, n, "{a:%d}", &a) == 1);
  ASSERT(a == 12345);
  free(buf);
  return NULL;
}

static int compare_file(const char *file_name, const char *s) {
  int res = -1;
  char *p = json_fread(file_name);
  if (p == NULL) return res;
  res = strcmp(p, s);
  free(p);
  return res == 0;
}

static const char *test_fprintf(void) {
  const char *fname = "a.json";
  const char *result = "{\"a\":123}\n";
  char *p;
  ASSERT(json_fprintf(fname, "{a:%d}", 123) > 0);
  p = json_fread(fname);
  ASSERT(p != NULL);
  ASSERT(strcmp(p, result) == 0);
  free(p);
  remove(fname);
  ASSERT(json_fread(fname) == NULL);
  return NULL;
}

static const char *test_json_setf(void) {
  char buf[200];
  const char *s1 = "{ \"a\": 123, \"b\": [ 1 ], \"c\": true }";

  {
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    const char *s2 = "{ \"a\": 7, \"b\": [ 1 ], \"c\": true }";
    int res = json_setf(s1, strlen(s1), &out, ".a", "%d", 7);
    ASSERT(res == 1);
    ASSERT(strcmp(buf, s2) == 0);
  }

  {
    /* Add Key with length > 1 */
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    const char *s2 =
        "{ \"a\": 123, \"b\": [ 1 ], \"c\": true,\"foo\":{\"bar\":42} }";
    int res = json_setf(s1, strlen(s1), &out, ".foo.bar", "%d", 42);
    ASSERT(res == 0);
    ASSERT(strcmp(buf, s2) == 0);
  }

  {
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    const char *s2 = "{ \"a\": 123, \"b\": false, \"c\": true }";
    int res = json_setf(s1, strlen(s1), &out, ".b", "%B", 0);
    ASSERT(res == 1);
    ASSERT(strcmp(buf, s2) == 0);
  }

  {
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    const char *s2 = "{ \"a\": 123, \"b\": [ 2 ], \"c\": true }";
    int res = json_setf(s1, strlen(s1), &out, ".b[0]", "%d", 2);
    ASSERT(res == 1);
    ASSERT(strcmp(buf, s2) == 0);
  }

  {
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    const char *s2 = "{ \"b\": [ 1 ], \"c\": true }";
    int res = json_setf(s1, strlen(s1), &out, ".a", NULL);
    ASSERT(res == 1);
    ASSERT(strcmp(buf, s2) == 0);
  }

  {
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    const char *s2 = "{ \"a\": 123, \"b\": [ 1 ] }";
    int res = json_setf(s1, strlen(s1), &out, ".c", NULL);
    ASSERT(res == 1);
    ASSERT(strcmp(buf, s2) == 0);
  }

  {
    /* Delete non-existent key */
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    const char *s1 = "{\"a\":1}";
    int res = json_setf(s1, strlen(s1), &out, ".d", NULL);
    ASSERT(res == 0);
    ASSERT(strcmp(buf, s1) == 0);
  }

  {
    /* Delete non-existent key, spaces in obj */
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    int res = json_setf(s1, strlen(s1), &out, ".d", NULL);
    ASSERT(res == 0);
    ASSERT(strcmp(buf, s1) == 0);
  }

  {
    /* Change the whole JSON object */
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    const char *s2 = "123";
    int res = json_setf(s1, strlen(s1), &out, "", "%d", 123);
    ASSERT(res == 1);
    ASSERT(strcmp(buf, s2) == 0);
  }

  {
    /* Add missing keys */
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    const char *s2 =
        "{ \"a\": 123, \"b\": [ 1 ], \"c\": true,\"d\":{\"e\":8} }";
    int res = json_setf(s1, strlen(s1), &out, ".d.e", "%d", 8);
    ASSERT(res == 0);
    ASSERT(strcmp(buf, s2) == 0);
  }

  {
    /* Append to arrays */
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    const char *s2 = "{ \"a\": 123, \"b\": [ 1,2 ], \"c\": true }";
    int res = json_setf(s1, strlen(s1), &out, ".b[]", "%d", 2);
    ASSERT(res == 0);
    ASSERT(strcmp(buf, s2) == 0);
  }

  {
    /* Delete from array */
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    const char *s2 = "{ \"a\": 123, \"b\": [ ], \"c\": true }";
    int res = json_setf(s1, strlen(s1), &out, ".b[0]", NULL);
    ASSERT(res == 1);
    ASSERT(strcmp(buf, s2) == 0);
  }

  {
    /* Create array and push value  */
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    const char *s2 = "{ \"a\": 123, \"b\": [ 1 ], \"c\": true,\"d\":[3] }";
    int res = json_setf(s1, strlen(s1), &out, ".d[]", "%d", 3);
    // printf("[%s]\n[%s]\n", buf, s2);
    ASSERT(res == 0);
    ASSERT(strcmp(buf, s2) == 0);
  }

  return NULL;
}

static const char *test_prettify(void) {
  const char *fname = "a.json";
  char buf[200];

  {
    const char *s1 = "{ \"a\":   1, \"b\":2,\"c\":[null,\"aa\",{},true]}";
    struct json_out out = JSON_OUT_BUF(buf, sizeof(buf));
    const char *s2 =
        "{\n  \"a\": 1,\n  \"b\": 2,\n  \"c\": [\n    null,\n    \"aa\",\n    "
        "{},\n    true\n  ]\n}";
    ASSERT(json_prettify(s1, strlen(s1), &out) > 0);
    ASSERT(strcmp(buf, s2) == 0);
  }

  {
    remove(fname);
    ASSERT(json_prettify_file(fname) == -1);
  }

  {
    ASSERT(compare_file(fname, "") == -1);
    json_fprintf(fname, "::");
    ASSERT(json_prettify_file(fname) == JSON_STRING_INVALID);
    ASSERT(compare_file(fname, "::\n"));
    remove(fname);
  }

  {
    ASSERT(compare_file(fname, "") == -1);
    json_fprintf(fname, "{");
    ASSERT(json_prettify_file(fname) == JSON_STRING_INCOMPLETE);
    ASSERT(compare_file(fname, "{\n"));
    remove(fname);
  }

  {
    ASSERT(compare_file(fname, "") == -1);
    json_fprintf(fname, "%d", 123);
    ASSERT(compare_file(fname, "123\n"));
    ASSERT(json_prettify_file(fname) > 0);
    ASSERT(compare_file(fname, "123\n"));
    remove(fname);
  }

  {
    const char *s = "{\n  \"a\": 123\n}\n";
    ASSERT(compare_file(fname, "") == -1);
    json_fprintf(fname, "{a:%d}", 123);
    ASSERT(json_prettify_file(fname) > 0);
    ASSERT(compare_file(fname, s));
    (void) remove(fname);
  }

  return NULL;
}

static const char *test_json_next(void) {
  struct json_token key, val;
  char buf[100];

  {
    /* Traverse an object */
    void *h = NULL;
    int i = 0;
    const char *s = "{ \"a\": [], \"b\": [ 1, {} ], \"c\": true }";
    int len = strlen(s);
    const char *results[] = {"[a] -> [[]]", "[b] -> [[ 1, {} ]]",
                             "[c] -> [true]"};
    while ((h = json_next_key(s, len, h, "", &key, &val)) != NULL) {
      snprintf(buf, sizeof(buf), "[%.*s] -> [%.*s]", key.len, key.ptr, val.len,
               val.ptr);
      ASSERT(strcmp(results[i], buf) == 0);
      i++;
    }
    ASSERT(i == 3);
  }

  {
    /* Traverse an array */
    void *h = NULL;
    int i = 0, idx;
    const char *s = "{ \"a\": [], \"b\": [ 1, {} ], \"c\": true }";
    int len = strlen(s);
    const char *results[] = {"[0] -> [1]", "[1] -> [{}]"};
    while ((h = json_next_elem(s, len, h, ".b", &idx, &val)) != NULL) {
      snprintf(buf, sizeof(buf), "[%d] -> [%.*s]", idx, val.len, val.ptr);
      ASSERT(strcmp(results[i], buf) == 0);
      i++;
    }
    ASSERT(i == 2);
  }

  {
    /* Traverse more complex object */
    const char *s = "{ \"a\": 123, \"b\": { \"c\": true, \"d\": 1234 } }";
    void *h = NULL;
    int i = 0;
    int len = strlen(s);
    const char *results[] = {"[c] -> [true]", "[d] -> [1234]"};
    while ((h = json_next_key(s, len, h, ".b", &key, &val)) != NULL) {
      snprintf(buf, sizeof(buf), "[%.*s] -> [%.*s]", key.len, key.ptr, val.len,
               val.ptr);
      ASSERT(strcmp(results[i], buf) == 0);
      i++;
    }
    ASSERT(i == 2);
  }

  return NULL;
}

static const char *test_json_printf_hex(void) {
  char *s = json_asprintf("%H", 3, "abc");
#if JSON_ENABLE_HEX
  const char *r = "\"616263\"";
  ASSERT(strcmp(s, r) == 0);
#else
  ASSERT(s == NULL);
#endif
  free(s);
  return NULL;
}

static const char *test_json_printf_base64(void) {
  char *s = json_asprintf("{a:%d,b:%V}", 77, "hi", 2);
#if JSON_ENABLE_BASE64
  const char *r = "{\"a\":77,\"b\":\"aGk=\"}";
#else
  const char *r = "{\"a\":77,\"b\":}";
#endif
  ASSERT(strcmp(s, r) == 0);
  free(s);
  return NULL;
}

static const char *run_all_tests(void) {
  RUN_TEST(test_json_printf_hex);
  RUN_TEST(test_json_printf_base64);
  RUN_TEST(test_json_next);
  RUN_TEST(test_prettify);
  RUN_TEST(test_eos);
  RUN_TEST(test_scanf);
  RUN_TEST(test_errors);
  RUN_TEST(test_json_printf);
  RUN_TEST(test_system);
  RUN_TEST(test_callback_api);
  RUN_TEST(test_callback_api_long_path);
  RUN_TEST(test_json_unescape);
  RUN_TEST(test_parse_string);
  RUN_TEST(test_fprintf);
  RUN_TEST(test_json_setf);
  return NULL;
}

int main(void) {
  const char *fail_msg = run_all_tests();
  printf("%s, tests run: %d\n", fail_msg ? "FAIL" : "PASS", static_num_tests);
  return fail_msg == NULL ? EXIT_SUCCESS : EXIT_FAILURE;
}