org.obeonetwork.m2doc.properties.test.TemplateInfoTest.java Source code

Introduction

Here is the source code for org.obeonetwork.m2doc.properties.test.TemplateInfoTest.java

Source

/*******************************************************************************
 *  Copyright (c) 2016 Obeo. 
 *  All rights reserved. This program and the accompanying materials
 *  are made available under the terms of the Eclipse Public License v1.0
 *  which accompanies this distribution, and is available at
 *  http://www.eclipse.org/legal/epl-v10.html
 *   
 *   Contributors:
 *       Obeo - initial API and implementation
 *  
 *******************************************************************************/
package org.obeonetwork.m2doc.properties.test;

import java.io.FileInputStream;
import java.io.IOException;
import java.util.List;
import java.util.Map;

import org.apache.poi.openxml4j.exceptions.InvalidFormatException;
import org.apache.poi.openxml4j.opc.OPCPackage;
import org.apache.poi.xwpf.usermodel.XWPFDocument;
import org.junit.Test;
import org.obeonetwork.m2doc.properties.TemplateInfo;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;

/**
 * Test class for the {@link TemplateInfo}.
 * 
 * @author Romain Guider
 */
public class TemplateInfoTest {

    @Test
    public void testServiceToken() throws IOException, InvalidFormatException {
        FileInputStream is = new FileInputStream("templates/propertiesTest.docx");
        OPCPackage oPackage = OPCPackage.open(is);
        XWPFDocument document = new XWPFDocument(oPackage);
        TemplateInfo info = new TemplateInfo(document);
        List<String> serviceTokens = info.getServiceTokens();
        assertTrue(serviceTokens.contains("token1"));
        assertTrue(serviceTokens.contains("token2"));
    }

    @Test
    public void testVariableMap() throws IOException, InvalidFormatException {
        FileInputStream is = new FileInputStream("templates/propertiesTest.docx");
        OPCPackage oPackage = OPCPackage.open(is);
        XWPFDocument document = new XWPFDocument(oPackage);
        TemplateInfo info = new TemplateInfo(document);
        Map<String, String> variables = info.getVariables();
        assertEquals("database.Table", variables.get("variable1"));
        assertEquals("database.Column", variables.get("variable2"));
    }
}