You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
231 lines
7.1 KiB
231 lines
7.1 KiB
#include "config.h"
|
|
#include <common/utils.h>
|
|
#include <common/json.c>
|
|
#include <wire/wire.h>
|
|
|
|
/* AUTOGENERATED MOCKS START */
|
|
/* Generated stub for amount_asset_is_main */
|
|
bool amount_asset_is_main(struct amount_asset *asset UNNEEDED)
|
|
{ fprintf(stderr, "amount_asset_is_main called!\n"); abort(); }
|
|
/* Generated stub for amount_asset_to_sat */
|
|
struct amount_sat amount_asset_to_sat(struct amount_asset *asset UNNEEDED)
|
|
{ fprintf(stderr, "amount_asset_to_sat called!\n"); abort(); }
|
|
/* Generated stub for amount_sat_add */
|
|
bool amount_sat_add(struct amount_sat *val UNNEEDED,
|
|
struct amount_sat a UNNEEDED,
|
|
struct amount_sat b UNNEEDED)
|
|
{ fprintf(stderr, "amount_sat_add called!\n"); abort(); }
|
|
/* Generated stub for amount_sat_eq */
|
|
bool amount_sat_eq(struct amount_sat a UNNEEDED, struct amount_sat b UNNEEDED)
|
|
{ fprintf(stderr, "amount_sat_eq called!\n"); abort(); }
|
|
/* Generated stub for amount_sat_sub */
|
|
bool amount_sat_sub(struct amount_sat *val UNNEEDED,
|
|
struct amount_sat a UNNEEDED,
|
|
struct amount_sat b UNNEEDED)
|
|
{ fprintf(stderr, "amount_sat_sub called!\n"); abort(); }
|
|
/* Generated stub for amount_sat_to_msat */
|
|
bool amount_sat_to_msat(struct amount_msat *msat UNNEEDED,
|
|
struct amount_sat sat UNNEEDED)
|
|
{ fprintf(stderr, "amount_sat_to_msat called!\n"); abort(); }
|
|
/* Generated stub for fmt_wireaddr_without_port */
|
|
char *fmt_wireaddr_without_port(const tal_t *ctx UNNEEDED, const struct wireaddr *a UNNEEDED)
|
|
{ fprintf(stderr, "fmt_wireaddr_without_port called!\n"); abort(); }
|
|
/* Generated stub for fromwire_fail */
|
|
const void *fromwire_fail(const u8 **cursor UNNEEDED, size_t *max UNNEEDED)
|
|
{ fprintf(stderr, "fromwire_fail called!\n"); abort(); }
|
|
/* Generated stub for json_add_member */
|
|
void json_add_member(struct json_stream *js UNNEEDED,
|
|
const char *fieldname UNNEEDED,
|
|
bool quote UNNEEDED,
|
|
const char *fmt UNNEEDED, ...)
|
|
{ fprintf(stderr, "json_add_member called!\n"); abort(); }
|
|
/* Generated stub for json_array_end */
|
|
void json_array_end(struct json_stream *js UNNEEDED)
|
|
{ fprintf(stderr, "json_array_end called!\n"); abort(); }
|
|
/* Generated stub for json_array_start */
|
|
void json_array_start(struct json_stream *js UNNEEDED, const char *fieldname UNNEEDED)
|
|
{ fprintf(stderr, "json_array_start called!\n"); abort(); }
|
|
/* Generated stub for json_member_direct */
|
|
char *json_member_direct(struct json_stream *js UNNEEDED,
|
|
const char *fieldname UNNEEDED, size_t extra UNNEEDED)
|
|
{ fprintf(stderr, "json_member_direct called!\n"); abort(); }
|
|
/* Generated stub for json_object_end */
|
|
void json_object_end(struct json_stream *js UNNEEDED)
|
|
{ fprintf(stderr, "json_object_end called!\n"); abort(); }
|
|
/* Generated stub for json_object_start */
|
|
void json_object_start(struct json_stream *ks UNNEEDED, const char *fieldname UNNEEDED)
|
|
{ fprintf(stderr, "json_object_start called!\n"); abort(); }
|
|
/* Generated stub for type_to_string_ */
|
|
const char *type_to_string_(const tal_t *ctx UNNEEDED, const char *typename UNNEEDED,
|
|
union printable_types u UNNEEDED)
|
|
{ fprintf(stderr, "type_to_string_ called!\n"); abort(); }
|
|
/* AUTOGENERATED MOCKS END */
|
|
|
|
struct json {
|
|
jsmntok_t *toks;
|
|
char *buffer;
|
|
};
|
|
|
|
static void convert_quotes(char *first)
|
|
{
|
|
while (*first != '\0') {
|
|
if (*first == '\'')
|
|
*first = '"';
|
|
first++;
|
|
}
|
|
}
|
|
|
|
static struct json *json_parse(const tal_t * ctx, const char *str)
|
|
{
|
|
struct json *j = tal(ctx, struct json);
|
|
j->buffer = tal_strdup(j, str);
|
|
convert_quotes(j->buffer);
|
|
bool ok;
|
|
j->toks = json_parse_input(j, j->buffer, strlen(j->buffer), &ok);
|
|
assert(ok);
|
|
j->toks = json_tok_copy(j, j->toks);
|
|
return j;
|
|
}
|
|
|
|
static void test_toks(const struct json *j, ...)
|
|
{
|
|
const jsmntok_t * tok = j->toks + 1;
|
|
const char *value;
|
|
va_list(ap);
|
|
va_start(ap, j);
|
|
while ((value = va_arg(ap, char *)) != NULL) {
|
|
if (tok->type == JSMN_OBJECT)
|
|
assert(streq(value, "{"));
|
|
else if (tok->type == JSMN_ARRAY)
|
|
assert(streq(value, "["));
|
|
else
|
|
assert(json_tok_streq(j->buffer, tok, value));
|
|
tok++;
|
|
}
|
|
}
|
|
|
|
static void sanity(void)
|
|
{
|
|
struct json *j = json_parse(tmpctx, "[]");
|
|
json_tok_remove(&j->toks, j->toks, j->toks, 0);
|
|
assert(j);
|
|
}
|
|
|
|
static void remove_one(void)
|
|
{
|
|
struct json *j = json_parse(tmpctx, "['invoice']");
|
|
json_tok_remove(&j->toks, j->toks, j->toks + 1, 1);
|
|
assert(j);
|
|
}
|
|
|
|
static void remove_first(void)
|
|
{
|
|
struct json *j = json_parse(tmpctx, "['one', 'two', 'three']");
|
|
assert(j);
|
|
json_tok_remove(&j->toks, j->toks, j->toks + 1, 1);
|
|
assert(j->toks);
|
|
|
|
test_toks(j, "two", "three", NULL);
|
|
assert(tal_count(j->toks) == 3);
|
|
|
|
j = json_parse(tmpctx, "{'1':'one', '2':'two', '3':'three'}");
|
|
assert(j);
|
|
|
|
json_tok_remove(&j->toks, j->toks, j->toks + 1, 1);
|
|
assert(j);
|
|
test_toks(j, "2", "two", "3", "three", NULL);
|
|
assert(tal_count(j->toks) == 5);
|
|
|
|
j = json_parse(tmpctx, "{'1':'one', '2':'two', '3':'three'}");
|
|
json_tok_remove(&j->toks, j->toks, j->toks + 1, 1);
|
|
assert(j);
|
|
test_toks(j, "2", "two", "3", "three", NULL);
|
|
assert(tal_count(j->toks) == 5);
|
|
|
|
}
|
|
|
|
static void remove_last(void)
|
|
{
|
|
struct json *j = json_parse(tmpctx, "['one', 'two', 'three']");
|
|
json_tok_remove(&j->toks, j->toks, j->toks + 3, 1);
|
|
test_toks(j, "one", "two", NULL);
|
|
assert(tal_count(j->toks) == 3);
|
|
|
|
j = json_parse(tmpctx, "{'1':'one', '2':'two', '3':'three'}");
|
|
json_tok_remove(&j->toks, j->toks, j->toks + 5, 1);
|
|
assert(j);
|
|
test_toks(j, "1", "one", "2", "two", NULL);
|
|
assert(tal_count(j->toks) == 5);
|
|
}
|
|
|
|
static void remove_multiple(void)
|
|
{
|
|
struct json *j = json_parse(tmpctx, "['a', 'b', 'c', 'd', 'e']");
|
|
json_tok_remove(&j->toks, j->toks, j->toks + 1, 2);
|
|
test_toks(j, "c", "d", "e", NULL);
|
|
|
|
j = json_parse(tmpctx, "['a', 'b', 'c', 'd', 'e']");
|
|
json_tok_remove(&j->toks, j->toks, j->toks + 2, 2);
|
|
test_toks(j, "a", "d", "e", NULL);
|
|
|
|
j = json_parse(tmpctx, "{'1':'one', '2':'two', '3':'three', '4':'four'}");
|
|
json_tok_remove(&j->toks, j->toks, j->toks + 3, 2);
|
|
assert(j);
|
|
test_toks(j, "1", "one", "4", "four", NULL);
|
|
assert(tal_count(j->toks) == 5);
|
|
}
|
|
|
|
static void remove_all(void)
|
|
{
|
|
struct json *j = json_parse(tmpctx, "['a', 'b', 'c', 'd', 'e']");
|
|
json_tok_remove(&j->toks, j->toks, j->toks + 1, 5);
|
|
assert(tal_count(j->toks) == 1);
|
|
|
|
j = json_parse(tmpctx, "{'1':'one', '2':'two', '3':'three', '4':'four'}");
|
|
json_tok_remove(&j->toks, j->toks, j->toks + 1, 4);
|
|
assert(tal_count(j->toks) == 1);
|
|
}
|
|
|
|
static void remove_complex(void)
|
|
{
|
|
struct json *j = json_parse(tmpctx,
|
|
"{'1':'one', '2':'two',"
|
|
"'3': { '3.1': 'a', '3.2':'b', '3.3':'c' }, "
|
|
"'4': { '4.1': 'a', '4.2':'b', '4.3':'c' }, "
|
|
"'5':'five'}");
|
|
|
|
json_tok_remove(&j->toks, j->toks, j->toks + 5, 2);
|
|
test_toks(j, "1", "one", "2", "two", "5", "five", NULL);
|
|
}
|
|
|
|
static void remove_inside_obj(void)
|
|
{
|
|
jsmntok_t *tok;
|
|
struct json *j = json_parse(tmpctx,
|
|
"{'1':'one', '2':'two',"
|
|
"'3': { '3.1': 'a', '3.2':'b', '3.3':'c' }, "
|
|
"'4':'four'}");
|
|
|
|
tok = (jsmntok_t *)json_get_member(j->buffer, j->toks, "3");
|
|
json_tok_remove(&j->toks, tok, tok+1, 1);
|
|
test_toks(j, "1", "one", "2", "two", "3", "{", "3.2", "b", "3.3", "c",
|
|
"4", "four", NULL);
|
|
}
|
|
|
|
int main(void)
|
|
{
|
|
setup_locale();
|
|
setup_tmpctx();
|
|
|
|
sanity();
|
|
remove_one();
|
|
remove_first();
|
|
remove_last();
|
|
remove_multiple();
|
|
remove_all();
|
|
remove_complex();
|
|
remove_inside_obj();
|
|
|
|
tal_free(tmpctx);
|
|
printf("run-json_remove ok\n");
|
|
}
|
|
|