YamlToolsTest.java

package dk.kb.yggdrasil.utils;

import static org.junit.Assert.*;

import java.io.File;
import java.util.LinkedHashMap;

import org.junit.Assert;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.JUnit4;
import org.yaml.snakeyaml.scanner.ScannerException;

import dk.kb.yggdrasil.config.RunningMode;
import dk.kb.yggdrasil.exceptions.YggdrasilException;
import dk.kb.yggdrasil.utils.YamlTools;

/**
 * Tests for the methods in the YamlTools class.
 *
 */
@RunWith(JUnit4.class)
public class YamlToolsTest {

    public static String YAML_TEST_FILE = "src/test/resources/config/rabbitmq.yml";
    public static String NOT_YAML_TEST_FILE = "src/test/resources/config/rabbitmq.yaml";
    public static String NOT_YAML_TEST_FILE2 = "src/test/resources/config/file_with_no_yaml_content.xml";

    @Test
    public void testReadYamlFailed() throws Exception {
        File f = new File(NOT_YAML_TEST_FILE);
        try {
            YamlTools.loadYamlSettings(f);
            fail("Should throw YggdrasilException on non existing file");
        } catch (YggdrasilException e) {
            // expected
        }
    }

    @Test
    public void testReadNonYamlFile() throws Exception {
        File f = new File(NOT_YAML_TEST_FILE2);
        try {
            YamlTools.loadYamlSettings(f);
            fail("Should throw YAML ScannerException on reading non YAML file");
        } catch (ScannerException e) {
            // expected
        }
    }

    @SuppressWarnings("rawtypes")
    @Test
    public void testReadYamlFile() throws Exception {
        File f = new File(YAML_TEST_FILE);

        LinkedHashMap m = YamlTools.loadYamlSettings(f);
        Assert.assertNotNull(m);
        String mode = RunningMode.getMode().toString();
        Assert.assertTrue(m.containsKey(mode));
    }

}