[bs-22] Add tests for Yaml processing
* Gap in logic identified, so DocumentMatcher refactored to return an enum [#48127729] [bs-22] Add missing unit testspull/1/merge
Dave Syer
12 years ago
parent
7531c5acfb
commit
c675d9c76e
@ -0,0 +1,90 @@
|
||||
/*
|
||||
* Copyright 2012-2013 the original author or authors.
|
||||
*
|
||||
* 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 org.springframework.bootstrap.config;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.io.InputStream;
|
||||
import java.util.Map;
|
||||
|
||||
import org.junit.Test;
|
||||
import org.springframework.bootstrap.config.YamlProcessor.ResolutionMethod;
|
||||
import org.springframework.core.io.AbstractResource;
|
||||
import org.springframework.core.io.ByteArrayResource;
|
||||
import org.springframework.core.io.FileSystemResource;
|
||||
import org.springframework.core.io.Resource;
|
||||
|
||||
import static org.junit.Assert.assertEquals;
|
||||
|
||||
/**
|
||||
* @author Dave Syer
|
||||
*
|
||||
*/
|
||||
public class YamlMapFactoryBeanTests {
|
||||
|
||||
private YamlMapFactoryBean factory = new YamlMapFactoryBean();
|
||||
|
||||
@Test
|
||||
public void testSetIgnoreResourceNotFound() throws Exception {
|
||||
this.factory
|
||||
.setResolutionMethod(YamlMapFactoryBean.ResolutionMethod.OVERRIDE_AND_IGNORE);
|
||||
this.factory.setResources(new FileSystemResource[] { new FileSystemResource(
|
||||
"non-exsitent-file.yml") });
|
||||
assertEquals(0, this.factory.getObject().size());
|
||||
}
|
||||
|
||||
@Test(expected = IllegalStateException.class)
|
||||
public void testSetBarfOnResourceNotFound() throws Exception {
|
||||
this.factory.setResources(new FileSystemResource[] { new FileSystemResource(
|
||||
"non-exsitent-file.yml") });
|
||||
assertEquals(0, this.factory.getObject().size());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testGetObject() throws Exception {
|
||||
this.factory.setResources(new ByteArrayResource[] { new ByteArrayResource(
|
||||
"foo: bar".getBytes()) });
|
||||
assertEquals(1, this.factory.getObject().size());
|
||||
}
|
||||
|
||||
@SuppressWarnings("unchecked")
|
||||
@Test
|
||||
public void testOverrideAndremoveDefaults() throws Exception {
|
||||
this.factory.setResources(new ByteArrayResource[] {
|
||||
new ByteArrayResource("foo:\n bar: spam".getBytes()),
|
||||
new ByteArrayResource("foo:\n spam: bar".getBytes()) });
|
||||
assertEquals(1, this.factory.getObject().size());
|
||||
assertEquals(2,
|
||||
((Map<String, Object>) this.factory.getObject().get("foo")).size());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testFirstFound() throws Exception {
|
||||
this.factory.setResolutionMethod(ResolutionMethod.FIRST_FOUND);
|
||||
this.factory.setResources(new Resource[] { new AbstractResource() {
|
||||
@Override
|
||||
public String getDescription() {
|
||||
return "non-existent";
|
||||
}
|
||||
|
||||
@Override
|
||||
public InputStream getInputStream() throws IOException {
|
||||
throw new IOException("planned");
|
||||
}
|
||||
}, new ByteArrayResource("foo:\n spam: bar".getBytes()) });
|
||||
assertEquals(1, this.factory.getObject().size());
|
||||
}
|
||||
|
||||
}
|
||||
@ -0,0 +1,212 @@
|
||||
/*
|
||||
* Copyright 2012-2013 the original author or authors.
|
||||
*
|
||||
* 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 org.springframework.bootstrap.config;
|
||||
|
||||
import java.util.Arrays;
|
||||
import java.util.Map;
|
||||
import java.util.Properties;
|
||||
|
||||
import org.junit.Ignore;
|
||||
import org.junit.Test;
|
||||
import org.springframework.bootstrap.config.YamlProcessor.DocumentMatcher;
|
||||
import org.springframework.bootstrap.config.YamlProcessor.MatchStatus;
|
||||
import org.springframework.bootstrap.config.YamlProcessor.ResolutionMethod;
|
||||
import org.springframework.core.io.ByteArrayResource;
|
||||
import org.springframework.core.io.ClassPathResource;
|
||||
import org.springframework.core.io.Resource;
|
||||
import org.yaml.snakeyaml.Yaml;
|
||||
|
||||
import static org.junit.Assert.assertEquals;
|
||||
|
||||
/**
|
||||
* @author Dave Syer
|
||||
*
|
||||
*/
|
||||
public class YamlPropertiesFactoryBeanTests {
|
||||
|
||||
@Test
|
||||
public void testLoadResource() throws Exception {
|
||||
YamlPropertiesFactoryBean factory = new YamlPropertiesFactoryBean();
|
||||
factory.setResources(new Resource[] { new ByteArrayResource(
|
||||
"foo: bar\nspam:\n foo: baz".getBytes()) });
|
||||
Properties properties = factory.getObject();
|
||||
assertEquals("bar", properties.get("foo"));
|
||||
assertEquals("baz", properties.get("spam.foo"));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testLoadResourcesWithOverride() throws Exception {
|
||||
YamlPropertiesFactoryBean factory = new YamlPropertiesFactoryBean();
|
||||
factory.setResources(new Resource[] {
|
||||
new ByteArrayResource("foo: bar\nspam:\n foo: baz".getBytes()),
|
||||
new ByteArrayResource("foo:\n bar: spam".getBytes()) });
|
||||
Properties properties = factory.getObject();
|
||||
assertEquals("bar", properties.get("foo"));
|
||||
assertEquals("baz", properties.get("spam.foo"));
|
||||
assertEquals("spam", properties.get("foo.bar"));
|
||||
}
|
||||
|
||||
@Test
|
||||
@Ignore
|
||||
// We can't fail on duplicate keys because the Map is created by the YAML library
|
||||
public void testLoadResourcesWithInternalOverride() throws Exception {
|
||||
YamlPropertiesFactoryBean factory = new YamlPropertiesFactoryBean();
|
||||
factory.setResources(new Resource[] { new ByteArrayResource(
|
||||
"foo: bar\nspam:\n foo: baz\nfoo: bucket".getBytes()) });
|
||||
Properties properties = factory.getObject();
|
||||
assertEquals("bar", properties.get("foo"));
|
||||
}
|
||||
|
||||
@Test
|
||||
@Ignore
|
||||
// We can't fail on duplicate keys because the Map is created by the YAML library
|
||||
public void testLoadResourcesWithNestedInternalOverride() throws Exception {
|
||||
YamlPropertiesFactoryBean factory = new YamlPropertiesFactoryBean();
|
||||
factory.setResources(new Resource[] { new ByteArrayResource(
|
||||
"foo:\n bar: spam\n foo: baz\nbreak: it\nfoo: bucket".getBytes()) });
|
||||
Properties properties = factory.getObject();
|
||||
assertEquals("spam", properties.get("foo.bar"));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testLoadResourceWithMultipleDocuments() throws Exception {
|
||||
YamlPropertiesFactoryBean factory = new YamlPropertiesFactoryBean();
|
||||
factory.setResources(new Resource[] { new ByteArrayResource(
|
||||
"foo: bar\nspam: baz\n---\nfoo: bag".getBytes()) });
|
||||
Properties properties = factory.getObject();
|
||||
assertEquals("bag", properties.get("foo"));
|
||||
assertEquals("baz", properties.get("spam"));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testLoadResourceWithSelectedDocuments() throws Exception {
|
||||
YamlPropertiesFactoryBean factory = new YamlPropertiesFactoryBean();
|
||||
factory.setResources(new Resource[] { new ByteArrayResource(
|
||||
"foo: bar\nspam: baz\n---\nfoo: bag\nspam: bad".getBytes()) });
|
||||
factory.setDocumentMatchers(Arrays
|
||||
.<DocumentMatcher> asList(new DocumentMatcher() {
|
||||
@Override
|
||||
public MatchStatus matches(Properties properties) {
|
||||
return "bag".equals(properties.getProperty("foo")) ? MatchStatus.FOUND
|
||||
: MatchStatus.NOT_FOUND;
|
||||
}
|
||||
}));
|
||||
Properties properties = factory.getObject();
|
||||
assertEquals("bag", properties.get("foo"));
|
||||
assertEquals("bad", properties.get("spam"));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testLoadResourceWithDefaultMatch() throws Exception {
|
||||
YamlPropertiesFactoryBean factory = new YamlPropertiesFactoryBean();
|
||||
factory.setMatchDefault(true);
|
||||
factory.setResources(new Resource[] { new ByteArrayResource(
|
||||
"one: two\n---\nfoo: bar\nspam: baz\n---\nfoo: bag\nspam: bad".getBytes()) });
|
||||
factory.setDocumentMatchers(Arrays
|
||||
.<DocumentMatcher> asList(new DocumentMatcher() {
|
||||
@Override
|
||||
public MatchStatus matches(Properties properties) {
|
||||
if (!properties.containsKey("foo")) {
|
||||
return MatchStatus.ABSTAIN;
|
||||
}
|
||||
return "bag".equals(properties.getProperty("foo")) ? MatchStatus.FOUND
|
||||
: MatchStatus.NOT_FOUND;
|
||||
}
|
||||
}));
|
||||
Properties properties = factory.getObject();
|
||||
assertEquals("bag", properties.get("foo"));
|
||||
assertEquals("bad", properties.get("spam"));
|
||||
assertEquals("two", properties.get("one"));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testLoadResourceWithDefaultMatchSkippingMissedMatch() throws Exception {
|
||||
YamlPropertiesFactoryBean factory = new YamlPropertiesFactoryBean();
|
||||
factory.setMatchDefault(true);
|
||||
factory.setResources(new Resource[] { new ByteArrayResource(
|
||||
"one: two\n---\nfoo: bag\nspam: bad\n---\nfoo: bar\nspam: baz".getBytes()) });
|
||||
factory.setDocumentMatchers(Arrays
|
||||
.<DocumentMatcher> asList(new DocumentMatcher() {
|
||||
@Override
|
||||
public MatchStatus matches(Properties properties) {
|
||||
if (!properties.containsKey("foo")) {
|
||||
return MatchStatus.ABSTAIN;
|
||||
}
|
||||
return "bag".equals(properties.getProperty("foo")) ? MatchStatus.FOUND
|
||||
: MatchStatus.NOT_FOUND;
|
||||
}
|
||||
}));
|
||||
Properties properties = factory.getObject();
|
||||
assertEquals("bag", properties.get("foo"));
|
||||
assertEquals("bad", properties.get("spam"));
|
||||
assertEquals("two", properties.get("one"));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testLoadNonExistentResource() throws Exception {
|
||||
YamlPropertiesFactoryBean factory = new YamlPropertiesFactoryBean();
|
||||
factory.setResolutionMethod(ResolutionMethod.OVERRIDE_AND_IGNORE);
|
||||
factory.setResources(new Resource[] { new ClassPathResource("no-such-file.yml") });
|
||||
Properties properties = factory.getObject();
|
||||
assertEquals(0, properties.size());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testLoadNull() throws Exception {
|
||||
YamlPropertiesFactoryBean factory = new YamlPropertiesFactoryBean();
|
||||
factory.setResources(new Resource[] { new ByteArrayResource("foo: bar\nspam:"
|
||||
.getBytes()) });
|
||||
Properties properties = factory.getObject();
|
||||
assertEquals("bar", properties.get("foo"));
|
||||
assertEquals("", properties.get("spam"));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testLoadArrayOfString() throws Exception {
|
||||
YamlPropertiesFactoryBean factory = new YamlPropertiesFactoryBean();
|
||||
factory.setResources(new Resource[] { new ByteArrayResource("foo:\n- bar\n- baz"
|
||||
.getBytes()) });
|
||||
Properties properties = factory.getObject();
|
||||
assertEquals("bar", properties.get("foo[0]"));
|
||||
assertEquals("baz", properties.get("foo[1]"));
|
||||
assertEquals("bar,baz", properties.get("foo"));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testLoadArrayOfObject() throws Exception {
|
||||
YamlPropertiesFactoryBean factory = new YamlPropertiesFactoryBean();
|
||||
factory.setResources(new Resource[] { new ByteArrayResource(
|
||||
"foo:\n- bar:\n spam: crap\n- baz\n- one: two\n three: four"
|
||||
.getBytes()) });
|
||||
Properties properties = factory.getObject();
|
||||
// System.err.println(properties);
|
||||
assertEquals("crap", properties.get("foo[0].bar.spam"));
|
||||
assertEquals("baz", properties.get("foo[1]"));
|
||||
assertEquals("two", properties.get("foo[2].one"));
|
||||
assertEquals("four", properties.get("foo[2].three"));
|
||||
assertEquals("{bar={spam=crap}},baz,{one=two, three=four}", properties.get("foo"));
|
||||
}
|
||||
|
||||
@SuppressWarnings("unchecked")
|
||||
@Test
|
||||
public void testYaml() {
|
||||
Yaml yaml = new Yaml();
|
||||
Map<String, ?> map = yaml.loadAs("foo: bar\nspam:\n foo: baz", Map.class);
|
||||
assertEquals("bar", map.get("foo"));
|
||||
assertEquals("baz", ((Map<String, Object>) map.get("spam")).get("foo"));
|
||||
}
|
||||
|
||||
}
|
||||
Loading…
Reference in New Issue