From 2b012b06a0c5fc4ef50971bc803414dee6b91e68 Mon Sep 17 00:00:00 2001
From: "Johan T. Halseth" <johanth@gmail.com>
Date: Mon, 28 Jan 2019 09:06:48 +0100
Subject: [PATCH] channeldb/graph test: add test for update policy for unknown
 edge

---
 channeldb/graph_test.go | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/channeldb/graph_test.go b/channeldb/graph_test.go
index 27ed7ba5..378391ec 100644
--- a/channeldb/graph_test.go
+++ b/channeldb/graph_test.go
@@ -729,6 +729,14 @@ func TestEdgeInfoUpdates(t *testing.T) {
 
 	// Create an edge and add it to the db.
 	edgeInfo, edge1, edge2 := createChannelEdge(db, node1, node2)
+
+	// Make sure inserting the policy at this point, before the edge info
+	// is added, will fail.
+	if err := graph.UpdateEdgePolicy(edge1); err != ErrEdgeNotFound {
+		t.Fatalf("expected ErrEdgeNotFound, got: %v", err)
+	}
+
+	// Add the edge info.
 	if err := graph.AddChannelEdge(edgeInfo); err != nil {
 		t.Fatalf("unable to create channel edge: %v", err)
 	}