mapnik/tests/python_tests/load_map_test.py

#!/usr/bin/env python

from nose.tools import *
from utilities import execution_path, run_all, datasources_available

import os, sys, glob, mapnik

default_logging_severity = mapnik.logger.get_severity()

def setup():
    # make the tests silent to suppress unsupported params from harfbuzz tests
    # TODO: remove this after harfbuzz branch merges
    mapnik.logger.set_severity(mapnik.severity_type.None)
    # All of the paths used are relative, if we run the tests
    # from another directory we need to chdir()
    os.chdir(execution_path('.'))

def teardown():
    mapnik.logger.set_severity(default_logging_severity)

def test_broken_files():
    default_logging_severity = mapnik.logger.get_severity()
    mapnik.logger.set_severity(mapnik.severity_type.None)
    broken_files = glob.glob("../data/broken_maps/*.xml")
    # Add a filename that doesn't exist 
    broken_files.append("../data/broken/does_not_exist.xml")

    failures = [];
    for filename in broken_files:
        try:
            m = mapnik.Map(512, 512)
            strict = True
            mapnik.load_map(m, filename, strict)
            failures.append('Loading broken map (%s) did not raise RuntimeError!' % filename)
        except RuntimeError:
            pass
    eq_(len(failures),0,'\n'+'\n'.join(failures))
    mapnik.logger.set_severity(default_logging_severity)

def test_good_files():
    all_files = glob.glob("../data/good_maps/*.xml")
    all_files.extend(glob.glob("../visual_tests/styles/*.xml"))

    good_files = list()
    for xmlfile in all_files:
        missing_plugins = set()
        have_inputs = datasources_available(xmlfile, missing_plugins)
        if have_inputs:
            good_files.append(xmlfile)
        else:
            print 'Notice: skipping load_map_test for %s due to unavailable input plugins: %s' % (os.path.basename(xmlfile), list(missing_plugins))

    failures = [];
    strict = False
    for filename in good_files:
        try:
            m = mapnik.Map(512, 512)
            mapnik.load_map(m, filename, strict)
            base_path = os.path.dirname(filename)
            mapnik.load_map_from_string(m, open(filename, 'rb').read(), strict, base_path)
        except RuntimeError, e:
            failures.append('Failed to load valid map %s (%s)!' % (filename, str(e)))
    eq_(len(failures),0,'\n'+'\n'.join(failures))

if __name__ == "__main__":
    setup()
    exit(run_all(eval(x) for x in dir() if x.startswith("test_")))