fs-uae-wrapper/tests/test_wrapper.py

import os
import sys
from tempfile import mkstemp, mkdtemp
from unittest import TestCase
import shutil

try:
    from unittest import mock
except ImportError:
    import mock

from fs_uae_wrapper import wrapper


class TestWrapper(TestCase):

    def setUp(self):
        fd, self.fname = mkstemp()
        self.dirname = mkdtemp()
        os.close(fd)
        self._argv = sys.argv[:]
        sys.argv = ['fs-uae-wrapper']
        self.curdir = os.path.abspath(os.curdir)

    def tearDown(self):
        os.chdir(self.curdir)
        shutil.rmtree(self.dirname)
        os.unlink(self.fname)
        sys.argv = self._argv[:]

    def test_get_config_options(self):

        configs = ["[conf]\nwrapper=foo\n",
                   "[conf]\n wrapper =foo\n",
                   "[conf]\n wrapper =    foo\n",
                   "[conf]\nwrapper = foo    \n"]

        for cfg in configs:
            with open(self.fname, 'w') as fobj:
                fobj.write(cfg)

            val = wrapper.get_config_options(self.fname)
            self.assertDictEqual(val, {'wrapper': 'foo'})

        with open(self.fname, 'w') as fobj:
            fobj.write("[conf]\nwraper=foo\n")
        conf = wrapper.get_config_options(self.fname)
        self.assertDictEqual(conf, {'wraper': 'foo'})

        with open(self.fname, 'w') as fobj:
            fobj.write("[conf]\nwrapper\n")
        conf = wrapper.get_config_options(self.fname)
        self.assertIsNone(conf)

        with open(self.fname, 'w') as fobj:
            fobj.write("[conf]\nfullscreen = 1\n")
        conf = wrapper.get_config_options(self.fname)
        self.assertDictEqual(conf, {'fullscreen': '1'})

        with open(self.fname, 'w') as fobj:
            fobj.write("[conf]\nwrapper= = = something went wrong\n")
        conf = wrapper.get_config_options(self.fname)
        self.assertDictEqual(conf, {'wrapper': '= = something went wrong'})

        with open(self.fname, 'w') as fobj:
            fobj.write("[conf]\nwrapper = =    \n")
        conf = wrapper.get_config_options(self.fname)
        self.assertDictEqual(conf, {'wrapper': '='})

        with open(self.fname, 'w') as fobj:
            fobj.write("[conf]\nwrapper =     \n")
        conf = wrapper.get_config_options(self.fname)
        self.assertDictEqual(conf, {'wrapper': ''})

    @mock.patch('fs_uae_wrapper.plain.run')
    def test_run(self, mock_plain_run):

        sys.argv.append('--help')
        self.assertRaises(SystemExit, wrapper.run)

        sys.argv.pop()
        self.assertRaises(SystemExit, wrapper.run)

        sys.argv.append('--fullscreen')
        sys.argv.append('--fade_out_duration=0')
        # will exit due to not found configuration
        self.assertRaises(SystemExit, wrapper.run)

        os.chdir(self.dirname)
        with open('Config.fs-uae', 'w') as fobj:
            fobj.write('\n')

        wrapper.run()
        mock_plain_run.called_once_with('Config.fs-uae',
                                        ['--fullscreen',
                                         '--fade_out_duration=0'],
                                        [])

        # This will obviously fail for nonexistent module
        sys.argv.append('--wrapper=dummy_wrapper')
        self.assertRaises(SystemExit, wrapper.run)

    def test_run_wrong_conf(self):

        os.chdir(self.dirname)
        with open('Config.fs-uae', 'w') as fobj:
            fobj.write('foo\n')

        self.assertRaises(SystemExit, wrapper.run)

    def test_parse_option(self):

        # commandline origin
        self.assertEqual(wrapper.parse_option('--fullscreen'),
                         ('fullscreen', '1'))

        self.assertEqual(wrapper.parse_option('--fade_out_duration=0'),
                         ('fade_out_duration', '0'))

        # pass the wrong parameter to fs-uae
        self.assertEqual(wrapper.parse_option('-typo=true'), ('-typo', 'true'))

        # pass the wrong parameter to fs-uae again
        self.assertEqual(wrapper.parse_option('typo=true'), ('typo', 'true'))

        # We have no idea what to do with this - might be a conf file
        self.assertEqual(wrapper.parse_option('this-is-odd'), (None, None))

    def test_parse_args(self):

        # Looking for configuration file... first, we have nothing
        self.assertEqual(wrapper.parse_args(), (None, [], {}))

        # still no luck - nonexistent file
        sys.argv.append('there-is-no-config.fs-uae')
        self.assertEqual(wrapper.parse_args(), (None, [], {}))

        # lets make it
        os.chdir(self.dirname)
        with open('there-is-no-config.fs-uae', 'w') as fobj:
            fobj.write('\n')

        self.assertEqual(wrapper.parse_args(),
                         ('there-is-no-config.fs-uae', [], {}))

        # remove argument, try to find default one
        sys.argv.pop()
        self.assertListEqual(sys.argv, ['fs-uae-wrapper'])

        with open('Config.fs-uae', 'w') as fobj:
            fobj.write('\n')

        self.assertEqual(wrapper.parse_args(), ('Config.fs-uae', [], {}))

        # add --wrapper-foo and --wrapper-bar options
        sys.argv.extend(['--wrapper=plain', '--wrapper_foo=1',
                         '--wrapper_bar=false'])
        self.assertListEqual(sys.argv,
                             ['fs-uae-wrapper', '--wrapper=plain',
                              '--wrapper_foo=1', '--wrapper_bar=false'])

        with open('Config.fs-uae', 'w') as fobj:
            fobj.write('\n')

        conf, fsopts, wrapopts = wrapper.parse_args()
        self.assertEqual(conf, 'Config.fs-uae')
        self.assertListEqual(fsopts, [])
        self.assertDictEqual(wrapopts, {'wrapper': 'plain',
                                        'wrapper_foo': '1',
                                        'wrapper_bar': 'false'})

        # mix wrapper* params in commandline and config
        sys.argv = ['fs-uae-wrapper',
                    '--wrapper=plain',
                    '--wrapper_bar=false',
                    '--fullscreen',
                    '--fast_memory=4096']
        with open('Config.fs-uae', 'w') as fobj:
            fobj.write('[conf]\nwrapper = cd32\nwrapper_foo = /some/path\n')

        conf, fsopts, wrapopts = wrapper.parse_args()
        self.assertEqual(conf, 'Config.fs-uae')
        self.assertListEqual(fsopts, ['--fullscreen', '--fast_memory=4096'])
        self.assertDictEqual(wrapopts, {'wrapper': 'plain',
                                        'wrapper_bar': 'false'})