org.artifactory.descriptor.reader.CentralConfigReaderTest.java Source code

Introduction

Here is the source code for org.artifactory.descriptor.reader.CentralConfigReaderTest.java

Source

/*
 * Artifactory is a binaries repository manager.
 * Copyright (C) 2012 JFrog Ltd.
 *
 * Artifactory is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Artifactory is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with Artifactory.  If not, see <http://www.gnu.org/licenses/>.
 */

package org.artifactory.descriptor.reader;

import org.apache.commons.io.FileUtils;
import org.artifactory.common.property.ArtifactorySystemProperties;
import org.artifactory.descriptor.config.CentralConfigDescriptor;
import org.artifactory.descriptor.repo.RemoteRepoDescriptor;
import org.artifactory.jaxb.JaxbHelper;
import org.artifactory.test.ArtifactoryHomeBoundTest;
import org.artifactory.test.TestUtils;
import org.artifactory.util.ResourceUtils;
import org.artifactory.version.ArtifactoryConfigVersion;
import org.testng.Assert;
import org.testng.annotations.Test;

import java.io.File;
import java.util.Collection;
import java.util.Map;
import java.util.Properties;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;

import static org.artifactory.common.ConstantValues.substituteRepoKeys;
import static org.testng.Assert.*;

/**
 * Test the {@link CentralConfigReader}
 *
 * @author Tomer Cohen
 */
@Test

public class CentralConfigReaderTest extends ArtifactoryHomeBoundTest {

    private String failureMessage;

    public void readV7Config() throws Exception {
        File oldConfigFile = ResourceUtils.getResourceAsFile("/config/install/config.1.4.7.xml");
        CentralConfigDescriptor newConfig = new CentralConfigReader().read(oldConfigFile);
        Map<String, RemoteRepoDescriptor> descriptorOrderedMap = newConfig.getRemoteRepositoriesMap();
        Assert.assertEquals(descriptorOrderedMap.size(), 12, "Should contain 12 remote repository");
    }

    public void readAllConfigFiles() throws Exception {
        Properties propTest = new Properties();
        propTest.setProperty(substituteRepoKeys.getPropertyName() + "3rdp-releases", "third-party-releases");
        propTest.setProperty(substituteRepoKeys.getPropertyName() + "3rdp-snapshots", "third-party-snapshots");
        propTest.setProperty(substituteRepoKeys.getPropertyName() + "3rd-party", "third-party");
        // load the repo key substitute
        Map<String, String> subs = (Map<String, String>) TestUtils.invokeStaticMethod(
                ArtifactorySystemProperties.class, "fillRepoKeySubstitute", new Class[] { Properties.class },
                new Object[] { propTest });
        assertEquals(subs.size(), 3);
        TestUtils.setField(getBound().getArtifactoryProperties(), "substituteRepoKeys", subs);
        File backupDirs = ResourceUtils.getResourceAsFile("/config");
        Collection<File> oldArtifactoryConfigs = FileUtils.listFiles(backupDirs, new String[] { "xml" }, true);
        assertTrue(oldArtifactoryConfigs.size() > 10, "Where are all my test files??");

        // run in parallel for better speed
        ExecutorService executorService = Executors.newFixedThreadPool(Runtime.getRuntime().availableProcessors());
        for (File oldArtifactoryConfig : oldArtifactoryConfigs) {
            executorService.submit(new ConfigReadWriteTester(oldArtifactoryConfig));
        }
        executorService.shutdown();
        //executorService.awaitTermination(15, TimeUnit.SECONDS);
        executorService.awaitTermination(1500, TimeUnit.SECONDS);

        assertNull(failureMessage, failureMessage);
    }

    private class ConfigReadWriteTester implements Runnable {
        private final File configFile;

        private ConfigReadWriteTester(File configFile) {
            this.configFile = configFile;
        }

        @Override
        public void run() {
            try {
                CentralConfigReader centralConfigReader = new CentralConfigReader();
                CentralConfigDescriptor newConfig = centralConfigReader.read(configFile);
                ArtifactoryConfigVersion configVersion = ArtifactoryConfigVersion
                        .getConfigVersion(JaxbHelper.toXml(newConfig));
                assertNotNull(configVersion,
                        "Null value returned from config reader for file " + configFile.getAbsolutePath());
                assertTrue(configVersion.isCurrent(), "Artifactory config version is not up to date");
            } catch (Throwable t) {
                t.printStackTrace();
                failureMessage = "Failed to convert " + configFile + ": " + t.getMessage();
            }
        }
    }
}