Geografic-Information-System/scripts/g.search.modules/testsuite/test_g_search_modules.py

"""
TEST:      test_g.search.modules.py

AUTHOR(S): Jachym Cepicky <jachym.cepicky gmail com>

PURPOSE:   Test g.search.modules script outputs

COPYRIGHT: (C) 2015 Jachym Ceppicky, and by the GRASS Development Team

           This program is free software under the GNU General Public
           License (>=v2). Read the file COPYING that comes with GRASS
           for details.
"""

from grass.gunittest.case import TestCase
from grass.gunittest.main import test
from grass.gunittest.gmodules import SimpleModule

import termcolor

import os


class TestSearchModule(TestCase):

    def test_terminal_output(self):
        """ """
        module = SimpleModule('g.search.modules', keyword="water")
        self.assertModule(module)
        stdout = module.outputs.stdout
        self.assertEqual(stdout.split()[0], 'r.watershed')

    def test_json_output(self):
        import json
        module = SimpleModule('g.search.modules', keyword="water", flags="j")
        self.assertModule(module)
        stdout = json.loads(module.outputs.stdout)
        self.assertEqual(len(stdout), 6, 'Six modules found')
        self.assertEqual(stdout[3]['name'], 'r.basins.fill', 'r.basins.fill')
        self.assertTrue('keywords' in stdout[3]['attributes'])

    def test_shell_outout(self):
        module = SimpleModule('g.search.modules', keyword="water", flags="g")
        self.assertModule(module)
        stdout = module.outputs.stdout.split()
        self.assertEqual(len(stdout), 6)
        self.assertEqual(stdout[3], 'r.basins.fill')

    def test_colored_terminal(self):
        module = SimpleModule('g.search.modules', keyword="water", flags="c")
        self.assertModule(module)
        stdout = module.outputs.stdout.split()
        self.assertEqual(stdout[0],
                         termcolor.colored('r.watershed',
                         attrs=['bold']))

    def test_manual_pages(self):
        module = SimpleModule('g.search.modules', keyword="kapri", flags="gm")
        self.assertModule(module)
        stdout = module.outputs.stdout.split()
        self.assertEqual(len(stdout), 2)

if __name__ == '__main__':
    test()