TypeSafeMap2Test.java

/**
 * Copyright (c) 2008-2011, http://www.snakeyaml.org
 *
 * 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 examples.collections;

import java.util.Iterator;
import java.util.LinkedHashMap;
import java.util.Map;

import junit.framework.TestCase;

import org.yaml.snakeyaml.JavaBeanDumper;
import org.yaml.snakeyaml.JavaBeanLoader;
import org.yaml.snakeyaml.Util;

/**
 * Test MapBean->Map<Enum, Developer> developers <br/>
 * Developer class must be properly recognised
 */
public class TypeSafeMap2Test extends TestCase {
    public void testDumpMap() {
        MapBean2 bean = new MapBean2();
        Map<Developer2, Color> data = new LinkedHashMap<Developer2, Color>();
        data.put(new Developer2("Andy", "tester"), Color.BLACK);
        data.put(new Developer2("Lisa", "owner"), Color.RED);
        bean.setData(data);
        Map<Color, Developer2> developers = new LinkedHashMap<Color, Developer2>();
        developers.put(Color.WHITE, new Developer2("Fred", "creator"));
        developers.put(Color.BLACK, new Developer2("John", "committer"));
        bean.setDevelopers(developers);
        JavaBeanDumper dumper = new JavaBeanDumper(false);
        String output = dumper.dump(bean);
        // System.out.println(output);
        String etalon = Util.getLocalResource("examples/map-bean-12.yaml");
        assertEquals(etalon, output);
    }

    public void testMap2() {
        MapBean2 bean = new MapBean2();
        Map<Developer2, Color> data = new LinkedHashMap<Developer2, Color>();
        data.put(new Developer2("Andy", "tester"), Color.BLACK);
        data.put(new SuperMan("Bill", "cleaner", false), Color.BLACK);
        data.put(new Developer2("Lisa", "owner"), Color.RED);
        bean.setData(data);
        Map<Color, Developer2> developers = new LinkedHashMap<Color, Developer2>();
        developers.put(Color.WHITE, new Developer2("Fred", "creator"));
        developers.put(Color.RED, new SuperMan("Jason", "contributor", true));
        developers.put(Color.BLACK, new Developer2("John", "committer"));
        bean.setDevelopers(developers);
        JavaBeanDumper dumper = new JavaBeanDumper(false);
        String output = dumper.dump(bean);
        // System.out.println(output);
        String etalon = Util.getLocalResource("examples/map-bean-13.yaml");
        assertEquals(etalon, output);
        // load
        JavaBeanLoader<MapBean2> beanLoader = new JavaBeanLoader<MapBean2>(MapBean2.class);
        MapBean2 parsed = beanLoader.load(etalon);
        assertNotNull(parsed);
        Map<Developer2, Color> parsedData = parsed.getData();
        assertEquals(3, parsedData.size());
        assertTrue(parsedData.containsKey(new SuperMan("Bill", "cleaner", false)));
        assertEquals(Color.BLACK, parsedData.get(new SuperMan("Bill", "cleaner", false)));
        //
        Map<Color, Developer2> parsedDevelopers = parsed.getDevelopers();
        assertEquals(3, parsedDevelopers.size());
        assertEquals(new SuperMan("Jason", "contributor", true), parsedDevelopers.get(Color.RED));
    }

    public void testLoadMap() {
        String output = Util.getLocalResource("examples/map-bean-12.yaml");
        // System.out.println(output);
        JavaBeanLoader<MapBean2> beanLoader = new JavaBeanLoader<MapBean2>(MapBean2.class);
        MapBean2 parsed = beanLoader.load(output);
        assertNotNull(parsed);
        Map<Developer2, Color> data = parsed.getData();
        assertEquals(2, data.size());
        Iterator<Developer2> iter = data.keySet().iterator();
        Developer2 first = iter.next();
        assertEquals("Andy", first.getName());
        assertEquals("tester", first.getRole());
        assertEquals(Color.BLACK, data.get(first));
        Developer2 second = iter.next();
        assertEquals("Lisa", second.getName());
        assertEquals("owner", second.getRole());
        assertEquals(Color.RED, data.get(second));
        //
        Map<Color, Developer2> developers = parsed.getDevelopers();
        assertEquals(2, developers.size());
        Iterator<Color> iter2 = developers.keySet().iterator();
        Color firstColor = iter2.next();
        assertEquals(Color.WHITE, firstColor);
        Developer2 dev1 = developers.get(firstColor);
        assertEquals("Fred", dev1.getName());
        assertEquals("creator", dev1.getRole());
        Color secondColor = iter2.next();
        assertEquals(Color.BLACK, secondColor);
        Developer2 dev2 = developers.get(secondColor);
        assertEquals("John", dev2.getName());
        assertEquals("committer", dev2.getRole());
    }

    public static enum Color {
        WHITE, BLACK, RED;
    }

    public static class MapBean2 {
        private Map<Developer2, Color> data;
        private String name;
        private Map<Color, Developer2> developers;

        public MapBean2() {
            name = "Bean123";
        }

        public String getName() {
            return name;
        }

        public void setName(String name) {
            this.name = name;
        }

        public Map<Color, Developer2> getDevelopers() {
            return developers;
        }

        public void setDevelopers(Map<Color, Developer2> developers) {
            this.developers = developers;
        }

        public Map<Developer2, Color> getData() {
            return data;
        }

        public void setData(Map<Developer2, Color> data) {
            this.data = data;
        }

    }

    public static class Developer2 implements Comparable<Developer2> {
        private String name;
        private String role;

        public Developer2() {
        }

        private Developer2(String name, String role) {
            this.name = name;
            this.role = role;
        }

        public String getName() {
            return name;
        }

        public void setName(String name) {
            this.name = name;
        }

        public String getRole() {
            return role;
        }

        public void setRole(String role) {
            this.role = role;
        }

        public int compareTo(Developer2 o) {
            return name.compareTo(o.name);
        }

        @Override
        public boolean equals(Object obj) {
            if (obj instanceof Developer2) {
                return toString().equals(obj.toString());
            } else {
                return false;
            }
        }

        @Override
        public int hashCode() {
            return toString().hashCode();
        }

        @Override
        public String toString() {
            return "Developer " + name + " " + role;
        }

    }

    public static class SuperMan extends Developer2 {
        private boolean smart;

        public SuperMan() {
            super();
        }

        private SuperMan(String name, String role, boolean smart) {
            super(name, role);
            this.smart = smart;
        }

        public boolean isSmart() {
            return smart;
        }

        public void setSmart(boolean smart) {
            this.smart = smart;
        }

        @Override
        public String toString() {
            return "Super" + super.toString();
        }
    }
}