forked from typedb/typedb
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Adding a new end-to-end test 'ClientJavaE2' which performs basic quer…
…ies against a real Grakn distribution (typedb#2917) We need to be sure that Grakn's basic functionality works as it should whenever we're merging a new PR or releasing the distribution to public. 1. Adds a new module under test-integration/test-client-java which has the test ClientJavaE2E. It performs the following queries against a real Grakn distribution: define query match; get; match; insert; match; delete; match; aggregate; compute 2. Adds client-java-e2e job to CircleCI 3. Rename DistributionE2E to GraknGraqlCommandsE2E and Distribution_WithARunningGraknE2E to GraknGraqlCommands_WithARunningGraknE2E
- Loading branch information
Showing
7 changed files
with
400 additions
and
41 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,46 @@ | ||
<?xml version="1.0" encoding="UTF-8"?> | ||
<!-- | ||
~ Grakn - A Distributed Semantic Database | ||
~ Copyright (C) 2016-2018 Grakn Labs Limited | ||
~ | ||
~ Grakn is free software: you can redistribute it and/or modify | ||
~ it under the terms of the GNU Affero General Public License as published by | ||
~ the Free Software Foundation, either version 3 of the License, or | ||
~ (at your option) any later version. | ||
~ | ||
~ Grakn is distributed in the hope that it will be useful, | ||
~ but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
~ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | ||
~ GNU General Public License for more details. | ||
~ | ||
~ You should have received a copy of the GNU General Public License | ||
~ along with Grakn. If not, see <http://www.gnu.org/licenses/gpl.txt>. | ||
--> | ||
|
||
<project xmlns="http://maven.apache.org/POM/4.0.0" | ||
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" | ||
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd"> | ||
<parent> | ||
<artifactId>grakn-test</artifactId> | ||
<groupId>ai.grakn</groupId> | ||
<version>1.3.0-SNAPSHOT</version> | ||
</parent> | ||
<modelVersion>4.0.0</modelVersion> | ||
|
||
<artifactId>test-client-java</artifactId> | ||
|
||
<dependencies> | ||
<dependency> | ||
<groupId>ai.grakn</groupId> | ||
<artifactId>client-java</artifactId> | ||
</dependency> | ||
<dependency> | ||
<groupId>junit</groupId> | ||
<artifactId>junit</artifactId> | ||
</dependency> | ||
<dependency> | ||
<groupId>org.zeroturnaround</groupId> | ||
<artifactId>zt-exec</artifactId> | ||
</dependency> | ||
</dependencies> | ||
</project> |
272 changes: 272 additions & 0 deletions
272
grakn-test/test-client-java/src/test/java/ai/grakn/client/ClientJavaE2E.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,272 @@ | ||
package ai.grakn.client; | ||
|
||
import ai.grakn.GraknTxType; | ||
import ai.grakn.Keyspace; | ||
import ai.grakn.concept.AttributeType; | ||
import ai.grakn.graql.AggregateQuery; | ||
import ai.grakn.graql.ComputeQuery; | ||
import ai.grakn.graql.DefineQuery; | ||
import ai.grakn.graql.DeleteQuery; | ||
import ai.grakn.graql.GetQuery; | ||
import ai.grakn.graql.InsertQuery; | ||
import ai.grakn.graql.answer.ConceptMap; | ||
import ai.grakn.graql.answer.Value; | ||
import ai.grakn.util.GraqlSyntax; | ||
import ai.grakn.util.SimpleURI; | ||
import org.apache.commons.io.FileUtils; | ||
import org.junit.AfterClass; | ||
import org.junit.BeforeClass; | ||
import org.junit.Test; | ||
import org.slf4j.Logger; | ||
import org.slf4j.LoggerFactory; | ||
import org.zeroturnaround.exec.ProcessExecutor; | ||
|
||
import java.io.IOException; | ||
import java.util.List; | ||
import java.util.concurrent.TimeoutException; | ||
import java.util.function.Consumer; | ||
import java.util.stream.Collectors; | ||
|
||
import static ai.grakn.client.ClientJavaE2EConstants.GRAKN_UNZIPPED_DIRECTORY; | ||
import static ai.grakn.client.ClientJavaE2EConstants.assertGraknRunning; | ||
import static ai.grakn.client.ClientJavaE2EConstants.assertGraknStopped; | ||
import static ai.grakn.client.ClientJavaE2EConstants.assertZipExists; | ||
import static ai.grakn.client.ClientJavaE2EConstants.unzipGrakn; | ||
import static ai.grakn.graql.Graql.and; | ||
import static ai.grakn.graql.Graql.count; | ||
import static ai.grakn.graql.Graql.define; | ||
import static ai.grakn.graql.Graql.label; | ||
import static ai.grakn.graql.Graql.var; | ||
import static org.hamcrest.MatcherAssert.assertThat; | ||
import static org.hamcrest.Matchers.*; | ||
|
||
/** | ||
* | ||
* Performs various queries with the client-java library: | ||
* - define a schema with a rule | ||
* - match; get; | ||
* - match; get of an inferred relationship | ||
* - match; insert; | ||
* - match; delete; | ||
* - match; aggregate; | ||
* - and compute count | ||
* | ||
* The tests are highly interconnected hence why they are grouped into a single test. | ||
* If you split them into multiple tests, there is no guarantee that they are ran in the order they are defined, | ||
* and there is a chance that the match; get; test is performed before the define a schema test, which would cause it to fail. | ||
* | ||
* The schema describes a lion family which consists of a lion, lioness, and the offspring - three young lions. The mating | ||
* relationship captures the mating act between the male and female partners (ie., the lion and lioness). The child-bearing | ||
* relationship captures the child-bearing act which results from the mating act. | ||
* | ||
* The rule is one such that if there is an offspring which is the result of a certain child-bearing act, then | ||
* that offspring is the child of the male and female partners which are involved in the mating act. | ||
* | ||
* @author Ganeshwara Herawan Hananda | ||
*/ | ||
public class ClientJavaE2E { | ||
private static Logger LOG = LoggerFactory.getLogger(ClientJavaE2E.class); | ||
|
||
@BeforeClass | ||
public static void setup_prepareDistribution() throws IOException, InterruptedException, TimeoutException { | ||
ProcessExecutor commandExecutor = new ProcessExecutor() | ||
.directory(GRAKN_UNZIPPED_DIRECTORY.toFile()) | ||
.redirectOutput(System.out) | ||
.redirectError(System.err) | ||
.readOutput(true); | ||
|
||
LOG.info("setup_prepareDistribution() - check if there is no existing Grakn process running..."); | ||
assertGraknStopped(); | ||
assertZipExists(); | ||
unzipGrakn(); | ||
LOG.info("setup_prepareDistribution() - done."); | ||
|
||
LOG.info("setup_prepareDistribution() - starting a new Grakn instance..."); | ||
commandExecutor.command("./grakn", "server", "start").execute(); | ||
assertGraknRunning(); | ||
LOG.info("setup_prepareDistribution() - Grakn started successfully."); | ||
} | ||
|
||
@AfterClass | ||
public static void cleanup_cleanupDistribution() throws IOException, InterruptedException, TimeoutException { | ||
ProcessExecutor commandExecutor = new ProcessExecutor() | ||
.directory(GRAKN_UNZIPPED_DIRECTORY.toFile()) | ||
.redirectOutput(System.out) | ||
.redirectError(System.err) | ||
.readOutput(true); | ||
LOG.info("cleanup_cleanupDistribution() - stopping the Grakn instance..."); | ||
commandExecutor.command("./grakn", "server", "stop").execute(); | ||
assertGraknStopped(); | ||
LOG.info("cleanup_cleanupDistribution() - done."); | ||
|
||
LOG.info("cleanup_cleanupDistribution() - deleting the Grakn distribution..."); | ||
FileUtils.deleteDirectory(GRAKN_UNZIPPED_DIRECTORY.toFile()); | ||
LOG.info("cleanup_cleanupDistribution() - cleanup successful."); | ||
} | ||
|
||
@Test | ||
public void clientJavaE2E() { | ||
LOG.info("clientJavaE2E() - starting client-java E2E..."); | ||
|
||
localhostGraknTx(tx -> { | ||
DefineQuery defineQuery = tx.graql().define( | ||
label("child-bearing").sub("relationship").relates("offspring").relates("child-bearer"), | ||
label("mating").sub("relationship").relates("male-partner").relates("female-partner").plays("child-bearer"), | ||
label("parentship").sub("relationship").relates("parent").relates("child"), | ||
|
||
label("name").sub("attribute").datatype(AttributeType.DataType.STRING), | ||
label("lion").sub("entity").has("name").plays("male-partner").plays("female-partner").plays("offspring"), | ||
|
||
label("infer-parentship-from-mating-and-child-bearing").sub("rule") | ||
.when(and( | ||
var().rel("male-partner", var("male")).rel("female-partner", var("female")).isa("mating"), | ||
var("childbearing").rel("child-bearer").rel("offspring", var("offspring")).isa("child-bearing") | ||
)) | ||
.then(and( | ||
var().rel("parent", var("male")).rel("parent", var("female")).rel("child", var("offspring")).isa("parentship") | ||
)) | ||
); | ||
LOG.info("clientJavaE2E() - define a schema..."); | ||
LOG.info("clientJavaE2E() - '" + defineQuery + "'"); | ||
defineQuery.execute(); | ||
tx.commit(); | ||
LOG.info("clientJavaE2E() - done."); | ||
}); | ||
|
||
localhostGraknTx(tx -> { | ||
GetQuery getThingQuery = tx.graql().match(var("t").sub("thing")).get(); | ||
LOG.info("clientJavaE2E() - assert if schema defined..."); | ||
LOG.info("clientJavaE2E() - '" + getThingQuery + "'"); | ||
List<String> definedSchema = getThingQuery.execute().stream() | ||
.map(answer -> answer.get(var("t")).asType().label().getValue()).collect(Collectors.toList()); | ||
String[] correctSchema = new String[] { "thing", "entity", "relationship", "attribute", | ||
"lion", "mating", "parentship", "child-bearing", "@has-name", "name" }; | ||
assertThat(definedSchema, hasItems(correctSchema)); | ||
LOG.info("clientJavaE2E() - done."); | ||
}); | ||
|
||
localhostGraknTx(tx -> { | ||
String[] names = lionNames(); | ||
InsertQuery insertLionQuery = tx.graql().insert( | ||
var().isa("lion").has("name").val(names[0]), | ||
var().isa("lion").has("name").val(names[1]), | ||
var().isa("lion").has("name").val(names[2]) | ||
); | ||
LOG.info("clientJavaE2E() - insert some data..."); | ||
LOG.info("clientJavaE2E() - '" + insertLionQuery + "'"); | ||
insertLionQuery.execute(); | ||
tx.commit(); | ||
LOG.info("clientJavaE2E() - done."); | ||
}); | ||
|
||
localhostGraknTx(tx -> { | ||
String[] familyMembers = lionNames(); | ||
LOG.info("clientJavaE2E() - inserting mating relationships..."); | ||
InsertQuery insertMatingQuery = tx.graql() | ||
.match( | ||
var("lion").isa("lion").has("name").val(familyMembers[0]), | ||
var("lioness").isa("lion").has("name").val(familyMembers[1])) | ||
.insert(var().isa("mating").rel("male-partner", var("lion")).rel("female-partner", var("lioness"))); | ||
LOG.info("clientJavaE2E() - '" + insertMatingQuery + "'"); | ||
List<ConceptMap> insertedMating = insertMatingQuery.execute(); | ||
|
||
LOG.info("clientJavaE2E() - inserting child-bearing relationships..."); | ||
InsertQuery insertChildBearingQuery = tx.graql() | ||
.match( | ||
var("lion").isa("lion").has("name").val(familyMembers[0]), | ||
var("lioness").isa("lion").has("name").val(familyMembers[1]), | ||
var("offspring").isa("lion").has("name").val(familyMembers[2]), | ||
var("mating").rel("male-partner", var("lion")).rel("female-partner", var("lioness")).isa("mating") | ||
) | ||
.insert(var("childbearing").rel("child-bearer", var("mating")).rel("offspring", var("offspring")).isa("child-bearing")); | ||
LOG.info("clientJavaE2E() - '" + insertChildBearingQuery + "'"); | ||
List<ConceptMap> insertedChildBearing = insertChildBearingQuery.execute(); | ||
|
||
tx.commit(); | ||
|
||
assertThat(insertedMating, hasSize(1)); | ||
assertThat(insertedChildBearing, hasSize(1)); | ||
LOG.info("clientJavaE2E() - done."); | ||
}); | ||
|
||
localhostGraknTx(tx -> { | ||
LOG.info("clientJavaE2E() - execute match get on the lion instances..."); | ||
GetQuery getLionQuery = tx.graql().match(var("p").isa("lion").has("name", var("n"))).get(); | ||
LOG.info("clientJavaE2E() - '" + getLionQuery + "'"); | ||
List<ConceptMap> insertedLions = getLionQuery.execute(); | ||
List<String> insertedNames = insertedLions.stream().map(answer -> answer.get("n").asAttribute().value().toString()).collect(Collectors.toList()); | ||
assertThat(insertedNames, containsInAnyOrder(lionNames())); | ||
|
||
LOG.info("clientJavaE2E() - execute match get on the mating relationships..."); | ||
GetQuery getMatingQuery = tx.graql().match(var().isa("mating")).get(); | ||
LOG.info("clientJavaE2E() - '" + getMatingQuery + "'"); | ||
List<ConceptMap> insertedMating = getMatingQuery.execute(); | ||
assertThat(insertedMating, hasSize(1)); | ||
|
||
LOG.info("clientJavaE2E() - execute match get on the child-bearing..."); | ||
GetQuery getChildBearingQuery = tx.graql().match(var().isa("child-bearing")).get(); | ||
LOG.info("clientJavaE2E() - '" + getChildBearingQuery + "'"); | ||
List<ConceptMap> insertedChildBearing = getChildBearingQuery.execute(); | ||
assertThat(insertedChildBearing, hasSize(1)); | ||
LOG.info("clientJavaE2E() - done."); | ||
}); | ||
|
||
localhostGraknTx(tx -> { | ||
LOG.info("clientJavaE2E() - match get inferred relationships..."); | ||
GetQuery getParentship = tx.graql().match( | ||
var("parentship") | ||
.isa("parentship") | ||
.rel("parent", var("parent")) | ||
.rel("child", var("child"))).get(); | ||
LOG.info("clientJavaE2E() - '" + getParentship + "'"); | ||
List<ConceptMap> parentship = getParentship.execute(); | ||
assertThat(parentship, hasSize(2)); | ||
LOG.info("clientJavaE2E() - done."); | ||
}); | ||
|
||
localhostGraknTx(tx -> { | ||
LOG.info("clientJavaE2E() - match aggregate..."); | ||
AggregateQuery<Value> aggregateQuery = tx.graql().match(var("p").isa("lion")).aggregate(count()); | ||
LOG.info("clientJavaE2E() - '" + aggregateQuery + "'"); | ||
int aggregateCount = aggregateQuery.execute().number().intValue(); | ||
assertThat(aggregateCount, equalTo(lionNames().length)); | ||
LOG.info("clientJavaE2E() - done."); | ||
}); | ||
|
||
localhostGraknTx(tx -> { | ||
LOG.info("clientJavaE2E() - compute count..."); | ||
final ComputeQuery<Value> computeQuery = tx.graql().compute(GraqlSyntax.Compute.Method.COUNT).in("lion"); | ||
LOG.info("clientJavaE2E() - '" + computeQuery + "'"); | ||
int computeCount = computeQuery.execute().get(0).number().intValue(); | ||
assertThat(computeCount, equalTo(lionNames().length)); | ||
LOG.info("clientJavaE2E() - done."); | ||
}); | ||
|
||
localhostGraknTx(tx -> { | ||
LOG.info("clientJavaE2E() - match delete..."); | ||
DeleteQuery deleteQuery = tx.graql().match(var("m").isa("mating")).delete(var("m")); | ||
LOG.info("clientJavaE2E() - '" + deleteQuery + "'"); | ||
deleteQuery.execute(); | ||
List<ConceptMap> matings = tx.graql().match(var("m").isa("mating")).get().execute(); | ||
assertThat(matings, hasSize(0)); | ||
LOG.info("clientJavaE2E() - done."); | ||
}); | ||
|
||
LOG.info("clientJavaE2E() - client-java E2E test done."); | ||
} | ||
|
||
private String[] lionNames() { | ||
return new String[] { "male-partner", "female-partner", "young-lion" }; | ||
} | ||
|
||
private void localhostGraknTx(Consumer<Grakn.Transaction> fn) { | ||
SimpleURI graknHost = new SimpleURI("localhost", 48555); | ||
Keyspace keyspace = Keyspace.of("grakn"); | ||
|
||
try (Grakn.Session session = Grakn.session(graknHost, keyspace)) { | ||
try (Grakn.Transaction transaction = session.transaction(GraknTxType.WRITE)) { | ||
fn.accept(transaction); | ||
} | ||
} | ||
} | ||
} |
Oops, something went wrong.