channeldb: update TestGraphZombieIndex to use require

channeldb/graph_test.go

@@ -3067,16 +3067,12 @@ func TestGraphZombieIndex(t *testing.T) {
 	}
 
 	edge, _, _ := createChannelEdge(graph.db, node1, node2)
-	if err := graph.AddChannelEdge(edge); err != nil {
+	require.NoError(t, graph.AddChannelEdge(edge))
-		t.Fatalf("unable to create channel edge: %v", err)
-	}
 
 	// Since the edge is known the graph and it isn't a zombie, IsZombieEdge
 	// should not report the channel as a zombie.
 	isZombie, _, _ := graph.IsZombieEdge(edge.ChannelID)
-	if isZombie {
+	require.False(t, isZombie)
-		t.Fatal("expected edge to not be marked as zombie")
-	}
 	assertNumZombies(t, graph, 0)
 
 	// If we delete the edge and mark it as a zombie, then we should expect
@@ -3084,28 +3080,18 @@ func TestGraphZombieIndex(t *testing.T) {
 	err = graph.DeleteChannelEdges(false, true, edge.ChannelID)
 	require.NoError(t, err, "unable to mark edge as zombie")
 	isZombie, pubKey1, pubKey2 := graph.IsZombieEdge(edge.ChannelID)
-	if !isZombie {
+	require.True(t, isZombie)
-		t.Fatal("expected edge to be marked as zombie")
+	require.Equal(t, node1.PubKeyBytes, pubKey1)
-	}
+	require.Equal(t, node2.PubKeyBytes, pubKey2)
-	if pubKey1 != node1.PubKeyBytes {
-		t.Fatalf("expected pubKey1 %x, got %x", node1.PubKeyBytes,
-			pubKey1)
-	}
-	if pubKey2 != node2.PubKeyBytes {
-		t.Fatalf("expected pubKey2 %x, got %x", node2.PubKeyBytes,
-			pubKey2)
-	}
 	assertNumZombies(t, graph, 1)
 
 	// Similarly, if we mark the same edge as live, we should no longer see
 	// it within the index.
-	if err := graph.MarkEdgeLive(edge.ChannelID); err != nil {
+	require.NoError(t, graph.MarkEdgeLive(edge.ChannelID))
-		t.Fatalf("unable to mark edge as live: %v", err)
+
-	}
 	isZombie, _, _ = graph.IsZombieEdge(edge.ChannelID)
-	if isZombie {
+	require.False(t, isZombie)
-		t.Fatal("expected edge to not be marked as zombie")
+
-	}
 	assertNumZombies(t, graph, 0)
 
 	// If we mark the edge as a zombie manually, then it should show up as
@@ -3114,10 +3100,9 @@ func TestGraphZombieIndex(t *testing.T) {
 		edge.ChannelID, node1.PubKeyBytes, node2.PubKeyBytes,
 	)
 	require.NoError(t, err, "unable to mark edge as zombie")
+
 	isZombie, _, _ = graph.IsZombieEdge(edge.ChannelID)
-	if !isZombie {
+	require.True(t, isZombie)
-		t.Fatal("expected edge to be marked as zombie")
-	}
 	assertNumZombies(t, graph, 1)
 }