From f55df28bcd04a286db020757bd0007c462feac2e Mon Sep 17 00:00:00 2001
From: Michael Schmoock <michael@schmoock.net>
Date: Mon, 1 Feb 2021 12:51:37 +0100
Subject: [PATCH] rebalance: adds a very basic rebalanceall test

---
 rebalance/test_rebalance.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/rebalance/test_rebalance.py b/rebalance/test_rebalance.py
index 5c8361c..31293c2 100644
--- a/rebalance/test_rebalance.py
+++ b/rebalance/test_rebalance.py
@@ -2,6 +2,7 @@ import os
 from pyln.testing.fixtures import *  # noqa: F401,F403
 
 plugin_path = os.path.join(os.path.dirname(__file__), "rebalance.py")
+plugin_opt = {'plugin': plugin_path}
 
 
 def test_rebalance_starts(node_factory):
@@ -14,3 +15,12 @@ def test_rebalance_starts(node_factory):
     # Then statically
     l1.daemon.opts["plugin"] = plugin_path
     l1.start()
+
+
+def test_rebalance_all(node_factory):
+    l1, l2 = node_factory.line_graph(2, opts=plugin_opt)
+
+    # for now just check we get an error if theres just one channel and we
+    # could not perform an auto rebalance
+    result = l1.rpc.rebalanceall()
+    assert result['message'] == 'Error: Not enough open channels to balance anything'