Graph.Add() returns pointer to new Vertex and error

graph/graph.go

@@ -63,5 +63,5 @@ type Graph interface {
 	Path(int, int) ([]*Vertex, error)
 	// Add adds the specified Vertex to the graph.  Add does not do any uniqueness checks - it is up to the
 	// user to keep track of which vertices have already been added.
-	Add(Vertex)
+	Add(Vertex) (*Vertex, error)
 }

graph/list.go

@@ -192,9 +192,11 @@ func New(vertices []Vertex) (Graph, error) {
 
 var addMutex sync.Mutex
 
-func (a AdjacencyList) Add(v Vertex) {
+func (a AdjacencyList) Add(v Vertex) (*Vertex, error) {
 	addMutex.Lock()
 	v.index = len(a)
 	a = append(a, &v)
 	addMutex.Unlock()
+
+	return a.AtIndex(v.index)
 }

graph/list_test.go

@@ -162,4 +162,25 @@ var _ = Describe("List", func() {
 			})
 		})
 	})
+
+	Describe("Add", func() {
+		var (
+			newVertex graph.Vertex
+		)
+
+		BeforeEach(func() {
+			newVertex = graph.Vertex{
+				Attributes: graph.Attributes{
+					"foo": "peter",
+				},
+			}
+		})
+
+		It("Adds the new Vertex", func() {
+			nvp, err := g.Add(newVertex)
+			Expect(nvp).NotTo(BeNil())
+			Expect(err).NotTo(HaveOccurred())
+			Expect(nvp.Get("foo").(string)).To(Equal("peter"))
+		})
+	})
 })