~launchpad-pqm/launchpad/devel

# Copyright 2009 Canonical Ltd.  This software is licensed under the

# GNU Affero General Public License version 3 (see the file LICENSE).

"""

Run the doctests and pagetests.

"""

import logging

import os

import unittest

from canonical.database.sqlbase import flush_database_updates

from canonical.launchpad.testing.pages import PageTestSuite

from canonical.launchpad.testing.systemdocs import (

    LayeredDocFileSuite, setUp, tearDown)

from canonical.testing import (

    DatabaseFunctionalLayer, DatabaseLayer, LaunchpadFunctionalLayer,

    LaunchpadZopelessLayer)

from lp.registry.tests import mailinglists_helper

here = os.path.dirname(os.path.realpath(__file__))

special = {}

def test_suite():

    suite = unittest.TestSuite()

    stories_dir = os.path.join(os.path.pardir, 'stories')

    suite.addTest(PageTestSuite(stories_dir))

    stories_path = os.path.join(here, stories_dir)

    for story_dir in os.listdir(stories_path):

        full_story_dir = os.path.join(stories_path, story_dir)

        if not os.path.isdir(full_story_dir):

            continue

        story_path = os.path.join(stories_dir, story_dir)

        suite.addTest(PageTestSuite(story_path))

    testsdir = os.path.abspath(

        os.path.normpath(os.path.join(here, os.path.pardir, 'doc'))

        )

    # Add special needs tests

    for key in sorted(special):

        special_suite = special[key]

        suite.addTest(special_suite)

    # Add tests using default setup/teardown

    filenames = [filename

                 for filename in os.listdir(testsdir)

                 if filename.endswith('.txt') and filename not in special]

    # Sort the list to give a predictable order.

    filenames.sort()

    for filename in filenames:

        path = os.path.join('../doc/', filename)

        one_test = LayeredDocFileSuite(

            path, setUp=setUp, tearDown=tearDown,

            layer=LaunchpadFunctionalLayer,

            stdout_logging_level=logging.WARNING

            )

        suite.addTest(one_test)

    return suite