Source code for sqla_inspect.tests.test_base

# -*- coding: utf-8 -*-
# * Authors:
#       * Arezki Feth <f.a@majerti.fr>;
#       * Miotte Julien <j.m@majerti.fr>;
#       * TJEBBES Gaston <g.t@majerti.fr>

from sqla_inspect.base import FormatterRegistry


[docs]class Dummy(object): pass
[docs]def dummy_func(a): return a
[docs]def test_registry(): f = FormatterRegistry() assert f.get_formatter(Dummy()) is None f.add_formatter(Dummy, dummy_func) assert f.get_formatter(Dummy()) is dummy_func assert f.get_formatter(Dummy(), 'non_existingkey') is dummy_func f = FormatterRegistry() f.add_formatter(Dummy, dummy_func, 'key') assert f.get_formatter(Dummy(), 'key') is dummy_func assert f.get_formatter(Dummy()) is None