public void GetEdgesShouldReturnTheEdgesReceivedFromTheDb() { var record = new Mock <IRecord>(); record.Setup(r => r["n.id"]).Returns(1); record.Setup(r => r["m.id"]).Returns(2); var result = new Mock <IStatementResult>(); result.Setup(r => r.GetEnumerator()).Returns(new List <IRecord> { record.Object }.GetEnumerator()); session.Setup(s => s.Run(It.IsAny <string>(), null)).Returns(result.Object); var sut = new GraphDataServiceQuery(loggerFactory.Object, driver.Object); var edges = sut.GetEdges(); Assert.AreEqual(1, edges.Count); Assert.AreEqual(1, edges[0].Vertex1); Assert.AreEqual(2, edges[0].Vertex2); }
public void GetGraphShouldReturnTheVerticesReceivedFromTheDb() { var record = new Mock <IRecord>(); record.Setup(r => r["n.id"]).Returns(1); record.Setup(r => r["m.id"]).Returns(2); record.Setup(r => r["v.id"]).Returns(1); record.Setup(r => r["v.label"]).Returns("aaa"); var result = new Mock <IStatementResult>(); result.Setup(r => r.GetEnumerator()).Returns(new List <IRecord> { record.Object }.GetEnumerator()); session.Setup(s => s.Run(It.IsAny <string>(), null)).Returns(result.Object); transaction.Setup(s => s.Run(It.IsAny <string>(), null)).Returns(result.Object); var sut = new GraphDataServiceQuery(loggerFactory.Object, driver.Object); var graph = sut.GetGraph(); Assert.AreEqual(1, graph.Vertices.Count); Assert.AreEqual(1, graph.Vertices[0].Id); Assert.AreEqual("aaa", graph.Vertices[0].Label); }