RubyTest.java

/**
 * Copyright (c) 2008-2010 Andrey Somov
 *
 * 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.yaml.snakeyaml.ruby;

import junit.framework.TestCase;

import org.yaml.snakeyaml.Dumper;
import org.yaml.snakeyaml.DumperOptions;
import org.yaml.snakeyaml.JavaBeanLoader;
import org.yaml.snakeyaml.Loader;
import org.yaml.snakeyaml.TypeDescription;
import org.yaml.snakeyaml.Util;
import org.yaml.snakeyaml.Yaml;
import org.yaml.snakeyaml.constructor.Constructor;
import org.yaml.snakeyaml.nodes.Tag;
import org.yaml.snakeyaml.representer.Representer;

public class RubyTest extends TestCase {

    public void testParse() {
        TestObject result = parseObject(Util.getLocalResource("ruby/ruby1.yaml"));
        assertNotNull(result);
        assertEquals(0, result.getSub1().getAtt2());
        assertEquals("MyString", result.getSub2().getAtt1());
        assertEquals(1, result.getSub2().getAtt2().size());
        assertEquals(12345, result.getSub2().getAtt3());
    }

    public void testEmitNoTags() {
        TestObject result = parseObject(Util.getLocalResource("ruby/ruby1.yaml"));
        DumperOptions options = new DumperOptions();
        options.setExplicitStart(true);
        options.setExplicitRoot(Tag.MAP);
        Yaml yaml2 = new Yaml(options);
        String output = yaml2.dump(result);
        assertFalse("No tags expected.", output.contains("Sub1"));
        // System.out.println(output);
        // parse back. Without tags it shall still work
        JavaBeanLoader<TestObject> beanLoader = new JavaBeanLoader<TestObject>(TestObject.class);
        TestObject result2 = beanLoader.load(output);
        assertEquals(0, result2.getSub1().getAtt2());
        assertEquals("MyString", result2.getSub2().getAtt1());
        assertEquals(1, result2.getSub2().getAtt2().size());
        assertEquals(12345, result2.getSub2().getAtt3());
    }

    public void testEmitWithTags() {
        TestObject result = parseObject(Util.getLocalResource("ruby/ruby1.yaml"));
        DumperOptions options = new DumperOptions();
        options.setExplicitStart(true);
        Representer repr = new Representer();
        repr.addClassTag(TestObject.class, new Tag("!ruby/object:Test::Module::Object"));
        repr.addClassTag(Sub1.class, new Tag("!ruby/object:Test::Module::Sub1"));
        repr.addClassTag(Sub2.class, new Tag("!ruby/object:Test::Module::Sub2"));
        Yaml yaml2 = new Yaml(new Dumper(repr, options));
        String output = yaml2.dump(result);
        // System.out.println(output);
        assertTrue("Tags must be present.", output
                .startsWith("--- !ruby/object:Test::Module::Object"));
        assertTrue("Tags must be present: " + output, output
                .contains("!ruby/object:Test::Module::Sub1"));
        assertTrue("Tags must be present.", output.contains("!ruby/object:Test::Module::Sub2"));
        // parse back.
        TestObject result2 = parseObject(output);
        assertEquals(0, result2.getSub1().getAtt2());
        assertEquals("MyString", result2.getSub2().getAtt1());
        assertEquals(1, result2.getSub2().getAtt2().size());
        assertEquals(12345, result2.getSub2().getAtt3());
    }

    private TestObject parseObject(String input) {
        Constructor con = new Constructor(TestObject.class);
        con.addTypeDescription(new TypeDescription(TestObject.class,
                "!ruby/object:Test::Module::Object"));
        con.addTypeDescription(new TypeDescription(Sub1.class, "!ruby/object:Test::Module::Sub1"));
        con.addTypeDescription(new TypeDescription(Sub2.class, "!ruby/object:Test::Module::Sub2"));

        Yaml yaml = new Yaml(new Loader(con));
        return (TestObject) yaml.load(input);
    }
}