org.wikidata.wdtk.datamodel.json.jackson.JsonSerializerTest.java Source code

Introduction

Here is the source code for org.wikidata.wdtk.datamodel.json.jackson.JsonSerializerTest.java

Source

package org.wikidata.wdtk.datamodel.json.jackson;

/*
 * #%L
 * Wikidata Toolkit Data Model
 * %%
 * Copyright (C) 2014 - 2015 Wikidata Toolkit Developers
 * %%
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * #L%
 */

import static org.junit.Assert.assertEquals;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Collections;

import org.junit.Test;
import org.wikidata.wdtk.datamodel.helpers.Datamodel;
import org.wikidata.wdtk.datamodel.implementation.DataObjectFactoryImplTest;
import org.wikidata.wdtk.datamodel.interfaces.DatatypeIdValue;
import org.wikidata.wdtk.datamodel.interfaces.EntityDocument;
import org.wikidata.wdtk.datamodel.interfaces.EntityIdValue;
import org.wikidata.wdtk.datamodel.interfaces.ItemDocument;
import org.wikidata.wdtk.datamodel.interfaces.MonolingualTextValue;
import org.wikidata.wdtk.datamodel.interfaces.PropertyDocument;
import org.wikidata.wdtk.datamodel.interfaces.SiteLink;
import org.wikidata.wdtk.datamodel.interfaces.StatementGroup;

import com.fasterxml.jackson.databind.MappingIterator;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.ObjectReader;

public class JsonSerializerTest {

    @Test
    public void testSerializer() throws IOException {
        ByteArrayOutputStream out = new ByteArrayOutputStream();
        JsonSerializer serializer = new JsonSerializer(out);

        ItemDocument id1 = Datamodel.makeItemDocument(DataObjectFactoryImplTest.getTestItemIdValue(1),
                Collections
                        .<MonolingualTextValue>singletonList(Datamodel.makeMonolingualTextValue("Label1", "lang1")),
                Collections.<MonolingualTextValue>emptyList(), Collections.<MonolingualTextValue>emptyList(),
                DataObjectFactoryImplTest.getTestStatementGroups(1, 24, 1, EntityIdValue.ET_ITEM),
                Collections.<String, SiteLink>emptyMap());
        ItemDocument id2 = Datamodel.makeItemDocument(DataObjectFactoryImplTest.getTestItemIdValue(2),
                Collections.<MonolingualTextValue>emptyList(), Collections.<MonolingualTextValue>emptyList(),
                Collections.<MonolingualTextValue>emptyList(),
                DataObjectFactoryImplTest.getTestStatementGroups(2, 23, 1, EntityIdValue.ET_ITEM),
                Collections.<String, SiteLink>singletonMap("enwiki",
                        Datamodel.makeSiteLink("Title2", "enwiki", Collections.<String>emptyList())));
        PropertyDocument pd1 = Datamodel.makePropertyDocument(DataObjectFactoryImplTest.getTestPropertyIdValue(1),
                Collections.<MonolingualTextValue>emptyList(), Collections.<MonolingualTextValue>emptyList(),
                Collections
                        .<MonolingualTextValue>singletonList(Datamodel.makeMonolingualTextValue("Alias1", "lang1")),
                Collections.<StatementGroup>emptyList(),
                Datamodel.makeDatatypeIdValue(DatatypeIdValue.DT_COMMONS_MEDIA));

        serializer.open();
        serializer.processItemDocument(id1);
        serializer.processItemDocument(id2);
        serializer.processPropertyDocument(pd1);
        serializer.close();

        ArrayList<EntityDocument> inputDocuments = new ArrayList<>();
        inputDocuments.add(id1);
        inputDocuments.add(id2);
        inputDocuments.add(pd1);

        ArrayList<EntityDocument> outputDocuments = new ArrayList<>();

        ObjectMapper mapper = new ObjectMapper();
        ObjectReader documentReader = mapper.reader(JacksonTermedStatementDocument.class);

        MappingIterator<JacksonTermedStatementDocument> documentIterator = documentReader
                .readValues(out.toString());
        while (documentIterator.hasNextValue()) {
            JacksonTermedStatementDocument document = documentIterator.nextValue();
            document.setSiteIri("foo:");
            outputDocuments.add(document);
        }
        documentIterator.close();

        for (int i = 0; i < outputDocuments.size(); i++) {
            assertEquals(inputDocuments.get(i), outputDocuments.get(i));
        }
        assertEquals(serializer.getEntityDocumentCount(), 3);
    }
}