#!/usr/bin/env python

# Copyright 2013 OpenStack Foundation
#
# 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.

import ConfigParser
import os
import re

import testtools
import voluptuous
import yaml

import zuul.layoutvalidator
import zuul.lib.connections

FIXTURE_DIR = os.path.join(os.path.dirname(__file__),
                           'fixtures')
LAYOUT_RE = re.compile(r'^(good|bad)_.*\.yaml$')


class TestLayoutValidator(testtools.TestCase):
    def setUp(self):
        self.skip("Disabled for early v3 development")

    def test_layouts(self):
        """Test layout file validation"""
        print
        errors = []
        for fn in os.listdir(os.path.join(FIXTURE_DIR, 'layouts')):
            m = LAYOUT_RE.match(fn)
            if not m:
                continue
            print fn

            # Load any .conf file by the same name but .conf extension.
            config_file = ("%s.conf" %
                           os.path.join(FIXTURE_DIR, 'layouts',
                                        fn.split('.yaml')[0]))
            if not os.path.isfile(config_file):
                config_file = os.path.join(FIXTURE_DIR, 'layouts',
                                           'zuul_default.conf')
            config = ConfigParser.ConfigParser()
            config.read(config_file)
            connections = zuul.lib.connections.configure_connections(config)

            layout = os.path.join(FIXTURE_DIR, 'layouts', fn)
            data = yaml.load(open(layout))
            validator = zuul.layoutvalidator.LayoutValidator()
            if m.group(1) == 'good':
                try:
                    validator.validate(data, connections)
                except voluptuous.Invalid as e:
                    raise Exception(
                        'Unexpected YAML syntax error in %s:\n  %s' %
                        (fn, str(e)))
            else:
                try:
                    validator.validate(data, connections)
                    raise Exception("Expected a YAML syntax error in %s." %
                                    fn)
                except voluptuous.Invalid as e:
                    error = str(e)
                    print '  ', error
                    if error in errors:
                        raise Exception("Error has already been tested: %s" %
                                        error)
                    else:
                        errors.append(error)
                    pass
