pwman3/tests/test_config.py

# ============================================================================
# This file is part of Pwman3.
#
# Pwman3 is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License, version 2
# as published by the Free Software Foundation;
#
# Pwman3 is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Pwman3; if not, write to the Free Software
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
# ============================================================================
# Copyright (C) 2014 Oz Nahum Tiram <nahumoz@gmail.com>
# ============================================================================

import os
import sys
import unittest
import unittest.mock
from pwman.util import config
from pwman.util.config import find_config_dir


if sys.version_info.major > 2:
    from configparser import NoSectionError
else:
    from ConfigParser import NoSectionError

with open('testfile.conf', 'w') as f:
    f.write("""
[Global]
xsel = /usr/bin/xsel
colors = yes
umask = 0100
cls_timeout = 5

[Database]
type = SQLite
""")


class TestFindConfigWindows(unittest.TestCase):

    def test_windows_platfrom(self):

        with unittest.mock.patch('platform.system') as pl:
            pl.return_value = 'Windows'
            os.environ['APPDATA'] = 'balls'
            cdir = config.find_config_dir('zzzz')

            self.assertEqual(
                os.path.expandvars(os.path.join('$APPDATA', 'zzzz')),
                cdir)


class TestFindConfigCompat(unittest.TestCase):

    c_path = os.path.expanduser("~/.zzzz_compat_posix")

    def setUp(self):
        if not os.path.exists(self.c_path):
            os.mkdir(self.c_path)

    def tearDown(self):
        os.rmdir(self.c_path)

    def test_compat(self):
        cdir = config.find_config_dir('zzzz_compat_posix')
        self.assertEqual(self.c_path, cdir)


class TestFindConfigXDG(unittest.TestCase):

    c_path = os.path.expanduser("~/.zzzz_posix")

    def setUp(self):
        if os.path.exists(self.c_path):
            os.rmdir(self.c_path)

    def test_new_scheme(self):
        # assert we get xdg_fine with Linux
        cdir = config.find_config_dir('zzzz_posix')
        self.assertEqual(cdir, os.path.expanduser("~/.config/zzzz_posix"))


class TestConfig(unittest.TestCase):

    @classmethod
    def tearDownClass(cls):
        for item in ('testfile.conf', 'wrong_conf.conf', 'dummy.cfg',
                     'import_file.csv'):
            try:
                os.unlink(item)
            except OSError:
                continue

    def setUp(self):
        self.conf = config.Config(filename='testfile.conf',
                                  defaults=config.default_config)

    def test_has_defaults(self):
        self.assertTrue(self.conf.parser.has_section('Readline'))

    def test_has_user_history(self):
        cdir = find_config_dir('pwman')
        path = os.path.join(cdir, "history")
        config = self.conf.get_value('Readline', 'history')
        self.assertEqual(path, config)

    def test_has_user_db(self):
        self.assertNotEqual(os.path.join(config.find_config_dir("pwman"),'pwman.db'),
                            self.conf.get_value('Database', 'filename'))

    def test_wrong_config(self):
        with open('wrong_conf.conf', 'w') as f:
            f.write("""
[Encryption
algorithm = Blowfish
""")
        self.assertRaises(config.ConfigException, config.Config,
                          'wrong_conf.conf')

    def test_set_value(self):
        self.conf.set_value('Global', 'colors', 'no')
        self.assertEqual('no', self.conf.get_value('Global', 'colors'))

    def test_set_value_wrong(self):
        self.assertRaises(NoSectionError,
                          self.conf.set_value, *('Error', 'colors', 'no'))

    def test_get_pass_conf(self):
        ans = config.get_pass_conf(self.conf)
        self.assertFalse(any(ans))

    def test_open_file(self):
        cfg = config.Config("DoesNotExist")
        cfg.save()
        self.assertTrue(os.path.exists("DoesNotExist"))
        os.unlink("DoesNotExist")


if __name__ == '__main__':
    unittest.main(verbosity=2)