Java Code Examples for org.eclipse.rdf4j.common.iteration.Iterations#asList()

@Test
public void testMultipleBindings() throws Exception {

	List<BindingSet> leftBlock = new ArrayList<>();
	leftBlock.add(bindingSet(binding("x", irid("p1")), binding("y", l("P1"))));
	leftBlock.add(bindingSet(binding("x", irid("p2")), binding("y", l("P2"))));

	List<BindingSet> rightBlock = new ArrayList<>();
	rightBlock.add(bindingSet(binding("x", irid("p2")), binding("z", l("something"))));

	CloseableIteration<BindingSet, QueryEvaluationException> joinResultIter = HashJoin.join(leftBlock, rightBlock,
			Sets.newHashSet("x"),
			Collections.emptyList());
	List<BindingSet> joinResult = Iterations.asList(joinResultIter);

	Assertions.assertEquals(1, joinResult.size());
	Assertions.assertEquals(
			bindingSet(binding("x", irid("p2")), binding("y", l("P2")), binding("z", l("something"))),
			joinResult.get(0));
}
 

@Test
public void testGetData() {
	try (NotifyingSailConnection connection = elasticsearchStore.getConnection()) {
		connection.begin(IsolationLevels.NONE);
		connection.addStatement(RDF.TYPE, RDF.TYPE, RDFS.RESOURCE);
		connection.flush();
		List<? extends Statement> statements = Iterations.asList(connection.getStatements(null, null, null, true));

		connection.commit();

		System.out.println(Arrays.toString(statements.toArray()));

		assertEquals(1, statements.size());
		assertEquals(SimpleValueFactory.getInstance().createStatement(RDF.TYPE, RDF.TYPE, RDFS.RESOURCE),
				statements.get(0));
	}

}
 

/**
 * Test method for
 * {@link org.eclipse.rdf4j.sail.memory.MemTripleSource#getStatements(org.eclipse.rdf4j.model.Resource, org.eclipse.rdf4j.model.IRI, org.eclipse.rdf4j.model.Value, org.eclipse.rdf4j.model.Resource[])}
 * .
 */
@Test
public final void testGetStatementsNoContextsPredicateOwlClassNoContexts() throws Exception {
	loadTestData("/alp-testdata.ttl");
	TripleSource source = getTripleSourceCommitted();

	try (CloseableIteration<? extends Statement, QueryEvaluationException> statements = source.getStatements(null,
			RDF.TYPE, OWL.CLASS)) {
		List<Statement> list = Iterations.asList(statements);

		assertEquals(4, list.size());
	}
}
 

/**
 * Test method for
 * {@link org.eclipse.rdf4j.sail.memory.MemTripleSource#getStatements(org.eclipse.rdf4j.model.Resource, org.eclipse.rdf4j.model.IRI, org.eclipse.rdf4j.model.Value, org.eclipse.rdf4j.model.Resource[])}
 * .
 */
@Test
public final void testGetStatementsOneContextOnePredicate() throws Exception {
	loadTestData("/alp-testdata.ttl", this.alice);
	TripleSource source = getTripleSourceCommitted();

	try (CloseableIteration<? extends Statement, QueryEvaluationException> statements = source.getStatements(null,
			RDFS.SUBCLASSOF, null)) {
		List<Statement> list = Iterations.asList(statements);

		assertEquals(4, list.size());
	}
}
 

/**
 * Test method for
 * {@link org.eclipse.rdf4j.sail.memory.MemTripleSource#getStatements(org.eclipse.rdf4j.model.Resource, org.eclipse.rdf4j.model.IRI, org.eclipse.rdf4j.model.Value, org.eclipse.rdf4j.model.Resource[])}
 * .
 */
@Test
public final void testGetStatementsTwoContextsPredicateExClassTwoContexts() throws Exception {
	loadTestData("/alp-testdata.ttl", this.alice, this.bob);
	TripleSource source = getTripleSourceCommitted();

	try (CloseableIteration<? extends Statement, QueryEvaluationException> statements = source.getStatements(null,
			RDFS.SUBCLASSOF, f.createIRI(EX_NS, "A"), this.alice, this.bob)) {
		List<Statement> list = Iterations.asList(statements);

		assertEquals(6, list.size());
	}
}
 

/**
 * Test method for
 * {@link org.eclipse.rdf4j.sail.memory.MemTripleSource#getStatements(org.eclipse.rdf4j.model.Resource, org.eclipse.rdf4j.model.IRI, org.eclipse.rdf4j.model.Value, org.eclipse.rdf4j.model.Resource[])}
 * .
 */
@Test
public final void testGetStatementsNoContextsPredicateOwlClassTwoContexts() throws Exception {
	loadTestData("/alp-testdata.ttl");
	TripleSource source = getTripleSourceCommitted();

	try (CloseableIteration<? extends Statement, QueryEvaluationException> statements = source.getStatements(null,
			RDF.TYPE, OWL.CLASS, this.alice, this.bob)) {
		List<Statement> list = Iterations.asList(statements);

		assertEquals(0, list.size());
	}
}
 

/**
 * Test method for
 * {@link org.eclipse.rdf4j.sail.memory.MemTripleSource#getStatements(org.eclipse.rdf4j.model.Resource, org.eclipse.rdf4j.model.IRI, org.eclipse.rdf4j.model.Value, org.eclipse.rdf4j.model.Resource[])}
 * .
 */
@Test
public final void testGetStatementsTwoContextsExClassPredicateTwoContexts() throws Exception {
	loadTestData("/alp-testdata.ttl", this.alice, this.bob);
	TripleSource source = getTripleSourceCommitted();

	try (CloseableIteration<? extends Statement, QueryEvaluationException> statements = source
			.getStatements(f.createIRI(EX_NS, "C"), RDFS.SUBCLASSOF, null, this.alice, this.bob)) {
		List<Statement> list = Iterations.asList(statements);

		assertEquals(2, list.size());
	}
}
 

/**
 * Test method for
 * {@link org.eclipse.rdf4j.sail.memory.MemTripleSource#getStatements(org.eclipse.rdf4j.model.Resource, org.eclipse.rdf4j.model.IRI, org.eclipse.rdf4j.model.Value, org.eclipse.rdf4j.model.Resource[])}
 * .
 */
@Test
public final void testGetStatementsThreeContextsThreeContexts() throws Exception {
	loadTestData("/alp-testdata.ttl", this.alice, this.bob, this.mary);
	TripleSource source = getTripleSourceCommitted();

	try (CloseableIteration<? extends Statement, QueryEvaluationException> statements = source.getStatements(null,
			null, null, this.alice, this.bob, this.mary)) {
		List<Statement> list = Iterations.asList(statements);

		assertEquals(24, list.size());
	}
}
 

@Override
protected void service(TupleResultBuilder builder, RepositoryConnection con)
		throws RepositoryException, QueryResultHandlerException {
	for (Resource ctx : Iterations.asList(con.getContextIDs())) {
		builder.result(ctx);
	}
}
 

@Benchmark
public List<BindingSet> complexQueryWithoutCache() {

	try (SailRepositoryConnection connection = repoWithoutCache.getConnection()) {
		return Iterations.asList(connection
				.prepareTupleQuery(query5)
				.evaluate());
	}
}
 

/**
 * Test method for
 * {@link org.eclipse.rdf4j.sail.memory.MemTripleSource#getStatements(org.eclipse.rdf4j.model.Resource, org.eclipse.rdf4j.model.IRI, org.eclipse.rdf4j.model.Value, org.eclipse.rdf4j.model.Resource[])}
 * .
 */
@Test
public final void testGetStatementsOneContextPredicateOwlClassTwoContexts() throws Exception {
	loadTestData("/alp-testdata.ttl", this.alice);
	TripleSource source = getTripleSourceCommitted();

	try (CloseableIteration<? extends Statement, QueryEvaluationException> statements = source.getStatements(null,
			RDF.TYPE, OWL.CLASS, this.alice, this.bob)) {
		List<Statement> list = Iterations.asList(statements);

		assertEquals(4, list.size());
	}
}
 

@Benchmark
public List<BindingSet> groupByQuery() {

	try (SailRepositoryConnection connection = repository.getConnection()) {
		return Iterations.asList(connection
				.prepareTupleQuery(query1)
				.evaluate());
	}
}
 

/**
 * Test method for
 * {@link org.eclipse.rdf4j.sail.memory.MemTripleSource#getStatements(org.eclipse.rdf4j.model.Resource, org.eclipse.rdf4j.model.IRI, org.eclipse.rdf4j.model.Value, org.eclipse.rdf4j.model.Resource[])}
 * .
 */
@Test
public final void testGetStatementsOneContextPredicateExClassOneContext() throws Exception {
	loadTestData("/alp-testdata.ttl", this.alice);
	TripleSource source = getTripleSourceCommitted();

	try (CloseableIteration<? extends Statement, QueryEvaluationException> statements = source.getStatements(null,
			RDFS.SUBCLASSOF, f.createIRI(EX_NS, "A"), this.alice)) {
		List<Statement> list = Iterations.asList(statements);

		assertEquals(3, list.size());
	}
}
 

/**
 * Test method for
 * {@link org.eclipse.rdf4j.sail.memory.MemTripleSource#getStatements(org.eclipse.rdf4j.model.Resource, org.eclipse.rdf4j.model.IRI, org.eclipse.rdf4j.model.Value, org.eclipse.rdf4j.model.Resource[])}
 * .
 */
@Test
public final void testGetStatementsOneContextAllNull() throws Exception {
	loadTestData("/alp-testdata.ttl", this.alice);
	TripleSource source = getTripleSourceCommitted();

	try (CloseableIteration<? extends Statement, QueryEvaluationException> statements = source.getStatements(null,
			null, null)) {
		List<Statement> list = Iterations.asList(statements);

		assertEquals(8, list.size());
	}
}
 

@Test
public void testGetDataSailRepositoryStringObject() {
	SailRepository elasticsearchStore = new SailRepository(this.elasticsearchStore);
	try (SailRepositoryConnection connection = elasticsearchStore.getConnection()) {

		Literal label = vf.createLiteral("label1");
		connection.add(RDF.TYPE, RDFS.LABEL, label);

		List<? extends Statement> statements = Iterations
				.asList(connection.getStatements(null, null, label, true));

		System.out.println(Arrays.toString(statements.toArray()));

		assertEquals(1, statements.size());

		assertEquals(label, statements.get(0).getObject());

	}

}
 

@Test
public void testFederationFilter() throws Exception {

	assumeSparqlEndpoint();

	// load some data into endpoint1 and endpoint2
	loadDataSet(server.getRepository(1), "/tests/data/data1.ttl");
	loadDataSet(server.getRepository(2), "/tests/data/data2.ttl");

	final String repositoryId = "my-federation";
	final SPARQLEmbeddedServer rdf4jServer = (SPARQLEmbeddedServer) server;
	final File dataDir = rdf4jServer.getDataDir();
	String repoPath = "server/repositories";
	if (PlatformFactory.getPlatform().dataDirPreserveCase()) {
		repoPath = "Server/repositories";
	}
	final File repositoriesDir = new File(dataDir, repoPath);

	// preparation: add configuration files to the repository
	File fedXDataDir = new File(repositoriesDir, repositoryId);
	fedXDataDir.mkdirs();

	FileUtils.copyFile(toFile("/tests/rdf4jserver/config.ttl"), new File(fedXDataDir, "config.ttl"));

	String fedXSparqlUrl = rdf4jServer.getRepositoryUrl(repositoryId);
	SPARQLRepository repo = new SPARQLRepository(fedXSparqlUrl);
	repo.init();

	String query = "PREFIX foaf: <http://xmlns.com/foaf/0.1/>\n" +
			"PREFIX ns1: <http://namespace1.org/> "
			+ "SELECT * WHERE { "
			+ "?person a foaf:Person ."
			+ "?person foaf:name ?name .\n"
			+ "FILTER (?name IN ('Person1', 'Person2')) "
			+ "}";

	try (RepositoryConnection conn = repo.getConnection()) {
		try (TupleQueryResult tqr = conn.prepareTupleQuery(query).evaluate()) {

			List<BindingSet> res = Iterations.asList(tqr);
			assertContainsAll(res, "person",
					Sets.newHashSet(iri("http://namespace1.org/", "Person_1"),
							iri("http://namespace1.org/", "Person_2")));
		}
	}

	repo.shutDown();

	// temporary workaround: shutdown the federation repository explicitly here to
	// avoid a long running test. This is because the federation keeps an open
	// connection to other endpoints hosted in the same server, and the shutdown
	// sequence is arbitrary.
	Repository fedx = rdf4jServer.getRepositoryResolver().getRepository(repositoryId);
	fedx.shutDown();
}
 

@Setup(Level.Trial)
public void beforeClass() throws IOException {

	file = Files.newTemporaryFolder();

	repository = new SailRepository(new NativeStore(file, "spoc,ospc,psoc"));

	try (SailRepositoryConnection connection = repository.getConnection()) {
		connection.begin(IsolationLevels.NONE);
		connection.add(getResourceAsStream("benchmarkFiles/datagovbe-valid.ttl"), "", RDFFormat.TURTLE);
		connection.commit();
	}

	try (SailRepositoryConnection connection = repository.getConnection()) {

		statementList = Iterations.asList(connection.getStatements(null, RDF.TYPE, null, false));
	}

	System.gc();

}
 

@Test
public void testWithLocalRepositoryManager_CustomFederatedServiceResolver() throws Exception {

	addMemoryStore("repo1");
	addMemoryStore("repo2");
	addMemoryStore("serviceRepo");

	// register custom federated service resolver
	AbstractFederatedServiceResolver serviceResolver = new SPARQLServiceResolver() {
		@Override
		protected FederatedService createService(String serviceUrl) throws QueryEvaluationException {
			if (serviceUrl.equals("http://serviceRepo")) {
				Repository serviceRepo = repoManager.getRepository("serviceRepo");
				return new RepositoryFederatedService(serviceRepo, false);
			}
			throw new IllegalArgumentException("Service url cannot be resolved: " + serviceUrl);
		}
	};
	repoManager.externalResolver = serviceResolver;

	ValueFactory vf = SimpleValueFactory.getInstance();
	addData("repo1", Lists.newArrayList(
			vf.createStatement(vf.createIRI("http://ex.org/p1"), RDF.TYPE, FOAF.PERSON)));
	addData("repo2", Lists.newArrayList(
			vf.createStatement(vf.createIRI("http://ex.org/p2"), RDF.TYPE, FOAF.PERSON)));

	addData("serviceRepo", Lists.newArrayList(
			vf.createStatement(vf.createIRI("http://ex.org/p1"), FOAF.NAME, vf.createLiteral("Person 1")),
			vf.createStatement(vf.createIRI("http://ex.org/p2"), FOAF.NAME, vf.createLiteral("Person 2"))));

	FedXRepositoryConfig fedXRepoConfig = FedXRepositoryConfigBuilder.create()
			.withResolvableEndpoint(Arrays.asList("repo1", "repo2"))
			.build();
	repoManager.addRepositoryConfig(new RepositoryConfig("federation", fedXRepoConfig));

	Repository repo = repoManager.getRepository("federation");

	try (RepositoryConnection conn = repo.getConnection()) {

		TupleQuery tq = conn.prepareTupleQuery(
				"SELECT * WHERE { ?person a ?PERSON . { SERVICE <http://serviceRepo> { ?person ?NAME ?name} } }");
		tq.setBinding("PERSON", FOAF.PERSON);
		tq.setBinding("NAME", FOAF.NAME);

		List<BindingSet> bindings = Iterations.asList(tq.evaluate());
		Assertions.assertEquals(2, bindings.size()); // two persons

		BindingSet b1 = bindings.get(0);
		BindingSet b2 = bindings.get(1);

		if (b1.getValue("person").equals(vf.createIRI("http://ex.org/p1"))) {
			Assertions.assertEquals(vf.createLiteral("Person 1"), b1.getValue("name"));
			Assertions.assertEquals(vf.createLiteral("Person 2"), b2.getValue("name"));
		} else {
			Assertions.assertEquals(vf.createLiteral("Person 2"), b1.getValue("name"));
			Assertions.assertEquals(vf.createLiteral("Person 1"), b2.getValue("name"));
		}

	}

}
 

@Test
public void testGetDataSailRepositoryLang() {
	SailRepository elasticsearchStore = new SailRepository(this.elasticsearchStore);
	try (SailRepositoryConnection connection = elasticsearchStore.getConnection()) {

		Literal label = vf.createLiteral("norsk bokmål", "nb");
		connection.add(RDF.TYPE, RDFS.LABEL, label);

		List<? extends Statement> statements = Iterations
				.asList(connection.getStatements(null, null, label, true));

		System.out.println(Arrays.toString(statements.toArray()));

		assertEquals(1, statements.size());

		assertEquals(label, statements.get(0).getObject());

	}

}
 

@Test
public void testGetDataSailRepositoryBNodeObject() {
	SailRepository elasticsearchStore = new SailRepository(this.elasticsearchStore);
	try (SailRepositoryConnection connection = elasticsearchStore.getConnection()) {

		BNode bNode = vf.createBNode();
		connection.add(bNode, RDF.TYPE, bNode);

		List<? extends Statement> statements = Iterations
				.asList(connection.getStatements(bNode, RDF.TYPE, bNode, true));

		System.out.println(Arrays.toString(statements.toArray()));

		assertEquals(1, statements.size());

		assertEquals(bNode, statements.get(0).getSubject());
		assertEquals(bNode, statements.get(0).getObject());

	}

}