com.igormaznitsa.jcp.AbstractSpyPreprocessorContextTest.java Source code

Introduction

Here is the source code for com.igormaznitsa.jcp.AbstractSpyPreprocessorContextTest.java

Source

/*
 * Copyright 2016 Igor Maznitsa.
 *
 * 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.igormaznitsa.jcp;

import java.io.File;
import org.apache.commons.io.FileUtils;
import org.apache.commons.io.FilenameUtils;
import org.junit.AfterClass;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.rules.TemporaryFolder;
import org.junit.runner.RunWith;
import org.powermock.api.mockito.PowerMockito;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;
import com.igormaznitsa.jcp.containers.FileInfoContainer;
import com.igormaznitsa.jcp.context.PreprocessingState;
import com.igormaznitsa.jcp.context.PreprocessorContext;
import com.igormaznitsa.jcp.utils.ResetablePrinter;
import static org.junit.Assert.assertEquals;

@RunWith(PowerMockRunner.class)
@PrepareForTest({ PreprocessorContext.class, PreprocessingState.class })
public abstract class AbstractSpyPreprocessorContextTest {

    protected static TemporaryFolder destinationFolder;

    public interface ContextDataProvider {
        boolean getAllowSpaceBeforeDirectiveFlag();
    }

    @BeforeClass
    public static void prepareClassTests() throws Exception {
        destinationFolder = new TemporaryFolder(new File("./"));
        destinationFolder.create();
    }

    @AfterClass
    public static void doJanitor() throws Exception {
        destinationFolder.delete();
    }

    @Before
    public void beforeTest() throws Exception {
        FileUtils.cleanDirectory(destinationFolder.getRoot());
    }

    protected String getCurrentTestFolder() {
        final String testFolder = FilenameUtils.normalizeNoEndSeparator(System.getProperty("test.folder"));
        final String fullClassPath = this.getClass().getName().replace('.', File.separatorChar);
        return FilenameUtils.normalize(testFolder + File.separator
                + fullClassPath.substring(0, fullClassPath.lastIndexOf(File.separatorChar)));
    }

    protected File getDestinationFolder() {
        return destinationFolder.getRoot();
    }

    protected void assertDestinationFolderEmpty() throws Exception {
        assertEquals("Destination folder must be enpty", 0, destinationFolder.getRoot().list().length);
    }

    protected PreprocessorContext preparePreprocessorContext(final String sourceFolder) throws Exception {
        return this.preparePreprocessorContext(sourceFolder, new ContextDataProvider() {
            @Override
            public boolean getAllowSpaceBeforeDirectiveFlag() {
                return false;
            }
        });
    }

    protected PreprocessorContext preparePreprocessorContext(final String sourceFolder,
            final ContextDataProvider provider) throws Exception {
        final PreprocessorContext preprocessorcontext = PowerMockito.spy(new PreprocessorContext());
        final PreprocessingState stateMock = PowerMockito.mock(PreprocessingState.class);
        PowerMockito.when(stateMock.getRootFileInfo())
                .thenReturn(new FileInfoContainer(new File("src/fake.java"), "fake.java", false));
        PowerMockito.when(stateMock.getPrinter()).thenReturn(new ResetablePrinter(10));

        PowerMockito.when(preprocessorcontext.getPreprocessingState()).thenReturn(stateMock);

        preprocessorcontext.setAllowWhitespace(provider.getAllowSpaceBeforeDirectiveFlag());
        preprocessorcontext.setSourceDirectories(sourceFolder);
        preprocessorcontext.setDestinationDirectory(destinationFolder.getRoot().getAbsolutePath());

        return preprocessorcontext;
    }

}