com.intellij.struts2.dom.struts.StrutsHighlightingSpringTest.java Source code

Introduction

Here is the source code for com.intellij.struts2.dom.struts.StrutsHighlightingSpringTest.java

Source

/*
 * Copyright 2011 The authors
 * 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.
 */

package com.intellij.struts2.dom.struts;

import com.intellij.facet.FacetManager;
import com.intellij.openapi.application.Result;
import com.intellij.openapi.command.WriteCommandAction;
import com.intellij.openapi.util.Condition;
import com.intellij.openapi.vfs.VirtualFile;
import com.intellij.spring.facet.SpringFacet;
import com.intellij.spring.facet.SpringFacetConfiguration;
import com.intellij.spring.facet.SpringFacetType;
import com.intellij.spring.facet.SpringFileSet;
import com.intellij.testFramework.builders.JavaModuleFixtureBuilder;
import com.intellij.util.containers.ContainerUtil;
import junit.framework.Assert;
import org.apache.commons.collections.CollectionUtils;
import org.jetbrains.annotations.NonNls;
import org.jetbrains.annotations.NotNull;

import java.util.Arrays;
import java.util.List;
import java.util.Set;

/**
 * Tests highlighting with Spring plugin.
 */
public class StrutsHighlightingSpringTest extends BasicStrutsHighlightingTestCase<JavaModuleFixtureBuilder> {

    @NonNls
    private static final String SPRING_XML = "spring.xml";

    @Override
    @NotNull
    protected String getTestDataLocation() {
        return "strutsXmlHighlightingSpring";
    }

    @Override
    protected boolean hasJavaSources() {
        return true;
    }

    @Override
    protected void configureModule(final JavaModuleFixtureBuilder moduleBuilder) throws Exception {
        super.configureModule(moduleBuilder);

        addLibrary(moduleBuilder, "spring", "spring.jar");
        addLibrary(moduleBuilder, "struts2-spring-plugin", STRUTS2_SPRING_PLUGIN_JAR);
    }

    public void testStrutsSpringHighlighting() throws Throwable {
        createSpringFileSet(SPRING_XML);

        performHighlightingTest("struts-spring.xml");
    }

    public void testStrutsSpringCompletionVariants() throws Throwable {
        @NonNls
        final String strutsXml = "struts-completionvariants-spring.xml";
        createStrutsFileSet(strutsXml);

        createSpringFileSet(SPRING_XML);

        // TODO <alias> does not appear here, see com.intellij.spring.impl.SpringModelImpl#myOwnBeans
        final List<String> variants = myFixture.getCompletionVariants(strutsXml);

        Assert.assertTrue(CollectionUtils.isSubCollection(
                Arrays.asList("MyClass", "bean1", "bean2", "springInterceptor", "springResultType"), variants));
    }

    public void testStrutsSpringCompletionVariantsSubclass() throws Throwable {
        @NonNls
        final String strutsXml = "struts-completionvariants-subclass-spring.xml";
        createStrutsFileSet(strutsXml);

        createSpringFileSet(SPRING_XML);

        final List<String> variants = myFixture.getCompletionVariants(strutsXml);

        final List<String> springCompletionVariants = filterSpringBeanCompletionVariants(variants);

        assertSameElements(springCompletionVariants, "springInterceptor");
    }

    /**
     * Returns all completions variants without '.', i.e. all Spring bean names.
     *
     * @param variants Original variants.
     * @return Remaining (Spring Bean) variants.
     */
    private List<String> filterSpringBeanCompletionVariants(final List<String> variants) {
        return ContainerUtil.findAll(variants, new Condition<String>() {
            @Override
            public boolean value(final String s) {
                return !s.contains(".");
            }
        });
    }

    // stuff below is Spring related ===============================================

    protected void createSpringFileSet(final String... springXmlPaths) throws Throwable {
        final SpringFacet springFacet = createSpringFacet();

        final SpringFacetConfiguration configuration = springFacet.getConfiguration();
        final Set<SpringFileSet> list = configuration.getFileSets();
        @NonNls
        final SpringFileSet fileSet = new SpringFileSet("", "default", configuration);
        list.add(fileSet);

        for (final String springXmlPath : springXmlPaths) {
            myFixture.copyFileToProject(springXmlPath);
            final VirtualFile file = myFixture.getTempDirFixture().getFile(springXmlPath);
            assert file != null;
            fileSet.addFile(file);
        }

        springFacet.getConfiguration().setModified();
    }

    protected SpringFacet createSpringFacet() {
        return new WriteCommandAction<SpringFacet>(myFixture.getProject()) {
            @Override
            protected void run(final Result<SpringFacet> result) throws Throwable {
                final SpringFacetType springFacetType = SpringFacetType.getInstance();
                final SpringFacet facet = FacetManager.getInstance(myModule).addFacet(springFacetType, "spring",
                        null);
                result.setResult(facet);
            }
        }.execute().throwException().getResultObject();
    }

}