From e1709f20ebc1379a42fe554ee25fde8e1780a114 Mon Sep 17 00:00:00 2001
From: Elliott Jin <elliott.jin@gmail.com>
Date: Tue, 16 Feb 2021 01:03:04 -0800
Subject: [PATCH] netann: add tests for manually enabling/disabling channels

---
 netann/chan_status_manager_test.go | 51 ++++++++++++++++++++++++++++++
 1 file changed, 51 insertions(+)

diff --git a/netann/chan_status_manager_test.go b/netann/chan_status_manager_test.go
index 9f19bb225..e9ff61d62 100644
--- a/netann/chan_status_manager_test.go
+++ b/netann/chan_status_manager_test.go
@@ -799,6 +799,57 @@ var stateMachineTests = []stateMachineTest{
 			h.assertNoUpdates(h.safeDisableTimeout)
 		},
 	},
+	{
+		name:         "request manual enable",
+		startActive:  true,
+		startEnabled: false,
+		fn: func(h testHarness) {
+			// Request manual enables for all channels.
+			h.assertEnables(h.graph.chans(), nil, true)
+
+			// Expect to see them all enabled on the network.
+			h.assertUpdates(
+				h.graph.chans(), true, h.safeDisableTimeout,
+			)
+
+			// Subsequent request disables with manual = false should succeed.
+			h.assertDisables(
+				h.graph.chans(), nil, false,
+			)
+
+			// Expect to see them all disabled on the network again.
+			h.assertUpdates(
+				h.graph.chans(), false, h.safeDisableTimeout,
+			)
+		},
+	},
+	{
+		name:         "request manual disable",
+		startActive:  true,
+		startEnabled: true,
+		fn: func(h testHarness) {
+			// Request manual disables for all channels.
+			h.assertDisables(h.graph.chans(), nil, true)
+
+			// Expect to see them all disabled on the network.
+			h.assertUpdates(
+				h.graph.chans(), false, h.safeDisableTimeout,
+			)
+
+			// Request enables with manual = false should fail.
+			h.assertEnables(
+				h.graph.chans(), netann.ErrEnableManuallyDisabledChan, false,
+			)
+
+			// Request enables with manual = true should succeed.
+			h.assertEnables(h.graph.chans(), nil, true)
+
+			// Expect to see them all enabled on the network again.
+			h.assertUpdates(
+				h.graph.chans(), true, h.safeDisableTimeout,
+			)
+		},
+	},
 }
 
 // TestChanStatusManagerStateMachine tests the possible state transitions that