From 61a67dae2926712a2ec95ffc3c6e8537b4f00349 Mon Sep 17 00:00:00 2001
From: Adam Sigal <adamsigal@gmail.com>
Date: Sat, 6 Jul 2019 18:27:31 -0400
Subject: [PATCH] pgraph: Add a test for FindEdge() and HasVertex() functions

Added a new test called TestFindEdge1(). In it, a 7-node
graph is created on which to test the aforementioned functions.
---
 pgraph/pgraph_test.go | 44 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 44 insertions(+)

diff --git a/pgraph/pgraph_test.go b/pgraph/pgraph_test.go
index c769f561..e491ed9a 100644
--- a/pgraph/pgraph_test.go
+++ b/pgraph/pgraph_test.go
@@ -840,3 +840,47 @@ func TestDeleteEdge2(t *testing.T) {
 		t.Errorf("expected outgoing graph edges for vertex v1: e4, instead of: %s", res)
 	}
 }
+
+func TestFindEdge1(t *testing.T) {
+	g, _ := NewGraph("g")
+
+	v1 := NV("v1")
+	v2 := NV("v2")
+	v3 := NV("v3")
+	v4 := NV("v4")
+	v5 := NV("v5")
+	v6 := NV("v6")
+	v7 := NV("v7")
+
+	e1 := NE("e1")
+	e2 := NE("e2")
+	e3 := NE("e3")
+	e4 := NE("e4")
+	e5 := NE("e5")
+	e6 := NE("e6")
+	e7 := NE("e7")
+	e8 := NE("e8")
+	e9 := NE("e9")
+	e10 := NE("e10")
+
+	g.AddEdge(v1, v2, e1)
+	g.AddEdge(v1, v4, e2)
+	g.AddEdge(v1, v3, e3)
+	g.AddEdge(v2, v3, e4)
+	g.AddEdge(v2, v4, e5)
+	g.AddEdge(v2, v6, e6)
+	g.AddEdge(v2, v5, e7)
+	g.AddEdge(v3, v6, e8)
+	g.AddEdge(v4, v7, e9)
+	g.AddEdge(v5, v7, e10)
+
+	if !(g.HasVertex(v1) && g.HasVertex(v4) && g.HasVertex(v7)) {
+		t.Errorf("graph expected to have vertices v1, v4, and v7")
+	}
+	if g.FindEdge(v1, v4) != e2 {
+		t.Errorf("edge e2 was not returned")
+	}
+	if g.FindEdge(v1, v7) != nil {
+		t.Errorf("an edge was found although it did not exist")
+	}
+}