package org.simpleframework.xml.strategy;
import java.util.HashMap;
import java.util.Map;
import org.simpleframework.xml.Element;
import org.simpleframework.xml.ElementMap;
import org.simpleframework.xml.Root;
import org.simpleframework.xml.Serializer;
import org.simpleframework.xml.ValidationTestCase;
import org.simpleframework.xml.core.Persister;
import org.simpleframework.xml.strategy.CycleStrategy;
import org.simpleframework.xml.strategy.Strategy;
public class MapCycleTest extends ValidationTestCase {
private static final String ENTRY_MAP =
"\n"+
" \n"+
"";
private static final String COMPLEX_MAP =
"\n"+
" \n"+
"";
private static final String PRIMITIVE_MAP =
"\n"+
" \n"+
"";
@Root
private static class EntryMap {
@ElementMap(key="key", attribute=true)
private Map map;
public String getValue(String name) {
return map.get(name).value;
}
public MapEntry getEntry(String name) {
return map.get(name);
}
}
@Root
private static class MapEntry {
@Element
private String name;
@Element
private String value;
}
@Root
private static class StringMap {
@ElementMap(key="letter", attribute=true)
private Map map;
public String getValue(String name) {
return map.get(name);
}
}
@Root
private static class ComplexMap {
@ElementMap
private Map map;
public String getValue(CompositeKey key) {
return map.get(key).value;
}
}
@Root
private static class CompositeKey {
@Element
private String name;
@Element
private String address;
public CompositeKey() {
super();
}
public CompositeKey(String name, String address) {
this.name = name;
this.address = address;
}
public int hashCode() {
return name.hashCode() + address.hashCode();
}
public boolean equals(Object item) {
if(item instanceof CompositeKey) {
CompositeKey other = (CompositeKey)item;
return other.name.equals(name) && other.address.equals(address);
}
return false;
}
}
@Root
private static class BadMap {
@ElementMap
private Map map;
public BadMap() {
this.map = new HashMap();
}
public String getValue(CompositeKey key) {
return map.get(key);
}
public void setValue(CompositeKey key, String value) {
map.put(key, value);
}
}
@Root
private static class PrimitiveMap {
@ElementMap
private Map map;
public double getValue(String name) {
return map.get(name);
}
}
public void testEntryMap() throws Exception {
Strategy strategy = new CycleStrategy();
Serializer serializer = new Persister(strategy);
EntryMap example = serializer.read(EntryMap.class, ENTRY_MAP);
assertEquals("example 1", example.getValue("a"));
assertEquals("example 2", example.getValue("b"));
assertEquals("example 1", example.getValue("c"));
assertEquals("example 1", example.getValue("d"));
MapEntry a = example.getEntry("a");
MapEntry b = example.getEntry("b");
MapEntry c = example.getEntry("c");
MapEntry d = example.getEntry("d");
assertTrue(a == c);
assertTrue(c == d);
assertFalse(a == b);
validate(example, serializer);
}
public void testComplexMap() throws Exception {
Strategy strategy = new CycleStrategy();
Serializer serializer = new Persister(strategy);
ComplexMap example = serializer.read(ComplexMap.class, COMPLEX_MAP);
assertEquals("example 2", example.getValue(new CompositeKey("name 1", "address 1")));
assertEquals("example 2", example.getValue(new CompositeKey("name 3", "address 3")));
assertEquals("example 4", example.getValue(new CompositeKey("name 4", "address 4")));
validate(example, serializer);
}
public void testPrimitiveMap() throws Exception {
Strategy strategy = new CycleStrategy();
Serializer serializer = new Persister(strategy);
PrimitiveMap example = serializer.read(PrimitiveMap.class, PRIMITIVE_MAP);
assertEquals(1.0, example.getValue("one"));
assertEquals(1.0, example.getValue("two"));
assertEquals(4.0, example.getValue("three"));
validate(example, serializer);
}
}