source: nagslang/tests/test_yamlish.py@ 102:fd9c4db5ddfe

from unittest import TestCase, SkipTest
from StringIO import StringIO

try:
    import yaml
except ImportError:
    yaml = None  # pyflakes:ignore

from nagslang.yamlish import load, dump


class TestRoundTrip(TestCase):
    def roundtrip(self, data):
        f = StringIO()
        self.dump(data, f)
        f.seek(0)
        print '\n=== Begin ===\n%s\n=== End ===' % f.buf.rstrip()
        self.assertEqual(self.load(f), data)

    def dump(self, data, f):
        dump(data, f)

    def load(self, f):
        return load(f)

    def test_simple_dict(self):
        self.roundtrip({'foo': 'bar'})

    def test_dict_of_dicts(self):
        self.roundtrip({'foo': {'bar': 'baz'}})

    def test_dict_tree(self):
        self.roundtrip({
            'foo': {
                'bar': {
                    'baz': 'qux'
                },
                'quux': 'corge',
            }
        })

    def test_dict_list(self):
        self.roundtrip({
            'foo': ['bar', 'baz'],
        })

    def test_nested_lists(self):
        self.roundtrip({
            'foo': [['bar', 'baz', 'qux'], 'quux'],
        })

    def test_list_of_dicts(self):
        self.roundtrip({
            'foo': [
                {'bar': 'baz'},
                {'qux': 'quux'},
            ],
        })


class TestFromPyYAML(TestRoundTrip):
    def dump(self, data, f):
        if yaml is None:
            raise SkipTest('yaml module unavailable')
        yaml.dump(data, f, default_flow_style=False)


class TestToPyYAML(TestRoundTrip):
    def load(self, f):
        if yaml is None:
            raise SkipTest('yaml module unavailable')
        return yaml.load(f)