~launchpad-pqm/launchpad/devel

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
# Copyright 2009 Canonical Ltd.  This software is licensed under the
# GNU Affero General Public License version 3 (see the file LICENSE).

__metaclass__ = type

import unittest

from canonical.testing.layers import DatabaseFunctionalLayer
from lp.blueprints.browser.specification import (
    SpecificationActionMenu,
    SpecificationContextMenu,
    )
from lp.testing import TestCaseWithFactory
from lp.testing.menu import check_menu_links


class TestSpecificationMenus(TestCaseWithFactory):
    """Test specification menus links."""
    layer = DatabaseFunctionalLayer

    def setUp(self):
        TestCaseWithFactory.setUp(self)
        self.specification = self.factory.makeSpecification()

    def test_SpecificationContextMenu(self):
        menu = SpecificationContextMenu(self.specification)
        self.assertTrue(check_menu_links(menu))

    def test_SpecificationActionMenu(self):
        menu = SpecificationActionMenu(self.specification)
        self.assertTrue(check_menu_links(menu))


def test_suite():
    suite = unittest.TestSuite()
    suite.addTest(unittest.TestLoader().loadTestsFromName(__name__))
    return suite


if __name__ == '__main__':
    unittest.TextTestRunner().run(test_suite())