#!/usr/bin/env python


try:
    import unittest2 as unittest
except ImportError:
    import unittest

import ghstats


class TestStats(unittest.TestCase):

    def test_cli(self):
        """
        Test command line arguments.
        """
        count = ghstats.main_cli(["kefir500/apk-icon-editor", "-l", "-d"])
        self.assertTrue(count > 0)

    def test_releases(self):
        """
        Test statistics for all releases.
        """
        stats = ghstats.download_stats("kefir500", "apk-icon-editor", None, False, ghstats.get_env_token(), False)
        self.assertTrue(isinstance(stats, list))
        count = ghstats.get_stats_downloads(stats, False)
        self.assertTrue(count > 0)

    def test_release(self):
        """
        Test statistics for the latest release.
        """
        stats = ghstats.download_stats("kefir500", "apk-icon-editor", None, True, ghstats.get_env_token(), False)
        self.assertTrue(isinstance(stats, dict))
        count = ghstats.get_stats_downloads(stats, False)
        self.assertTrue(count > 0)

    def test_invalid_repo(self):
        """
        Test nonexistent repository.
        """
        with self.assertRaises(ghstats.GithubRepoError):
            ghstats.download_stats("kefir500", "foobar", None, False, ghstats.get_env_token(), True)

    def test_invalid_token(self):
        """
        Test invalid GitHub personal access token.
        """
        with self.assertRaises(ghstats.GithubTokenError):
            ghstats.download_stats("kefir500", "foobar", None, False, "FOOBAR", True)

    def test_exits(self):
        """
        Test if functions halt the script.
        """
        with self.assertRaises(SystemExit) as cm:
            ghstats.print_help()
        self.assertEqual(cm.exception.code, 0)
        with self.assertRaises(SystemExit) as cm:
            ghstats.error("Test")
        self.assertNotEqual(cm.exception.code, 0)


if __name__ == "__main__":
    unittest.main()

#!/usr/bin/env python


try:
    import unittest2 as unittest
except ImportError:
    import unittest

import ghstats
import sys


class TestStats(unittest.TestCase):

    def test_cli(self):
        """
        Test command line arguments.
        """
        count = ghstats.main_cli(["kefir500/apk-icon-editor", "-l", "-d"])
        self.assertTrue(count > 0)

    def test_releases(self):
        """
        Test statistics for all releases.
        """
        stats = ghstats.download_stats("kefir500", "apk-icon-editor", None, False, ghstats.get_env_token(), False)
        self.assertTrue(isinstance(stats, list))
        count = ghstats.get_stats_downloads(stats, False)
        self.assertTrue(count > 0)

    def test_release(self):
        """
        Test statistics for the latest release.
        """
        stats = ghstats.download_stats("kefir500", "apk-icon-editor", None, True, ghstats.get_env_token(), False)
        self.assertTrue(isinstance(stats, dict))
        count = ghstats.get_stats_downloads(stats, False)
        self.assertTrue(count > 0)

    def test_invalid_repo(self):
        """
        Test nonexistent repository.
        """
        with self.assertRaises(ghstats.GithubRepoError):
            ghstats.download_stats("kefir500", "foobar", None, False, ghstats.get_env_token(), True)

    def test_invalid_token(self):
        """
        Test invalid GitHub personal access token.
        """
        with self.assertRaises(ghstats.GithubTokenError):
            ghstats.download_stats("kefir500", "foobar", None, False, "FOOBAR", True)

    def test_exits(self):
        """
        Test if functions halt the script.
        """
        with self.assertRaises(SystemExit) as cm:
            ghstats.print_help()
        self.assertEqual(cm.exception.code if sys.version_info >= (2, 7) else cm.exception, 0)
        with self.assertRaises(SystemExit) as cm:
            ghstats.error("Test")
        self.assertNotEqual(cm.exception.code if sys.version_info >= (2, 7) else cm.exception, 0)


if __name__ == "__main__":
    unittest.main()

Fix test_exits() unit test for Python 2.6