com.ewcms.publication.freemarker.directive.IncludeDirectiveTest.java Source code

Introduction

Here is the source code for com.ewcms.publication.freemarker.directive.IncludeDirectiveTest.java

Source

/**
 * Copyright (c)2010-2011 Enterprise Website Content Management System(EWCMS), All rights reserved.
 * EWCMS PROPRIETARY/CONFIDENTIAL. Use is subject to license terms.
 * http://www.ewcms.com
 */

package com.ewcms.publication.freemarker.directive;

import static org.mockito.Matchers.any;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

import java.util.HashMap;
import java.util.Map;

import org.apache.commons.lang.StringUtils;
import org.junit.Assert;
import org.junit.Test;

import com.ewcms.core.site.model.Channel;
import com.ewcms.core.site.model.Site;
import com.ewcms.publication.freemarker.FreemarkerTest;
import com.ewcms.publication.freemarker.GlobalVariable;
import com.ewcms.publication.service.ChannelPublishServiceable;
import com.ewcms.publication.service.TemplatePublishServiceable;

import freemarker.template.Configuration;
import freemarker.template.Template;

/**
 * IncludeDirective?
 *
 * @author wangwei
 */
public class IncludeDirectiveTest extends FreemarkerTest {

    @Test
    public void testGetUniqueTemplatePath() {
        IncludeDirective directive = new IncludeDirective(null, null);
        String path = "/home/test.html";
        Integer siteId = new Integer(2);

        String uPath = directive.getUniqueTemplatePath(siteId, path);
        Assert.assertEquals("2/home/test.html", uPath);
        path = "home/test.html";
        uPath = directive.getUniqueTemplatePath(siteId, path);
        Assert.assertEquals("2/home/test.html", uPath);
    }

    @Override
    protected void currentConfiguration(Configuration cfg) {
        ChannelPublishServiceable channelLoaderService = mock(ChannelPublishServiceable.class);
        Channel channel = new Channel();
        channel.setId(1);
        when(channelLoaderService.getChannelByUrlOrPath(any(Integer.class), any(String.class))).thenReturn(channel);
        TemplatePublishServiceable templateLoaderService = mock(TemplatePublishServiceable.class);
        when(templateLoaderService.getUniquePathOfChannelTemplate(any(Integer.class), any(Integer.class),
                any(String.class))).thenReturn("2/1/include.html");
        IncludeDirective directive = new IncludeDirective(channelLoaderService, templateLoaderService);
        cfg.setSharedVariable("include", directive);
    }

    /**
     * ?
     * 
     * @param name
     *            ???
     * @return
     */
    private String getTemplatePath(String name) {
        return String.format("directive/include/%s", name);
    }

    @Test
    public void testPathTemplate() throws Exception {
        Template template = cfg.getTemplate(getTemplatePath("path.html"));
        Map<String, Object> params = new HashMap<String, Object>();
        Site site = new Site();
        site.setId(2);
        params.put(GlobalVariable.SITE.toString(), site);
        String value = process(template, params);
        value = StringUtils.deleteWhitespace(value);
        Assert.assertEquals("test-path-include", value);
    }

    @Test
    public void testChannelTemplate() throws Exception {
        Template template = cfg.getTemplate(getTemplatePath("channel.html"));

        Map<String, Object> params = new HashMap<String, Object>();
        Site site = new Site();
        site.setId(2);
        params.put(GlobalVariable.SITE.toString(), site);
        Channel channel = new Channel();
        channel.setId(1);
        params.put(GlobalVariable.CHANNEL.toString(), channel);
        String value = process(template, params);
        value = StringUtils.deleteWhitespace(value);
        Assert.assertEquals("test-channel-includetest-channel-includetest-channel-include", value);
    }
}