rattail-project/rattail Files · tests/test

Files @ 177478f7d054

Branch filter:

Location: rattail-project/rattail/tests/test_util.py

177478f7d054 1.8 KiB text/x-python Show Annotation Show as Raw Download as Raw

lance

Database config/init overhaul.

This contains some not-very-atomic changes:

* Get rid of `get_session_class()` function and return to global `Session`
class approach.
* Primary database `Session` is now configured as part of command
initialization, by default.
* Make `config` object available to subcommands, and `Daemon` instances
(the beginning of the end for `edbob.config`!).
* Add `--stdout` and `--stderr` arguments to primary `Command`. These are
in turn made available to subcommands.
* Overhauled some subcommand logic per new patterns.
* Get rid of a few other random references to `edbob`.
* Added and improved several tests.
* Added ability to run tests using arbitrary database engine.


import os
import shutil
from unittest import TestCase

from rattail import util


class ImportTests(TestCase):

    def setUp(self):
        dirname = os.path.abspath(os.path.dirname(__file__))
        os.mkdir(os.path.join(dirname, 'foo'))
        with open(os.path.join(dirname, 'foo', '__init__.py'), 'w') as f:
            f.write('\n')
        with open(os.path.join(dirname, 'foo', 'bar.py'), 'w') as f:
            f.write('\n')
        os.mkdir(os.path.join(dirname, 'foo', 'baz'))
        with open(os.path.join(dirname, 'foo', 'baz', '__init__.py'), 'w') as f:
            f.write('\n')

    def tearDown(self):
        shutil.rmtree(os.path.join(os.path.dirname(__file__), 'foo'))

    def test_module_already_imported(self):
        util_module = util.import_module_path('rattail.util')
        self.assertTrue(util_module is util)

    def test_new_module(self):
        dirname = os.path.abspath(os.path.dirname(__file__))

        foo = util.import_module_path('tests.foo')
        self.assertEqual(foo.__file__, os.path.abspath(
                os.path.join(dirname, 'foo', '__init__.py')))

        bar = util.import_module_path('tests.foo.bar')
        self.assertEqual(bar.__file__, os.path.abspath(
                os.path.join(dirname, 'foo', 'bar.py')))

        baz = util.import_module_path('tests.foo.baz')
        self.assertEqual(baz.__file__, os.path.abspath(
                os.path.join(dirname, 'foo', 'baz', '__init__.py')))

    def test_load_object(self):
        with open(os.path.join(os.path.dirname(__file__), 'foo', 'baz', '__init__.py'), 'w') as f:
            f.write("""

somevar = 42

def somefunc():
    return somevar * 10

""")

        somevar = util.load_object('tests.foo.baz:somevar')
        self.assertEqual(somevar, 42)

        somefunc = util.load_object('tests.foo.baz:somefunc')
        self.assertEqual(somefunc(), 420)