1#!/usr/bin/env python
2from __future__ import absolute_import, division, unicode_literals
3import argparse
4import unittest
5
6from cola import app
7
8from .helper import run_unittest
9
10
11class AppTestCase(unittest.TestCase):
12    def test_setup_environment(self):
13        # If the function doesn't throw an exception we are happy.
14        self.assertTrue(hasattr(app, 'setup_environment'))
15        app.setup_environment()
16
17    def test_add_common_arguments(self):
18        # If the function doesn't throw an exception we are happy.
19        parser = argparse.ArgumentParser()
20        self.assertTrue(hasattr(app, 'add_common_arguments'))
21        app.add_common_arguments(parser)
22
23
24def test_suite():
25    return unittest.makeSuite(AppTestCase)
26
27
28if __name__ == "__main__":
29    run_unittest(test_suite())
30