-
Notifications
You must be signed in to change notification settings - Fork 12
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
60 changed files
with
1,046 additions
and
486 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
129 changes: 129 additions & 0 deletions
129
core/src/main/java/org/arquillian/smart/testing/configuration/ConfigurationLoader.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,129 @@ | ||
| package org.arquillian.smart.testing.configuration; | ||
|
|
||
| import java.io.File; | ||
| import java.io.FileReader; | ||
| import java.io.IOException; | ||
| import java.io.InputStream; | ||
| import java.io.UncheckedIOException; | ||
| import java.nio.file.Files; | ||
| import java.nio.file.Path; | ||
| import java.nio.file.Paths; | ||
| import java.util.Arrays; | ||
| import java.util.HashMap; | ||
| import java.util.Map; | ||
| import org.arquillian.smart.testing.hub.storage.local.LocalStorage; | ||
| import org.arquillian.smart.testing.logger.Log; | ||
| import org.arquillian.smart.testing.logger.Logger; | ||
| import org.yaml.snakeyaml.Yaml; | ||
|
|
||
| import static org.arquillian.smart.testing.configuration.ObjectMapper.mapToObject; | ||
|
|
||
| public class ConfigurationLoader { | ||
|
|
||
| public static final String SMART_TESTING_YML = "smart-testing.yml"; | ||
| public static final String SMART_TESTING_YAML = "smart-testing.yaml"; | ||
| private static final Logger logger = Log.getLogger(); | ||
|
|
||
| public static Configuration load() { | ||
| return load(Paths.get("").toAbsolutePath().toFile()); | ||
| } | ||
|
|
||
| public static Configuration load(File projectDir) { | ||
| final File[] files = | ||
| projectDir.listFiles((dir, name) -> name.equals(SMART_TESTING_YML) || name.equals(SMART_TESTING_YAML)); | ||
|
|
||
| if (files == null) { | ||
| throw new RuntimeException("I/O errors occurs while listing dir " + projectDir); | ||
| } | ||
|
|
||
| Map<String, Object> yamlConfiguration = new HashMap<>(0); | ||
| if (files.length == 0) { | ||
| logger.info("Config file `" + SMART_TESTING_YAML + "` OR `" + SMART_TESTING_YML + "` is not found. " | ||
| + "Using system properties to load configuration for smart testing."); | ||
| } else { | ||
| yamlConfiguration = getConfigParametersFromFile(getConfigurationFilePath(files)); | ||
| } | ||
|
|
||
| final Object strategiesConfiguration = yamlConfiguration.get("strategiesConfiguration"); | ||
|
|
||
| final Configuration configuration = parseConfiguration(yamlConfiguration); | ||
| if (strategiesConfiguration != null) { | ||
| configuration.setStrategiesConfig((Map<String, Object>) strategiesConfiguration); | ||
| } | ||
|
|
||
| return configuration; | ||
| } | ||
|
|
||
| private static Map<String, Object> getConfigParametersFromFile(Path filePath) { | ||
| try (InputStream io = Files.newInputStream(filePath)) { | ||
| final Yaml yaml = new Yaml(); | ||
| Map<String, Object> yamlConfig = yaml.load(io); | ||
| if (yamlConfig == null) { | ||
| logger.warn(String.format("The configuration file %s is empty.", filePath)); | ||
| return new HashMap<>(); | ||
| } else { | ||
| return yamlConfig; | ||
| } | ||
| } catch (IOException e) { | ||
| throw new UncheckedIOException(e); | ||
| } | ||
| } | ||
|
|
||
| public static Configuration load(File projectDir, String... strategies) { | ||
| final Configuration configuration = load(projectDir); | ||
| configuration.loadStrategyConfigurations(strategies); | ||
|
|
||
| return configuration; | ||
| } | ||
|
|
||
| public static Configuration loadPrecalculated(File projectDir) { | ||
| final File configFile = | ||
| new LocalStorage(projectDir).duringExecution().temporary().file(SMART_TESTING_YML).getFile(); | ||
| if (configFile.exists()) { | ||
| return loadConfigurationFromFile(configFile); | ||
| } else { | ||
| return load(projectDir); | ||
| } | ||
| } | ||
|
|
||
| static Configuration loadConfigurationFromFile(File configFile) { | ||
| try (FileReader fileReader = new FileReader(configFile)) { | ||
| final Yaml yaml = new Yaml(); | ||
| return yaml.loadAs(fileReader, Configuration.class); | ||
| } catch (IOException e) { | ||
| throw new RuntimeException("Failed to load configuration from file " + configFile, e); | ||
| } | ||
| } | ||
|
|
||
| // testing | ||
| static Configuration load(Path path) { | ||
| try (InputStream io = Files.newInputStream(path)) { | ||
| final Yaml yaml = new Yaml(); | ||
| Map<String, Object> yamlConfiguration = (Map<String, Object>) yaml.load(io); | ||
| return parseConfiguration(yamlConfiguration); | ||
| } catch (IOException e) { | ||
| throw new UncheckedIOException(e); | ||
| } | ||
| } | ||
|
|
||
| private static Path getConfigurationFilePath(File[] files) { | ||
| if (files.length == 1) { | ||
| final File configFile = files[0]; | ||
| logger.info("Using configuration from " + configFile.getName()); | ||
| return configFile.toPath(); | ||
| } | ||
|
|
||
| logger.warn("Found multiple config files with supported names: " + SMART_TESTING_YAML + ", " + SMART_TESTING_YML); | ||
| logger.warn("Using configuration from " + SMART_TESTING_YML); | ||
|
|
||
| return Arrays.stream(files) | ||
| .filter(file -> file.getName().equals(SMART_TESTING_YML)) | ||
| .map(File::toPath) | ||
| .findFirst() | ||
| .get(); | ||
| } | ||
|
|
||
| private static Configuration parseConfiguration(Map<String, Object> yamlConfiguration) { | ||
| return mapToObject(Configuration.class, yamlConfiguration); | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
8 changes: 8 additions & 0 deletions
8
core/src/main/java/org/arquillian/smart/testing/spi/StrategyConfiguration.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,8 @@ | ||
| package org.arquillian.smart.testing.spi; | ||
|
|
||
| import org.arquillian.smart.testing.configuration.ConfigurationSection; | ||
|
|
||
| public interface StrategyConfiguration extends ConfigurationSection { | ||
|
|
||
| String name(); | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.