package com.thoughtworks.xstream.converters.enums;

import junit.framework.TestCase;

import com.thoughtworks.xstream.XStream;


public class EnumSingleValueConverterTest extends TestCase {
    
    private XStream xstream;

    protected void setUp() throws Exception {
        super.setUp();
        xstream = new XStream();
        xstream.alias("EnumTest", EnumTest.class);
        
        xstream.useAttributeFor(SimpleEnum.class);
        xstream.registerConverter(EnumSingleValueConverter.create(SimpleEnum.class));
        
        xstream.useAttributeFor(EnumTest.class,"polymorphic");
        xstream.registerConverter(EnumSingleValueConverter.create(PolymorphicEnum.class));
    }
    
    public final void testAsAttributes(){
        EnumTest enumTest = new EnumTest();
        
        String xml = xstream.toXML(enumTest);
        String expected = "<EnumTest simple=\"GREEN\" polymorphic=\"A\"/>";
        assertEquals(expected, xml);
        
        EnumTest newEnumTest = (EnumTest) xstream.fromXML(xml);
        assertSame(enumTest.simple, newEnumTest.simple);
        assertSame(enumTest.polymorphic, newEnumTest.polymorphic);
    }
    
    static class EnumTest{
        SimpleEnum simple = SimpleEnum.GREEN;
        PolymorphicEnum polymorphic = PolymorphicEnum.A;
    }
}