#!/usr/bin/env python3
# -*- coding: utf-8 -*-
#
#  This file is part of gns-deb-diff.
#
#  gns-deb-diff is under the Public Domain. See
#  <https://creativecommons.org/publicdomain/zero/1.0>

import builtins
import json
import os
import subprocess
import sys
import time
import xmlrpc.client

import requests

import gd_diff

from io import StringIO
from os import path
from shutil import rmtree
from unittest import mock

from nose.tools import *

from gd_diff import *
from gns_deb_diff._version import __version__


class TestGdDiff(object):

    def setup(self):
        """Setup method for this class."""
        self.pkgs_file = 'tests/files/parkes-pkgs.list'
        self.small_pkgs_file = 'tests/files/small-parkes-pkgs.list'
        self.tiny_pkgs_file = 'tests/files/tiny-parkes-pkgs.list'
        self.pkgs_file_ne = 'tests/nonexistent-file.list'
        self.gns_pkgs_dir = 'tests/gns-pkgs'
        self.test_home = 'tests/HOME'
        self.test_w_file = os.path.join(self.test_home, 'w_file')

        self.stderr_orig = sys.stderr

        # make test home
        os.mkdir(self.test_home, mode=0o700)

        self.env_func = lambda e: self.test_home


    def test_read_file_success(self):
        f_content = read_file(self.pkgs_file)

        assert isinstance(f_content, str)
        assert_equal(len(f_content.split('\n')), 82)


    @raises(SystemExit)
    def test_read_file_error(self):
        with open(os.devnull, 'w') as sys.stderr:
            f_content = read_file(self.pkgs_file_ne)


    def test_write_file(self):
        content = 'One Goodbye\n Stealing Romance'
        write_file(self.test_w_file, content)

        assert read_file(self.test_w_file) == content


    def test_execute_success(self):
        cmd = 'python --version'
        cp = execute(cmd, out=subprocess.PIPE)

        assert cp.returncode == 0
        assert cp.stdout.split()[0] == b'Python'
        assert cp.stdout.split()[1].startswith(b'3.5')


    def test_execute_cmderror(self):
        cmd = 'bzr cat bzr://bzr.sv.gnu.org/gnewsense/packages-parkes/nonexistent-packages/debian/README.gNewSense'
        cp = execute(cmd, err=subprocess.PIPE)

        assert cp.returncode == 3
        assert cp.stderr.startswith(b'bzr: ERROR:')


    @raises(SystemExit)
    def test_execute_raise_exception(self):
        cmd = 'cornhole computers'
        with open(os.devnull, 'w') as sys.stderr:
            cp = execute(cmd)


    def test_read_packages(self):
        pkgs = read_packages(self.pkgs_file)
        assert len(pkgs) == 82


    def test_read_packages_sanity(self):
        pkgs = read_packages(self.pkgs_file)

        for pkg in pkgs:
            assert not ' ' in pkg


    @raises(SystemExit)
    def test_get_packages_error(self):
        res = get_packages('foo_release')


    def test_get_packages(self):
        pkgs = get_packages('parkes')

        assert_equal(pkgs, 'antlr\napt\napt-setup\nautoconf\nautoconf2.59\nautoconf2.64\nbacula\nbase-files\nbase-installer\nbatik\ncairomm\ncdebootstrap\ncfitsio3\nchoose-mirror\nclaws-mail\ndb4.6\ndb4.7\ndb4.8\ndebian-cd\ndebian-edu\ndebian-installer\ndebian-installer-launcher\ndebootstrap\ndesktop-base\ndoc-linux\ndoc-linux-hr\ndoc-linux-it\ndoc-linux-ja\ndoc-linux-pl\nenscript\nepiphany-browser\nfop\nfreetype\ngalaxia\ngdm3\nglibmm2.4\ngnewsense-archive-keyring\ngnome-desktop\ngtkmm2.4\nicedove\niceweasel\nkde4libs\nkdebase\nkdebase-workspace\nkdenetwork\nkernel-wedge\nlensfun\nliferea\nlintian\nlinux-2.6\nlinux-kernel-di-amd64-2.6\nlinux-kernel-di-i386-2.6\nlinux-latest-2.6\nlive-build\nlive-config\nmeta-gnome2\nmplayer\nnet-retriever\nobjcryst-fox\nopenbox-themes\nopenjdk-6\nopenoffice.org\npangomm\nperl-tk\npkgsel\npopularity-contest\npsutils\npython-apt\nscreenlets\nsip4-qt3\nsoftware-center\ntcl8.4\ntcl8.5\ntexlive-extra\ntk8.4\ntk8.5\nupdate-manager\nvim\nwmaker\nxchat\nxdm\nxorg-server\nxserver-xorg-video-siliconmotion\nyeeloong-base\n')


    def test_config_dir(self):
        with mock.patch('os.getenv', new=self.env_func):
            c_dir = config_dir()
            assert_equal(c_dir, os.path.join(self.test_home,
                                             '.config', 'gns-deb-diff'))
            assert_equal(os.path.isdir(c_dir), True)


    def test_config_file(self):
        with mock.patch('os.getenv', new=self.env_func):
            c_file = config_file()
            assert_equal(c_file, os.path.join(self.test_home, '.config',
                                             'gns-deb-diff', 'config'))


    def test_read_config_file_fail(self):
        with mock.patch('os.getenv', new=self.env_func):
            config = read_config_file()
            assert_equal(config, False)


    def test_read_config_file_success(self):
        with mock.patch('os.getenv', new=self.env_func):
            c_file = config_file()

            # first write sample config file.
            json.dump({'user': 'usrnm', 'pass': 'weasaspeciesrfckd'},
                      open(c_file, 'w'))

            # now on to the test.
            config = read_config_file()
            assert_equal(config['user'], 'usrnm')
            assert_equal(config['pass'], 'weasaspeciesrfckd')


    def test_pkgs_dir(self):
        with mock.patch('os.getenv', new=self.env_func):
            pd = pkgs_dir()
            assert_equal(pd, os.path.join(self.test_home, '.config',
                                          'gns-deb-diff', 'pkgs'))
            assert_equal(os.path.isdir(pd), True)


    def test_mk_pkgs_list(self):
        with mock.patch('os.getenv', new=self.env_func):
            pkgs_file = mk_pkgs_list('parkes')
            # test
            assert_equal(read_file(pkgs_file), 'antlr\napt\napt-setup\nautoconf\nautoconf2.59\nautoconf2.64\nbacula\nbase-files\nbase-installer\nbatik\ncairomm\ncdebootstrap\ncfitsio3\nchoose-mirror\nclaws-mail\ndb4.6\ndb4.7\ndb4.8\ndebian-cd\ndebian-edu\ndebian-installer\ndebian-installer-launcher\ndebootstrap\ndesktop-base\ndoc-linux\ndoc-linux-hr\ndoc-linux-it\ndoc-linux-ja\ndoc-linux-pl\nenscript\nepiphany-browser\nfop\nfreetype\ngalaxia\ngdm3\nglibmm2.4\ngnewsense-archive-keyring\ngnome-desktop\ngtkmm2.4\nicedove\niceweasel\nkde4libs\nkdebase\nkdebase-workspace\nkdenetwork\nkernel-wedge\nlensfun\nliferea\nlintian\nlinux-2.6\nlinux-kernel-di-amd64-2.6\nlinux-kernel-di-i386-2.6\nlinux-latest-2.6\nlive-build\nlive-config\nmeta-gnome2\nmplayer\nnet-retriever\nobjcryst-fox\nopenbox-themes\nopenjdk-6\nopenoffice.org\npangomm\nperl-tk\npkgsel\npopularity-contest\npsutils\npython-apt\nscreenlets\nsip4-qt3\nsoftware-center\ntcl8.4\ntcl8.5\ntexlive-extra\ntk8.4\ntk8.5\nupdate-manager\nvim\nwmaker\nxchat\nxdm\nxorg-server\nxserver-xorg-video-siliconmotion\nyeeloong-base\n')


    def test_readmes_dir(self):
        with mock.patch('os.getenv', new=self.env_func):
            rd_parkes = readmes_dir('parkes')
            assert_equal(rd_parkes, os.path.join(self.test_home, '.config',
                                                 'gns-deb-diff', 'readmes',
                                                 'parkes'))
            assert_equal(os.path.isdir(rd_parkes), True)


    def test_wiki_page_dir(self):
        with mock.patch('os.getenv', new=self.env_func):
            wd_parkes = wiki_page_dir('parkes')
            assert_equal(wd_parkes, os.path.join(self.test_home, '.config',
                                                 'gns-deb-diff', 'wiki-page',
                                                 'parkes'))
            assert_equal(os.path.isdir(wd_parkes), True)


    def test_wiki_page_path(self):
        with mock.patch('os.getenv', new=self.env_func):
            assert_equal(wiki_page_path('parkes'),
                         path.join(os.getenv('HOME'), '.config',
                                             'gns-deb-diff', 'wiki-page',
                                             'parkes', 'wiki.page'))


    def test_write_wiki_page(self):
        with mock.patch('os.getenv', new=self.env_func):
            release = 'parkes'
            write_wiki_page(release, 'wiki content')
            wp_file = os.path.join(wiki_page_dir(release), 'wiki.page')
            assert_equal(read_file(wp_file), 'wiki content')


    def test_configured_p_no(self):
        with mock.patch('os.getenv', new=self.env_func):
            configured = configured_p()
            assert_equal(configured, False)


    def test_configured_p_yes(self):
        with mock.patch('os.getenv', new=self.env_func):
            c_path = config_dir()
            c_file = config_file()

            open(c_file, 'w').close()

            configured = configured_p()
            assert_equal(configured, True)


    def test_configure(self):
        inputs = ['usrnm', 'weasaspeciesrfckd']
        def mock_input(p):
            return inputs.pop(0)

        with mock.patch('os.getenv', new=self.env_func), \
             mock.patch('builtins.input', new=mock_input):
            configure()

            # read config file
            config = json.load(open(config_file(), 'r'))

            # tests
            assert_equal(config['user'], 'usrnm')
            assert_equal(config['pass'], 'weasaspeciesrfckd')
            assert_equal(oct(os.stat(config_file()).st_mode), '0o100600')


    def test_save_gns_readme(self):
        with mock.patch('os.getenv', new=self.env_func):
            cmd = 'bzr cat bzr://bzr.sv.gnu.org/gnewsense/packages-parkes/antlr/debian/README.gNewSense'
            cp = execute(cmd, out=subprocess.PIPE)
            readme_content = cp.stdout.decode() # convert to str

            # save it
            save_gns_readme(readme_content, 'parkes', 'antlr')

            gns_readme_file = path.join(self.test_home, '.config',
                                        'gns-deb-diff', 'readmes',
                                        'parkes', 'antlr', 'debian',
                                        'README.gNewSense')
            with open(gns_readme_file, 'rb') as f:
                assert f.read() == b'Changed-From-Debian: Removed example with non-free files.\nChange-Type: Modified\n\nFor gNewSense, the non-free unicode.IDENTs files are *actually* removed (see\nalso README.source). See gNewSense bug #34218 for details.\n'


    def test_save_gns_readme_double(self):
        with mock.patch('os.getenv', new=self.env_func):
            cmd = 'bzr cat bzr://bzr.sv.gnu.org/gnewsense/packages-parkes/antlr/debian/README.gNewSense'
            cp = execute(cmd, out=subprocess.PIPE)
            readme_content = cp.stdout.decode() # convert to str

            # save it twice
            save_gns_readme(readme_content, 'parkes', 'antlr')
            save_gns_readme(readme_content, 'parkes', 'antlr')

            gns_readme_file = path.join(self.test_home, '.config',
                                        'gns-deb-diff', 'readmes',
                                        'parkes', 'antlr', 'debian',
                                        'README.gNewSense')
            with open(gns_readme_file, 'rb') as f:
                assert f.read() == b'Changed-From-Debian: Removed example with non-free files.\nChange-Type: Modified\n\nFor gNewSense, the non-free unicode.IDENTs files are *actually* removed (see\nalso README.source). See gNewSense bug #34218 for details.\n'


    def test_slurp_gns_readme_success(self):
        with mock.patch('os.getenv', new=self.env_func):
            saved = slurp_gns_readme('parkes', 'antlr')
            assert saved == True

            gns_readme_file = path.join(self.test_home, '.config',
                                        'gns-deb-diff', 'readmes',
                                        'parkes', 'antlr', 'debian',
                                        'README.gNewSense')
            with open(gns_readme_file, 'rb') as f:
                assert f.read() == b'Changed-From-Debian: Removed example with non-free files.\nChange-Type: Modified\n\nFor gNewSense, the non-free unicode.IDENTs files are *actually* removed (see\nalso README.source). See gNewSense bug #34218 for details.\n'


    def test_slurp_gns_readme_error(self):
        with mock.patch('os.getenv', new=self.env_func):
            saved = slurp_gns_readme('parkes', 'non-existent-pkg')
            assert saved == False

            gns_readme_file = path.join(self.test_home, '.config',
                                        'gns-deb-diff', 'readmes',
                                        'parkes', 'non-existent-pkg',
                                        'debian', 'README.gNewSense')
            assert not path.exists(gns_readme_file)


    def test_slurp_all_gns_readmes(self):
        with mock.patch('os.getenv', new=self.env_func):
            pkgs = read_packages(self.small_pkgs_file)

            # expected packages with no readmes
            expected_pkgs_noreadmes = [
                'pkg-with-no-readme',
                'another-pkg-no-readme',
            ]

            pkgs_noreadmes = slurp_all_gns_readmes('parkes', pkgs)
            assert_equal(pkgs_noreadmes, expected_pkgs_noreadmes)


    def test_read_gns_readme(self):
        with mock.patch('os.getenv', new=self.env_func):
            # first download the antlr readme
            saved = slurp_gns_readme('parkes', 'antlr')
            assert saved

            antlr_readme_content = read_gns_readme('parkes', 'antlr')
            expected_antlr_readme_content = 'Changed-From-Debian: Removed example with non-free files.\nChange-Type: Modified\n\nFor gNewSense, the non-free unicode.IDENTs files are *actually* removed (see\nalso README.source). See gNewSense bug #34218 for details.\n'
            assert_equal(antlr_readme_content, expected_antlr_readme_content)


    def test_read_gns_readme_none(self):
        with mock.patch('os.getenv', new=self.env_func):
            readme_content = read_gns_readme('parkes', 'non-existent-pkg')
            assert_equal(readme_content, None)


    def test_slurp_fields_from_readme(self):
        readme_content = ''
        field_values = slurp_fields_from_readme(readme_content)
        assert_equal(field_values['Change-Type'], None)
        assert_equal(field_values['Changed-From-Debian'], None)

        readme_content = 'Changed-From-Debian: \n'
        field_values = slurp_fields_from_readme(readme_content)
        assert_equal(field_values['Change-Type'], None)
        assert_equal(field_values['Changed-From-Debian'], None)

        readme_content = 'Change-Type: Deblob\n'
        field_values = slurp_fields_from_readme(readme_content)
        assert_equal(field_values['Change-Type'], 'Deblob')
        assert_equal(field_values['Changed-From-Debian'], None)

        readme_content = 'Changed-From-Debian: \nChange-Type: \n'
        field_values = slurp_fields_from_readme(readme_content)
        assert_equal(field_values['Change-Type'], None)
        assert_equal(field_values['Changed-From-Debian'], None)

        readme_content = 'Changed-From-Debian: Branding.    \nChange-Type: \n'
        field_values = slurp_fields_from_readme(readme_content)
        assert_equal(field_values['Change-Type'], None)
        assert_equal(field_values['Changed-From-Debian'], 'Branding.')

        readme_content = 'Changed-From-Debian: \nChange-Type: Modified'
        field_values = slurp_fields_from_readme(readme_content)
        assert_equal(field_values['Change-Type'], 'Modified')
        assert_equal(field_values['Changed-From-Debian'], None)

        readme_content = 'Changed-From-Debian:Fixed ambiguous use of the word free.\n\n\nChange-Type:Deblob\n'
        field_values = slurp_fields_from_readme(readme_content)
        assert_equal(field_values['Change-Type'], 'Deblob')
        assert_equal(field_values['Changed-From-Debian'],
                     'Fixed ambiguous use of the word free.')

        readme_content  = 'Changed-From-Debian: Removed example with non-free files.\nChange-Type: Modified\n\nFor gNewSense, the non-free unicode.IDENTs files are *actually* removed (see\nalso README.source). See gNewSense bug #34218 for details.\n'
        field_values = slurp_fields_from_readme(readme_content)
        assert_equal(field_values['Change-Type'], 'Modified')
        assert_equal(field_values['Changed-From-Debian'],
                     'Removed example with non-free files.')


    def test_get_wiki_page_data(self):
        def mock_mk_pkgs_list(r):
            return self.small_pkgs_file

        pkgs = [p.strip()
                for p in open(self.small_pkgs_file, 'r').read() \
                .split('\n')]

        with mock.patch('os.getenv', new=self.env_func), \
             mock.patch('gd_diff.mk_pkgs_list', new=mock_mk_pkgs_list):
            pkgs_noreadmes, table_data = get_wiki_page_data('parkes')

            for pkg in pkgs_noreadmes:
                assert pkg in ['pkg-with-no-readme', 'another-pkg-no-readme']

            for pkg, field_values in table_data.items():
                assert pkg in pkgs
                for field, value in field_values.items():
                    assert field in gd_diff.field_list
                    assert (type(value) == str or
                            value is None)

    def test_construct_table_row(self):
        row = construct_table_row('ucclia', 'antlr', 'Modified',
                                  'Removed example with non-free files')
        columns = row.split('||')[1:-1]
        assert_equal(columns[0], 'antlr')
        assert_equal(columns[1], 'Modified')
        assert_equal(columns[2], 'Removed example with non-free files')
        assert_equal(columns[3] , '[[http://bzr.savannah.gnu.org/lh/gnewsense/packages-ucclia/antlr/annotate/head:/debian/README.gNewSense|more_info]]')

        row = construct_table_row('ucclia', 'antlr', None,
                                  'Removed example with non-free files')
        columns = row.split('||')[1:-1]
        assert_equal(columns[0], 'antlr')
        assert_equal(columns[1], ' ')
        assert_equal(columns[2], 'Removed example with non-free files')
        assert_equal(columns[3] , '[[http://bzr.savannah.gnu.org/lh/gnewsense/packages-ucclia/antlr/annotate/head:/debian/README.gNewSense|more_info]]')

        row = construct_table_row('ucclia', 'antlr', 'Modified',
                                  None)
        columns = row.split('||')[1:-1]
        assert_equal(columns[0], 'antlr')
        assert_equal(columns[1], 'Modified')
        assert_equal(columns[2], ' ')
        assert_equal(columns[3] , '[[http://bzr.savannah.gnu.org/lh/gnewsense/packages-ucclia/antlr/annotate/head:/debian/README.gNewSense|more_info]]')

        row = construct_table_row('ucclia', 'antlr', None,
                                  None)
        columns = row.split('||')[1:-1]
        assert_equal(columns[0], 'antlr')
        assert_equal(columns[1], ' ')
        assert_equal(columns[2], ' ')
        assert_equal(columns[3] , '[[http://bzr.savannah.gnu.org/lh/gnewsense/packages-ucclia/antlr/annotate/head:/debian/README.gNewSense|more_info]]')


    def test_generate_wiki_table(self):
        def mock_mk_pkgs_list(r):
            return self.tiny_pkgs_file

        pkgs = [p.strip()
                for p in open(self.tiny_pkgs_file, 'r').read() \
                .split('\n')]

        expected_pkgs_noreadmes = ['pkg-with-no-readme',
                                   'another-pkg-no-readme']

        with mock.patch('os.getenv', new=self.env_func), \
             mock.patch('gd_diff.mk_pkgs_list', new=mock_mk_pkgs_list):
            pkgs_noreadmes, wiki_table = generate_wiki_table('parkes')

            for pkg in pkgs_noreadmes:
                assert pkg in expected_pkgs_noreadmes

            pkgs = set(pkgs) - set(expected_pkgs_noreadmes)

            for row in wiki_table.split('\n')[:-1]:
                cols = row.split('||')[1:]
                pkg = cols[0]

                assert pkg in pkgs
                assert cols[3] == '[[{}|more_info]]'.format(
                    readme_link_fmt.format('parkes', pkg))

                pkgs.remove(pkg)

            assert len(pkgs) == 0


    def test_gns_wiki_header(self):
        header = gns_wiki_header()
        expected_header = open('gns_deb_diff/data/wiki-header.txt', 'r').read()
        assert header == expected_header


    def test_generate_wiki_page(self):
        def mock_mk_pkgs_list(r):
            return self.tiny_pkgs_file

        with mock.patch('os.getenv', new=self.env_func), \
             mock.patch('gd_diff.mk_pkgs_list', new=mock_mk_pkgs_list):
            pkgs_noreadmes, wiki_page = generate_wiki_page('parkes')
            pkgs_noreadmes, wiki_table = generate_wiki_table('parkes')

            assert wiki_page == gns_wiki_header() + '\n' + wiki_table


    def test_read_wiki_page_returns_none_if_wikipage_nonexistent(self):
        with mock.patch('os.getenv', new=self.env_func):
            wiki_page_content = read_wiki_page('bogus-release')
            assert wiki_page_content == None


    def test_read_wiki_page_returns_content_if_wikipage_existent(self):
        with mock.patch('os.getenv', new=self.env_func):
            wp_file = wiki_page_path('bogus-release')

            # write some bogus content
            bogus_content = 'bogus content'
            with open(wp_file, 'w') as f:
                f.write(bogus_content)

            wiki_page_content = read_wiki_page('bogus-release')
            assert wiki_page_content == bogus_content


    def test_push_wiki_page(self):
        url = 'http://localhost:8080'
        user = 's'
        passw = 'gtzarko' # "get the zark out"; if you've been wondering.
        version = '3'
        content = 'test zarks! @ {}'.format(time.ctime())

        # ensure moin server is running
        try:
            r = requests.get(url)
            a = ''
            if(r.status_code != 200):
                return # skip testing
        except requests.ConnectionError:
            return # skip testing

        # test
        expected_barfs = [
            'auth success.', 'wiki page updated.',
            'auth success.', 'wiki page not updated.',
            'auth failure',
            'auth failure'
        ]
        barfs = []
        with mock.patch('sys.stdout', new=StringIO()) as output:
            try:
                # success
                push_wiki_page(url, user, passw, version, content)
                # another success
                push_wiki_page(url, user, passw, version, content)
                # auth error
                push_wiki_page(url, 'z', passw, version, content)
                # auth error
                push_wiki_page(url, user, 'zark_out', version, content)
                barfs = output.getvalue().strip('\n').split('\n')
            except SystemExit:
                return # zark it. wiki has disabled xmlrpc.

        assert barfs == expected_barfs


    def test_make_push(self):
        # mock args
        mock_sys_argv = ['gd-diff', 'parkes', '3']

        # mock `read_wiki_page` function
        expected_old_wiki_page = 'wiki content old content.'
        def mock_rwp(r):
            return expected_old_wiki_page

        # mock `generate_wiki_page` function
        expected_wiki_page = 'wiki content.'
        expected_pkgs_noreadmes = ['noreadmepkg']
        def mock_gwg(r):
            return expected_pkgs_noreadmes, expected_wiki_page

        # mock `configured_p` function
        def mock_cp():
            return False

        # mock `configure` function
        def mock_c():
            return

        # mock `read_config_file`
        expected_config = {'user': 'usrnm', 'pass': 'weasaspeciesrfckd'}
        def mock_rcf():
            return  expected_config

        # mock `write_wiki_page`
        def mock_wwp(r, wp):
            return

        # mock `push_wiki_page`
        def mock_pwp(u, usr, p, v, wp):
            return

        with mock.patch('os.getenv', new=self.env_func), \
             mock.patch('gd_diff.read_wiki_page', new=mock_rwp), \
             mock.patch('gd_diff.generate_wiki_page', new=mock_gwg), \
             mock.patch('gd_diff.configured_p', new=mock_cp), \
             mock.patch('gd_diff.configure', new=mock_c), \
             mock.patch('gd_diff.read_config_file', new=mock_rcf), \
             mock.patch('gd_diff.write_wiki_page', new=mock_wwp), \
             mock.patch('gd_diff.push_wiki_page', new=mock_pwp), \
             mock.patch('sys.argv', new=mock_sys_argv):
            args = get_args()
            config, pkgs_noreadmes, old_wiki_page, wiki_page = make_push(args)

            assert expected_config == config
            assert expected_pkgs_noreadmes == pkgs_noreadmes
            assert expected_old_wiki_page == old_wiki_page
            assert expected_wiki_page == wiki_page


    def test_get_args_gd_diff_version(self):
        mock_sys_argv = ['gd-diff', '--version']
        with mock.patch('sys.stdout', new=StringIO()) as output, \
             mock.patch('sys.argv', new=mock_sys_argv):
            try:
                args = get_args()
            except SystemExit as se:
                assert_equal(__version__, output.getvalue().rstrip('\n'))


    def test_get_args(self):
        mock_sys_argv = ['gd-diff', 'parkes', '3']
        with mock.patch('sys.argv', new=mock_sys_argv):
            args = get_args()
            assert args.release == 'parkes'
            assert args.version == 3


    def teardown(self):
        """Teardown method for this class."""
        if(path.exists(self.gns_pkgs_dir)):
            os.chmod(self.gns_pkgs_dir, mode=0o700)
            rmtree(self.gns_pkgs_dir)

        if(path.exists(self.test_home)):
            rmtree(self.test_home)

        # restore sys.stderr
        sys.stderr = self.stderr_orig