Differential D1832 Diff 6184 java/server/src/test/java/org/softwareheritage/graph/NeighborsTest.java
Changeset View
Changeset View
Standalone View
Standalone View
java/server/src/test/java/org/softwareheritage/graph/NeighborsTest.java
package org.softwareheritage.graph; | package org.softwareheritage.graph; | ||||
import java.util.ArrayList; | import java.util.ArrayList; | ||||
import org.junit.Test; | import org.junit.Test; | ||||
import org.softwareheritage.graph.Endpoint; | import org.softwareheritage.graph.Endpoint; | ||||
import org.softwareheritage.graph.Graph; | import org.softwareheritage.graph.Graph; | ||||
import org.softwareheritage.graph.GraphTest; | import org.softwareheritage.graph.GraphTest; | ||||
import org.softwareheritage.graph.SwhId; | import org.softwareheritage.graph.SwhId; | ||||
// Avoid warnings concerning Endpoint.Output.result manual cast | |||||
@SuppressWarnings("unchecked") | |||||
public class NeighborsTest extends GraphTest { | public class NeighborsTest extends GraphTest { | ||||
@Test | @Test | ||||
public void zeroNeighbor() { | public void zeroNeighbor() { | ||||
Graph graph = getGraph(); | Graph graph = getGraph(); | ||||
ArrayList<SwhId> expectedNodes = new ArrayList<>(); | ArrayList<SwhId> expectedNodes = new ArrayList<>(); | ||||
SwhId src1 = new SwhId("swh:1:ori:0000000000000000000000000000000000000021"); | SwhId src1 = new SwhId("swh:1:ori:0000000000000000000000000000000000000021"); | ||||
Endpoint endpoint1 = new Endpoint(graph, "backward", "*"); | Endpoint endpoint1 = new Endpoint(graph, "backward", "*"); | ||||
GraphTest.assertEqualsAnyOrder(expectedNodes, endpoint1.neighbors(src1)); | GraphTest.assertEqualsAnyOrder(expectedNodes, (ArrayList) endpoint1.neighbors(src1).result); | ||||
SwhId src2 = new SwhId("swh:1:cnt:0000000000000000000000000000000000000004"); | SwhId src2 = new SwhId("swh:1:cnt:0000000000000000000000000000000000000004"); | ||||
Endpoint endpoint2 = new Endpoint(graph, "forward", "*"); | Endpoint endpoint2 = new Endpoint(graph, "forward", "*"); | ||||
GraphTest.assertEqualsAnyOrder(expectedNodes, endpoint2.neighbors(src2)); | GraphTest.assertEqualsAnyOrder(expectedNodes, (ArrayList) endpoint2.neighbors(src2).result); | ||||
SwhId src3 = new SwhId("swh:1:cnt:0000000000000000000000000000000000000015"); | SwhId src3 = new SwhId("swh:1:cnt:0000000000000000000000000000000000000015"); | ||||
Endpoint endpoint3 = new Endpoint(graph, "forward", "*"); | Endpoint endpoint3 = new Endpoint(graph, "forward", "*"); | ||||
GraphTest.assertEqualsAnyOrder(expectedNodes, endpoint3.neighbors(src3)); | GraphTest.assertEqualsAnyOrder(expectedNodes, (ArrayList) endpoint3.neighbors(src3).result); | ||||
SwhId src4 = new SwhId("swh:1:rel:0000000000000000000000000000000000000019"); | SwhId src4 = new SwhId("swh:1:rel:0000000000000000000000000000000000000019"); | ||||
Endpoint endpoint4 = new Endpoint(graph, "backward", "*"); | Endpoint endpoint4 = new Endpoint(graph, "backward", "*"); | ||||
GraphTest.assertEqualsAnyOrder(expectedNodes, endpoint4.neighbors(src4)); | GraphTest.assertEqualsAnyOrder(expectedNodes, (ArrayList) endpoint4.neighbors(src4).result); | ||||
SwhId src5 = new SwhId("swh:1:dir:0000000000000000000000000000000000000008"); | SwhId src5 = new SwhId("swh:1:dir:0000000000000000000000000000000000000008"); | ||||
Endpoint endpoint5 = new Endpoint(graph, "forward", "snp:*,rev:*,rel:*"); | Endpoint endpoint5 = new Endpoint(graph, "forward", "snp:*,rev:*,rel:*"); | ||||
GraphTest.assertEqualsAnyOrder(expectedNodes, endpoint5.neighbors(src5)); | GraphTest.assertEqualsAnyOrder(expectedNodes, (ArrayList) endpoint5.neighbors(src5).result); | ||||
} | } | ||||
@Test | @Test | ||||
public void oneNeighbor() { | public void oneNeighbor() { | ||||
Graph graph = getGraph(); | Graph graph = getGraph(); | ||||
SwhId src1 = new SwhId("swh:1:rev:0000000000000000000000000000000000000003"); | SwhId src1 = new SwhId("swh:1:rev:0000000000000000000000000000000000000003"); | ||||
Endpoint endpoint1 = new Endpoint(graph, "forward", "*"); | Endpoint endpoint1 = new Endpoint(graph, "forward", "*"); | ||||
ArrayList<SwhId> expectedNodes1 = new ArrayList<>(); | ArrayList<SwhId> expectedNodes1 = new ArrayList<>(); | ||||
expectedNodes1.add(new SwhId("swh:1:dir:0000000000000000000000000000000000000002")); | expectedNodes1.add(new SwhId("swh:1:dir:0000000000000000000000000000000000000002")); | ||||
GraphTest.assertEqualsAnyOrder(expectedNodes1, endpoint1.neighbors(src1)); | GraphTest.assertEqualsAnyOrder(expectedNodes1, (ArrayList) endpoint1.neighbors(src1).result); | ||||
SwhId src2 = new SwhId("swh:1:dir:0000000000000000000000000000000000000017"); | SwhId src2 = new SwhId("swh:1:dir:0000000000000000000000000000000000000017"); | ||||
Endpoint endpoint2 = new Endpoint(graph, "forward", "dir:cnt"); | Endpoint endpoint2 = new Endpoint(graph, "forward", "dir:cnt"); | ||||
ArrayList<SwhId> expectedNodes2 = new ArrayList<>(); | ArrayList<SwhId> expectedNodes2 = new ArrayList<>(); | ||||
expectedNodes2.add(new SwhId("swh:1:cnt:0000000000000000000000000000000000000014")); | expectedNodes2.add(new SwhId("swh:1:cnt:0000000000000000000000000000000000000014")); | ||||
GraphTest.assertEqualsAnyOrder(expectedNodes2, endpoint2.neighbors(src2)); | GraphTest.assertEqualsAnyOrder(expectedNodes2, (ArrayList) endpoint2.neighbors(src2).result); | ||||
SwhId src3 = new SwhId("swh:1:dir:0000000000000000000000000000000000000012"); | SwhId src3 = new SwhId("swh:1:dir:0000000000000000000000000000000000000012"); | ||||
Endpoint endpoint3 = new Endpoint(graph, "backward", "*"); | Endpoint endpoint3 = new Endpoint(graph, "backward", "*"); | ||||
ArrayList<SwhId> expectedNodes3 = new ArrayList<>(); | ArrayList<SwhId> expectedNodes3 = new ArrayList<>(); | ||||
expectedNodes3.add(new SwhId("swh:1:rev:0000000000000000000000000000000000000013")); | expectedNodes3.add(new SwhId("swh:1:rev:0000000000000000000000000000000000000013")); | ||||
GraphTest.assertEqualsAnyOrder(expectedNodes3, endpoint3.neighbors(src3)); | GraphTest.assertEqualsAnyOrder(expectedNodes3, (ArrayList) endpoint3.neighbors(src3).result); | ||||
SwhId src4 = new SwhId("swh:1:rev:0000000000000000000000000000000000000009"); | SwhId src4 = new SwhId("swh:1:rev:0000000000000000000000000000000000000009"); | ||||
Endpoint endpoint4 = new Endpoint(graph, "backward", "rev:rev"); | Endpoint endpoint4 = new Endpoint(graph, "backward", "rev:rev"); | ||||
ArrayList<SwhId> expectedNodes4 = new ArrayList<>(); | ArrayList<SwhId> expectedNodes4 = new ArrayList<>(); | ||||
expectedNodes4.add(new SwhId("swh:1:rev:0000000000000000000000000000000000000013")); | expectedNodes4.add(new SwhId("swh:1:rev:0000000000000000000000000000000000000013")); | ||||
GraphTest.assertEqualsAnyOrder(expectedNodes4, endpoint4.neighbors(src4)); | GraphTest.assertEqualsAnyOrder(expectedNodes4, (ArrayList) endpoint4.neighbors(src4).result); | ||||
SwhId src5 = new SwhId("swh:1:snp:0000000000000000000000000000000000000020"); | SwhId src5 = new SwhId("swh:1:snp:0000000000000000000000000000000000000020"); | ||||
Endpoint endpoint5 = new Endpoint(graph, "backward", "*"); | Endpoint endpoint5 = new Endpoint(graph, "backward", "*"); | ||||
ArrayList<SwhId> expectedNodes5 = new ArrayList<>(); | ArrayList<SwhId> expectedNodes5 = new ArrayList<>(); | ||||
expectedNodes5.add(new SwhId("swh:1:ori:0000000000000000000000000000000000000021")); | expectedNodes5.add(new SwhId("swh:1:ori:0000000000000000000000000000000000000021")); | ||||
GraphTest.assertEqualsAnyOrder(expectedNodes5, endpoint5.neighbors(src5)); | GraphTest.assertEqualsAnyOrder(expectedNodes5, (ArrayList) endpoint5.neighbors(src5).result); | ||||
} | } | ||||
@Test | @Test | ||||
public void twoNeighbors() { | public void twoNeighbors() { | ||||
Graph graph = getGraph(); | Graph graph = getGraph(); | ||||
SwhId src1 = new SwhId("swh:1:snp:0000000000000000000000000000000000000020"); | SwhId src1 = new SwhId("swh:1:snp:0000000000000000000000000000000000000020"); | ||||
Endpoint endpoint1 = new Endpoint(graph, "forward", "*"); | Endpoint endpoint1 = new Endpoint(graph, "forward", "*"); | ||||
ArrayList<SwhId> expectedNodes1 = new ArrayList<>(); | ArrayList<SwhId> expectedNodes1 = new ArrayList<>(); | ||||
expectedNodes1.add(new SwhId("swh:1:rel:0000000000000000000000000000000000000010")); | expectedNodes1.add(new SwhId("swh:1:rel:0000000000000000000000000000000000000010")); | ||||
expectedNodes1.add(new SwhId("swh:1:rev:0000000000000000000000000000000000000009")); | expectedNodes1.add(new SwhId("swh:1:rev:0000000000000000000000000000000000000009")); | ||||
GraphTest.assertEqualsAnyOrder(expectedNodes1, endpoint1.neighbors(src1)); | GraphTest.assertEqualsAnyOrder(expectedNodes1, (ArrayList) endpoint1.neighbors(src1).result); | ||||
SwhId src2 = new SwhId("swh:1:dir:0000000000000000000000000000000000000008"); | SwhId src2 = new SwhId("swh:1:dir:0000000000000000000000000000000000000008"); | ||||
Endpoint endpoint2 = new Endpoint(graph, "forward", "dir:cnt"); | Endpoint endpoint2 = new Endpoint(graph, "forward", "dir:cnt"); | ||||
ArrayList<SwhId> expectedNodes2 = new ArrayList<>(); | ArrayList<SwhId> expectedNodes2 = new ArrayList<>(); | ||||
expectedNodes2.add(new SwhId("swh:1:cnt:0000000000000000000000000000000000000001")); | expectedNodes2.add(new SwhId("swh:1:cnt:0000000000000000000000000000000000000001")); | ||||
expectedNodes2.add(new SwhId("swh:1:cnt:0000000000000000000000000000000000000007")); | expectedNodes2.add(new SwhId("swh:1:cnt:0000000000000000000000000000000000000007")); | ||||
GraphTest.assertEqualsAnyOrder(expectedNodes2, endpoint2.neighbors(src2)); | GraphTest.assertEqualsAnyOrder(expectedNodes2, (ArrayList) endpoint2.neighbors(src2).result); | ||||
SwhId src3 = new SwhId("swh:1:cnt:0000000000000000000000000000000000000001"); | SwhId src3 = new SwhId("swh:1:cnt:0000000000000000000000000000000000000001"); | ||||
Endpoint endpoint3 = new Endpoint(graph, "backward", "*"); | Endpoint endpoint3 = new Endpoint(graph, "backward", "*"); | ||||
ArrayList<SwhId> expectedNodes3 = new ArrayList<>(); | ArrayList<SwhId> expectedNodes3 = new ArrayList<>(); | ||||
expectedNodes3.add(new SwhId("swh:1:dir:0000000000000000000000000000000000000008")); | expectedNodes3.add(new SwhId("swh:1:dir:0000000000000000000000000000000000000008")); | ||||
expectedNodes3.add(new SwhId("swh:1:dir:0000000000000000000000000000000000000002")); | expectedNodes3.add(new SwhId("swh:1:dir:0000000000000000000000000000000000000002")); | ||||
GraphTest.assertEqualsAnyOrder(expectedNodes3, endpoint3.neighbors(src3)); | GraphTest.assertEqualsAnyOrder(expectedNodes3, (ArrayList) endpoint3.neighbors(src3).result); | ||||
SwhId src4 = new SwhId("swh:1:rev:0000000000000000000000000000000000000009"); | SwhId src4 = new SwhId("swh:1:rev:0000000000000000000000000000000000000009"); | ||||
Endpoint endpoint4 = new Endpoint(graph, "backward", "rev:snp,rev:rel"); | Endpoint endpoint4 = new Endpoint(graph, "backward", "rev:snp,rev:rel"); | ||||
ArrayList<SwhId> expectedNodes4 = new ArrayList<>(); | ArrayList<SwhId> expectedNodes4 = new ArrayList<>(); | ||||
expectedNodes4.add(new SwhId("swh:1:snp:0000000000000000000000000000000000000020")); | expectedNodes4.add(new SwhId("swh:1:snp:0000000000000000000000000000000000000020")); | ||||
expectedNodes4.add(new SwhId("swh:1:rel:0000000000000000000000000000000000000010")); | expectedNodes4.add(new SwhId("swh:1:rel:0000000000000000000000000000000000000010")); | ||||
GraphTest.assertEqualsAnyOrder(expectedNodes4, endpoint4.neighbors(src4)); | GraphTest.assertEqualsAnyOrder(expectedNodes4, (ArrayList) endpoint4.neighbors(src4).result); | ||||
} | } | ||||
@Test | @Test | ||||
public void threeNeighbors() { | public void threeNeighbors() { | ||||
Graph graph = getGraph(); | Graph graph = getGraph(); | ||||
SwhId src1 = new SwhId("swh:1:dir:0000000000000000000000000000000000000008"); | SwhId src1 = new SwhId("swh:1:dir:0000000000000000000000000000000000000008"); | ||||
Endpoint endpoint1 = new Endpoint(graph, "forward", "*"); | Endpoint endpoint1 = new Endpoint(graph, "forward", "*"); | ||||
ArrayList<SwhId> expectedNodes1 = new ArrayList<>(); | ArrayList<SwhId> expectedNodes1 = new ArrayList<>(); | ||||
expectedNodes1.add(new SwhId("swh:1:dir:0000000000000000000000000000000000000006")); | expectedNodes1.add(new SwhId("swh:1:dir:0000000000000000000000000000000000000006")); | ||||
expectedNodes1.add(new SwhId("swh:1:cnt:0000000000000000000000000000000000000001")); | expectedNodes1.add(new SwhId("swh:1:cnt:0000000000000000000000000000000000000001")); | ||||
expectedNodes1.add(new SwhId("swh:1:cnt:0000000000000000000000000000000000000007")); | expectedNodes1.add(new SwhId("swh:1:cnt:0000000000000000000000000000000000000007")); | ||||
GraphTest.assertEqualsAnyOrder(expectedNodes1, endpoint1.neighbors(src1)); | GraphTest.assertEqualsAnyOrder(expectedNodes1, (ArrayList) endpoint1.neighbors(src1).result); | ||||
SwhId src2 = new SwhId("swh:1:rev:0000000000000000000000000000000000000009"); | SwhId src2 = new SwhId("swh:1:rev:0000000000000000000000000000000000000009"); | ||||
Endpoint endpoint2 = new Endpoint(graph, "backward", "*"); | Endpoint endpoint2 = new Endpoint(graph, "backward", "*"); | ||||
ArrayList<SwhId> expectedNodes2 = new ArrayList<>(); | ArrayList<SwhId> expectedNodes2 = new ArrayList<>(); | ||||
expectedNodes2.add(new SwhId("swh:1:snp:0000000000000000000000000000000000000020")); | expectedNodes2.add(new SwhId("swh:1:snp:0000000000000000000000000000000000000020")); | ||||
expectedNodes2.add(new SwhId("swh:1:rel:0000000000000000000000000000000000000010")); | expectedNodes2.add(new SwhId("swh:1:rel:0000000000000000000000000000000000000010")); | ||||
expectedNodes2.add(new SwhId("swh:1:rev:0000000000000000000000000000000000000013")); | expectedNodes2.add(new SwhId("swh:1:rev:0000000000000000000000000000000000000013")); | ||||
GraphTest.assertEqualsAnyOrder(expectedNodes2, endpoint2.neighbors(src2)); | GraphTest.assertEqualsAnyOrder(expectedNodes2, (ArrayList) endpoint2.neighbors(src2).result); | ||||
} | } | ||||
} | } |