mirrors/core-lightning
1
0
Fork 0
mirror of https://github.com/ElementsProject/lightning.git synced 2025-02-23 06:55:13 +01:00
Code Issues Projects Releases Wiki Activity

pytest: test hooks and notifications with libplugin

Browse source
This commit is contained in:
darosior 2020-02-01 16:00:32 +01:00 • committed by Rusty Russell
parent ceeb5503cc
commit 75becf7227
2 changed files with 50 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

45
tests/plugins/test_libplugin.c
View file

@ -22,6 +22,37 @@ static struct command_result *json_helloworld(struct command *cmd,
return command_success_str(cmd, tal_fmt(tmpctx, "hello %s", name)); return command_success_str(cmd, tal_fmt(tmpctx, "hello %s", name));
} }
static struct command_result *
json_peer_connected(struct command *cmd,
const char *buf,
const jsmntok_t *params)
{
const jsmntok_t *peertok, *idtok;
struct json_stream *response;
peertok = json_get_member(buf, params, "peer");
assert(peertok);
idtok = json_get_member(buf, peertok, "id");
assert(idtok);
plugin_log(cmd->plugin, LOG_INFORM, "%s peer_connected",
json_strdup(tmpctx, buf, idtok));
response = jsonrpc_stream_success(cmd);
json_add_string(response, "result", "continue");
return command_finished(cmd, response);
}
static void json_connected(struct command *cmd,
const char *buf,
const jsmntok_t *params)
{
const jsmntok_t *idtok = json_get_member(buf, params, "id");
assert(idtok);
plugin_log(cmd->plugin, LOG_INFORM, "%s connected",
json_strdup(tmpctx, buf, idtok));
}
static void init(struct plugin *p, static void init(struct plugin *p,
const char *buf UNUSED, const jsmntok_t *config UNUSED) const char *buf UNUSED, const jsmntok_t *config UNUSED)
{ {
@ -39,11 +70,23 @@ static const struct plugin_command commands[] = { {
} }
}; };
static const struct plugin_hook hooks[] = { {
"peer_connected",
json_peer_connected,
}
};
static const struct plugin_notification notifs[] = { {
"connect",
json_connected,
}
};
int main(int argc, char *argv[]) int main(int argc, char *argv[])
{ {
setup_locale(); setup_locale();
plugin_main(argv, init, PLUGIN_RESTARTABLE, commands, ARRAY_SIZE(commands), plugin_main(argv, init, PLUGIN_RESTARTABLE, commands, ARRAY_SIZE(commands),
NULL, 0, NULL, 0, notifs, ARRAY_SIZE(notifs), hooks, ARRAY_SIZE(hooks),
plugin_option("name", plugin_option("name",
"string", "string",
"Who to say hello to.", "Who to say hello to.",

6
tests/test_plugin.py
View file

@ -813,3 +813,9 @@ def test_libplugin(node_factory):
assert l1.rpc.call("helloworld") == "hello test_opt" assert l1.rpc.call("helloworld") == "hello test_opt"
# But param takes over! # But param takes over!
assert l1.rpc.call("helloworld", {"name": "test"}) == "hello test" assert l1.rpc.call("helloworld", {"name": "test"}) == "hello test"
# Test hooks and notifications
l2 = node_factory.get_node()
l2.connect(l1)
assert l1.daemon.is_in_log("{} peer_connected".format(l2.info["id"]))
l1.daemon.wait_for_log("{} connected".format(l2.info["id"]))