From 5fa3a50896f004d9b8479743f7e2fd0c78b521fd Mon Sep 17 00:00:00 2001
From: trueptolemy <823220586@qq.com>
Date: Tue, 25 Jun 2019 16:48:11 +0800
Subject: [PATCH] pytest: Add a simple plugin to test 'sendpay_success' and
 'sendpay_failure'

---
 tests/plugins/sendpay_notifications.py | 34 ++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)
 create mode 100755 tests/plugins/sendpay_notifications.py

diff --git a/tests/plugins/sendpay_notifications.py b/tests/plugins/sendpay_notifications.py
new file mode 100755
index 000000000..35d3058fc
--- /dev/null
+++ b/tests/plugins/sendpay_notifications.py
@@ -0,0 +1,34 @@
+#!/usr/bin/env python3
+"""This plugin is used to check that sendpay_success and sendpay_failure calls are working correctly.
+"""
+from lightning import Plugin
+
+plugin = Plugin()
+
+
+@plugin.init()
+def init(configuration, options, plugin):
+    plugin.success_list = []
+    plugin.failure_list = []
+
+
+@plugin.subscribe("sendpay_success")
+def notify_sendpay_success(plugin, sendpay_success):
+    plugin.log("receive a sendpay_success recored, id: {}, payment_hash: {}".format(sendpay_success['id'], sendpay_success['payment_hash']))
+    plugin.success_list.append(sendpay_success)
+
+
+@plugin.subscribe("sendpay_failure")
+def notify_sendpay_failure(plugin, sendpay_failure):
+    plugin.log("receive a sendpay_failure recored, id: {}, payment_hash: {}".format(sendpay_failure['data']['id'],
+               sendpay_failure['data']['payment_hash']))
+    plugin.failure_list.append(sendpay_failure)
+
+
+@plugin.method('listsendpays_plugin')
+def record_lookup(plugin):
+    return {'sendpay_success': plugin.success_list,
+            'sendpay_failure': plugin.failure_list}
+
+
+plugin.run()