/** * Construct configuration from raw yaml string. * * @param clazz * @param resourcePath * @throws Exception */ public YamlConfiguration(Class<?> clazz, String resourcePath) throws IOException { this(new YamlFile(clazz, resourcePath)); }
@Test public void testCanGetFormatted() throws IOException { assertEquals(FORMATTED, yamlFile.formatted()); } }
@Test public void testCanGetOriginal() throws IOException { String expected = Files.asCharSource( new File(YamlFileTest.class.getResource(YAMLCONFIGURATION_YAML).getFile()), StandardCharsets.UTF_8) .read(); assertEquals(expected, yamlFile.original()); }
@Test @SuppressWarnings("unchecked") public void testCreateDataTree() throws IOException { Object data = yamlFile.dataTree(); assertTrue(data instanceof Map); Map<String, Object> dataTree = (Map<String, Object>) data; assertNotNull(dataTree); assertTrue(dataTree.containsKey("example")); assertTrue(dataTree.containsKey("collectionreader")); assertTrue(dataTree.containsKey("annotators")); assertTrue(dataTree.containsKey("consumers")); assertTrue(dataTree.get("example") instanceof Map); }
/** * Read configuration from a file. * * @param file * @throws Exception */ public YamlConfiguration(File file) throws IOException { this(new YamlFile(file)); }
/** * Construct configuration from raw yaml string. * * @param clazz * @param resourcePath * @throws Exception */ public YamlConfiguration(Class<?> clazz, String resourcePath) throws IOException { this(new YamlFile(clazz, resourcePath)); }
/** * Read configuration from a file. * * @param file * @throws Exception */ public YamlConfiguration(File file) throws IOException { this(new YamlFile(file)); }
/** * Construct configuration from yaml file, allows yaml to use include statements * * @param file to read */ public YamlPipelineConfiguration(File file) throws IOException { super(new IncludedYaml(new YamlFile(file))); }
/** * Construct configuration from yaml file, allows yaml to use include statements * * @param file to read */ public YamlPiplineConfiguration(File file) throws IOException { super(new IncludedYaml(new YamlFile(file))); }
/** * Construct the IncludeYaml class wrapping the given yaml. Use, the default, file resolver to * find included configuration. * * @param yaml to wrap */ public IncludedYaml(Yaml yaml) { this(yaml, key -> new IncludedYaml(new YamlFile(new File(key)))); }
/** * Construct the IncludeYaml class wrapping the given yaml. Use, the default, file resolver to * find included configuration. * * @param yaml to wrap */ public IncludedYaml(Yaml yaml) { this(yaml, key -> new IncludedYaml(new YamlFile(new File(key)))); }
@Override public Yaml apply(String key) { return new IncludedYaml(new YamlFile(YamlFileTest.class, key), this); } };
Optional<File> configurationFile = getConfigurationFile(argList.get(0)); if (configurationFile.isPresent()) { yamls.add(new IncludedYaml(new YamlFile(configurationFile.get())));
@Before public void setup() throws IOException { yamlFile = new YamlFile(YamlFileTest.class, YAMLCONFIGURATION_YAML); }
source = "string"; } else if (!Strings.isNullOrEmpty(file)) { yaml = new IncludedYaml(new YamlFile(new File(file))); source = "file"; } else {
source = "string"; } else if (!Strings.isNullOrEmpty(file)) { yaml = new IncludedYaml(new YamlFile(new File(file))); source = "file"; } else {
@Test public void mapInclude() throws IOException { Yaml yaml = new IncludedYaml(new YamlFile(YamlFileTest.class, PARENT_INCLUDE_YAML), mapping); String expected = getRaw(PARENT_INCLUDE_YAML); assertEquals(expected, yaml.original()); Object data = yaml.dataTree(); assertTrue(data instanceof Map); Map<String, Object> dataTree = (Map<String, Object>) data; assertTrue(dataTree.containsKey("collectionreader")); List<Object> annotators = (List<Object>) dataTree.get("annotators"); assertEquals(2, annotators.size()); assertEquals( "uk.gov.dstl.baleen.testing.DummyAnnotator", ((Map<String, Object>) annotators.get(1)).get("class")); assertEquals( "uk.gov.dstl.baleen.testing.DummyResourceAnnotator", ((Map<String, Object>) annotators.get(0)).get("class")); }
@Test public void simpleInclude() throws IOException { Yaml yaml = new IncludedYaml(new YamlFile(YamlFileTest.class, PARENT_YAML), mapping); String expected = getRaw(PARENT_YAML); assertEquals(expected, yaml.original()); Object data = yaml.dataTree(); assertTrue(data instanceof Map); Map<String, Object> dataTree = (Map<String, Object>) data; assertTrue(dataTree.containsKey("collectionreader")); List<Object> annotators = (List<Object>) dataTree.get("annotators"); assertEquals(2, annotators.size()); assertEquals( "uk.gov.dstl.baleen.testing.DummyAnnotator", ((Map<String, Object>) annotators.get(1)).get("class")); assertEquals( "uk.gov.dstl.baleen.testing.DummyResourceAnnotator", ((Map<String, Object>) annotators.get(0)).get("class")); }
@Test public void nestedInclude() throws IOException { Yaml yaml = new IncludedYaml(new YamlFile(YamlFileTest.class, GRAND_PARENT_YAML), mapping); String expected = getRaw(GRAND_PARENT_YAML); assertEquals(expected, yaml.original()); Object data = yaml.dataTree(); assertTrue(data instanceof Map); Map<String, Object> dataTree = (Map<String, Object>) data; assertTrue(dataTree.containsKey("collectionreader")); List<Object> annotators = (List<Object>) dataTree.get("annotators"); assertEquals(4, annotators.size()); assertEquals( "uk.gov.dstl.baleen.testing.DummyResourceAnnotator", ((Map<String, Object>) annotators.get(0)).get("class")); assertEquals("uk.gov.dstl.baleen.testing.DummyAnnotator1", annotators.get(1)); assertEquals( "uk.gov.dstl.baleen.testing.DummyAnnotator2", ((Map<String, Object>) annotators.get(2)).get("class")); assertEquals(6, ((Map<String, Object>) annotators.get(2)).get("example.count")); assertEquals("uk.gov.dstl.baleen.testing.DummyAnnotator4", annotators.get(3)); }
@Test public void noIncludes() throws IOException { Yaml yaml = new IncludedYaml(new YamlFile(YamlFileTest.class, YAMLCONFIGURATION_YAML), mapping); String expected = getRaw(YAMLCONFIGURATION_YAML); assertEquals(expected, yaml.original()); Object data = yaml.dataTree(); assertTrue(data instanceof Map); Map<String, Object> dataTree = (Map<String, Object>) data; assertNotNull(dataTree); assertTrue(dataTree.containsKey("example")); assertTrue(dataTree.containsKey("collectionreader")); assertTrue(dataTree.containsKey("annotators")); assertTrue(dataTree.containsKey("consumers")); assertTrue(dataTree.get("example") instanceof Map); }