001/*
002 * Copyright 2015 DuraSpace, Inc.
003 *
004 * Licensed under the Apache License, Version 2.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 *
008 *     http://www.apache.org/licenses/LICENSE-2.0
009 *
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 */
016package org.fcrepo.kernel.api;
017
018import static java.util.Arrays.asList;
019import static com.hp.hpl.jena.graph.NodeFactory.createURI;
020import static org.junit.Assert.assertEquals;
021
022import java.util.List;
023
024import com.hp.hpl.jena.graph.Node;
025import com.hp.hpl.jena.graph.Triple;
026import com.hp.hpl.jena.rdf.model.Model;
027
028import org.junit.Test;
029
030/**
031 * @author acoburn
032 */
033public class RdfCollectorsTest {
034
035    @Test
036    public void streamToModel() {
037        final Node subject = createURI("subject");
038        final Node prop1 = createURI("prop1");
039        final Node prop2 = createURI("prop2");
040        final List<Triple> triples = asList(
041                new Triple(subject, prop1, createURI("obj1")),
042                new Triple(subject, prop1, createURI("obj2")),
043                new Triple(subject, prop1, createURI("obj3")),
044                new Triple(subject, prop2, createURI("obj1")),
045                new Triple(subject, prop2, createURI("obj2")),
046                new Triple(subject, prop2, createURI("obj3")));
047
048        final Model filtered = triples.stream().filter(x -> x.getPredicate().equals(prop1))
049                .collect(RdfCollectors.toModel());
050
051        assertEquals(3, filtered.size());
052    }
053}
054