mockit.internal.startup.StartupConfiguration Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jmockit Show documentation

JMockit is a Java toolkit for automated developer testing. It contains mocking/faking APIs and a code coverage tool, supporting both JUnit and TestNG. The mocking APIs allow all kinds of Java code, without testability restrictions, to be tested in isolation from selected dependencies.

There is a newer version: 1.49

Show newest version

/*
 * Copyright (c) 2006 Rogério Liesenfeld
 * This file is subject to the terms of the MIT license (see LICENSE.txt).
 */
package mockit.internal.startup;

import java.io.*;
import java.net.*;
import java.util.*;
import java.util.Map.*;
import java.util.regex.Pattern;
import javax.annotation.*;

final class StartupConfiguration
{
   @Nonnull private static final Collection NO_VALUES = Collections.emptyList();
   @Nonnull private static final Pattern COMMA_OR_SPACES = Pattern.compile("\\s*,\\s*|\\s+");

   @Nonnull private final Properties config;
   @Nonnull final Collection mockClasses;

   StartupConfiguration() throws IOException
   {
      config = new Properties();
      loadJMockitPropertiesFilesFromClasspath();
      loadJMockitPropertiesIntoSystemProperties();
      mockClasses = getMultiValuedProperty("mockups");
   }

   @SuppressWarnings("ThrowFromFinallyBlock")
   private void loadJMockitPropertiesFilesFromClasspath() throws IOException
   {
      ClassLoader loader = Thread.currentThread().getContextClassLoader();

      if (loader == null) {
         loader = getClass().getClassLoader();
      }

      Enumeration allFiles = loader.getResources("jmockit.properties");
      int numFiles = 0;

      while (allFiles.hasMoreElements()) {
         URL url = allFiles.nextElement();
         InputStream propertiesFile = url.openStream();

         if (numFiles == 0) {
            try { config.load(propertiesFile); } finally { propertiesFile.close(); }
         }
         else {
            Properties properties = new Properties();
            try { properties.load(propertiesFile); } finally { propertiesFile.close(); }
            addPropertyValues(properties);
         }

         numFiles++;
      }
   }

   @SuppressWarnings("UseOfPropertiesAsHashtable")
   private void addPropertyValues(@Nonnull Map propertiesToAdd)
   {
      for (Entry propertyToAdd : propertiesToAdd.entrySet()) {
         Object key = propertyToAdd.getKey();
         String valueToAdd = (String) propertyToAdd.getValue();
         String existingValue = (String) config.get(key);
         String newValue;

         if (existingValue == null || existingValue.isEmpty()) {
            newValue = valueToAdd;
         }
         else {
            newValue = existingValue + ' ' + valueToAdd;
         }

         config.put(key, newValue);
      }
   }

   private void loadJMockitPropertiesIntoSystemProperties()
   {
      Map systemProperties = System.getProperties();

      for (Entry property : config.entrySet()) {
         String key = (String) property.getKey();
         String name = key.startsWith("jmockit-") ? key : "jmockit-" + key;

         if (!systemProperties.containsKey(name)) {
            systemProperties.put(name, property.getValue());
         }
      }
   }

   @Nonnull
   private static Collection getMultiValuedProperty(@Nonnull String key)
   {
      String commaOrSpaceSeparatedValues = System.getProperty(key);
      
      if (commaOrSpaceSeparatedValues == null) {
         return NO_VALUES;
      }

      List allValues = Arrays.asList(COMMA_OR_SPACES.split(commaOrSpaceSeparatedValues));
      Set uniqueValues = new HashSet(allValues);
      uniqueValues.remove("");

      return uniqueValues;
   }
}