Function test_json_stream

lightningd/test/run-jsonrpc.c:233–283

Test that we can segment and parse a stream of json objects correctly. */

/* Test that we can segment and parse a stream of json objects correctly. */
static void test_json_stream(void)
{
	bool valid, complete;
	char *input, *talstr;
	jsmntok_t *toks;
	jsmn_parser parser;

	/* Multiple full messages in a single buffer (happens when buffer
	 * boundary coincides with message boundary, or read returned after
	 * timeout. */
	input = "{\"x\":\"x\"}{\"y\":\"y\"}";
	talstr = tal_strndup(NULL, input, strlen(input));
	toks = toks_alloc(NULL);
	jsmn_init(&parser);
	valid = json_parse_input(&parser, &toks, talstr, strlen(talstr), &complete);
	assert(tal_count(toks) == 4);
	assert(toks[0].start == 0 && toks[0].end == 9);
	assert(valid);
	assert(complete);
	tal_free(talstr);

	/* Multiple messages, and the last one is partial, far more likely than
	 * accidentally getting the boundaries to match. */
	input = "{\"x\":\"x\"}{\"y\":\"y\"}{\"z\":\"z";
	talstr = tal_strndup(NULL, input, strlen(input));
	toks_reset(toks);
	jsmn_init(&parser);
	valid = json_parse_input(&parser, &toks, talstr, strlen(talstr), &complete);
	assert(toks);
	assert(tal_count(toks) == 4);
	assert(toks[0].start == 0 && toks[0].end == 9);
	assert(valid);
	assert(complete);
	tal_free(talstr);

	/* We can do this incrementally, too. */
	toks_reset(toks);
	input = "{\"x\":\"x\"}";
	jsmn_init(&parser);
	for (size_t i = 0; i <= strlen(input); i++) {
		valid = json_parse_input(&parser, &toks, input, i, &complete);
		assert(valid);
		if (i == strlen(input))
			assert(complete);
		else
			assert(!complete);
	}
	assert(tal_count(toks) == 4);
	assert(toks[0].start == 0 && toks[0].end == 9);
	tal_free(toks);
}

int main(int argc, char *argv[])
{