3
0
Fork 0

fix: format all code with black

and from now on should not deviate from that...
This commit is contained in:
Lance Edgar 2025-08-30 21:25:44 -05:00
parent 49f9a0228b
commit a6bb538ce9
59 changed files with 2762 additions and 2131 deletions

View file

@ -26,6 +26,7 @@ from wuttjamaican.batch import BatchHandler
class MockBatchHandler(BatchHandler):
pass
class AnotherBatchHandler(BatchHandler):
pass
@ -34,14 +35,14 @@ class TestAppHandler(FileTestCase):
def setUp(self):
self.setup_files()
self.config = WuttaConfig(appname='wuttatest')
self.config = WuttaConfig(appname="wuttatest")
self.app = mod.AppHandler(self.config)
self.config.app = self.app
def test_init(self):
self.assertIs(self.app.config, self.config)
self.assertEqual(self.app.handlers, {})
self.assertEqual(self.app.appname, 'wuttatest')
self.assertEqual(self.app.appname, "wuttatest")
def test_get_enum(self):
self.assertIs(self.app.get_enum(), wuttjamaican.enum)
@ -50,48 +51,50 @@ class TestAppHandler(FileTestCase):
# just confirm the method works on a basic level; the
# underlying function is tested elsewhere
obj = self.app.load_object('wuttjamaican.util:UNSPECIFIED')
obj = self.app.load_object("wuttjamaican.util:UNSPECIFIED")
self.assertIs(obj, UNSPECIFIED)
def test_get_appdir(self):
mockdir = self.mkdir('mockdir')
mockdir = self.mkdir("mockdir")
# default appdir
with patch.object(sys, 'prefix', new=mockdir):
with patch.object(sys, "prefix", new=mockdir):
# default is returned by default
appdir = self.app.get_appdir()
self.assertEqual(appdir, os.path.join(mockdir, 'app'))
self.assertEqual(appdir, os.path.join(mockdir, "app"))
# but not if caller wants config only
appdir = self.app.get_appdir(configured_only=True)
self.assertIsNone(appdir)
# also, cannot create if appdir path not known
self.assertRaises(ValueError, self.app.get_appdir, configured_only=True, create=True)
self.assertRaises(
ValueError, self.app.get_appdir, configured_only=True, create=True
)
# configured appdir
self.config.setdefault('wuttatest.appdir', mockdir)
self.config.setdefault("wuttatest.appdir", mockdir)
appdir = self.app.get_appdir()
self.assertEqual(appdir, mockdir)
# appdir w/ subpath
appdir = self.app.get_appdir('foo', 'bar')
self.assertEqual(appdir, os.path.join(mockdir, 'foo', 'bar'))
appdir = self.app.get_appdir("foo", "bar")
self.assertEqual(appdir, os.path.join(mockdir, "foo", "bar"))
# subpath is created
self.assertEqual(len(os.listdir(mockdir)), 0)
appdir = self.app.get_appdir('foo', 'bar', create=True)
self.assertEqual(appdir, os.path.join(mockdir, 'foo', 'bar'))
self.assertEqual(os.listdir(mockdir), ['foo'])
self.assertEqual(os.listdir(os.path.join(mockdir, 'foo')), ['bar'])
appdir = self.app.get_appdir("foo", "bar", create=True)
self.assertEqual(appdir, os.path.join(mockdir, "foo", "bar"))
self.assertEqual(os.listdir(mockdir), ["foo"])
self.assertEqual(os.listdir(os.path.join(mockdir, "foo")), ["bar"])
def test_make_appdir(self):
# appdir is created, and 3 subfolders added by default
tempdir = tempfile.mkdtemp()
appdir = os.path.join(tempdir, 'app')
appdir = os.path.join(tempdir, "app")
self.assertFalse(os.path.exists(appdir))
self.app.make_appdir(appdir)
self.assertTrue(os.path.exists(appdir))
@ -107,23 +110,30 @@ class TestAppHandler(FileTestCase):
shutil.rmtree(tempdir)
def test_render_mako_template(self):
output_conf = self.write_file('output.conf', '')
template = Template("""\
output_conf = self.write_file("output.conf", "")
template = Template(
"""\
[wutta]
app_title = WuttaTest
""")
"""
)
output = self.app.render_mako_template(template, {}, output_path=output_conf)
self.assertEqual(output, """\
self.assertEqual(
output,
"""\
[wutta]
app_title = WuttaTest
""")
""",
)
with open(output_conf, 'rt') as f:
with open(output_conf, "rt") as f:
self.assertEqual(f.read(), output)
def test_resource_path(self):
result = self.app.resource_path('wuttjamaican:templates')
self.assertEqual(result, os.path.join(os.path.dirname(mod.__file__), 'templates'))
result = self.app.resource_path("wuttjamaican:templates")
self.assertEqual(
result, os.path.join(os.path.dirname(mod.__file__), "templates")
)
def test_make_session(self):
try:
@ -138,11 +148,12 @@ app_title = WuttaTest
short_session = MagicMock()
mockdb = MagicMock(short_session=short_session)
with patch.dict('sys.modules', **{'wuttjamaican.db': mockdb}):
with patch.dict("sys.modules", **{"wuttjamaican.db": mockdb}):
with self.app.short_session(foo='bar') as s:
with self.app.short_session(foo="bar") as s:
short_session.assert_called_once_with(
foo='bar', factory=self.app.make_session)
foo="bar", factory=self.app.make_session
)
def test_get_setting(self):
try:
@ -152,22 +163,26 @@ app_title = WuttaTest
pytest.skip("test is not relevant without sqlalchemy")
Session = orm.sessionmaker()
engine = sa.create_engine('sqlite://')
engine = sa.create_engine("sqlite://")
session = Session(bind=engine)
session.execute(sa.text("""
session.execute(
sa.text(
"""
create table setting (
name varchar(255) primary key,
value text
);
"""))
"""
)
)
session.commit()
value = self.app.get_setting(session, 'foo')
value = self.app.get_setting(session, "foo")
self.assertIsNone(value)
session.execute(sa.text("insert into setting values ('foo', 'bar');"))
value = self.app.get_setting(session, 'foo')
self.assertEqual(value, 'bar')
value = self.app.get_setting(session, "foo")
self.assertEqual(value, "bar")
def test_save_setting(self):
try:
@ -177,25 +192,29 @@ app_title = WuttaTest
pytest.skip("test is not relevant without sqlalchemy")
Session = orm.sessionmaker()
engine = sa.create_engine('sqlite://')
engine = sa.create_engine("sqlite://")
session = Session(bind=engine)
session.execute(sa.text("""
session.execute(
sa.text(
"""
create table setting (
name varchar(255) primary key,
value text
);
"""))
"""
)
)
session.commit()
# value null by default
value = self.app.get_setting(session, 'foo')
value = self.app.get_setting(session, "foo")
self.assertIsNone(value)
# unless we save a value
self.app.save_setting(session, 'foo', '1')
self.app.save_setting(session, "foo", "1")
session.commit()
value = self.app.get_setting(session, 'foo')
self.assertEqual(value, '1')
value = self.app.get_setting(session, "foo")
self.assertEqual(value, "1")
def test_delete_setting(self):
try:
@ -205,43 +224,48 @@ app_title = WuttaTest
pytest.skip("test is not relevant without sqlalchemy")
Session = orm.sessionmaker()
engine = sa.create_engine('sqlite://')
engine = sa.create_engine("sqlite://")
session = Session(bind=engine)
session.execute(sa.text("""
session.execute(
sa.text(
"""
create table setting (
name varchar(255) primary key,
value text
);
"""))
"""
)
)
session.commit()
# value null by default
value = self.app.get_setting(session, 'foo')
value = self.app.get_setting(session, "foo")
self.assertIsNone(value)
# unless we save a value
self.app.save_setting(session, 'foo', '1')
self.app.save_setting(session, "foo", "1")
session.commit()
value = self.app.get_setting(session, 'foo')
self.assertEqual(value, '1')
value = self.app.get_setting(session, "foo")
self.assertEqual(value, "1")
# but then if we delete it, should be null again
self.app.delete_setting(session, 'foo')
self.app.delete_setting(session, "foo")
session.commit()
value = self.app.get_setting(session, 'foo')
value = self.app.get_setting(session, "foo")
self.assertIsNone(value)
def test_continuum_is_enabled(self):
# false by default
with patch.object(self.app, 'providers', new={}):
with patch.object(self.app, "providers", new={}):
self.assertFalse(self.app.continuum_is_enabled())
# but "any" provider technically could enable it...
class MockProvider:
def continuum_is_enabled(self):
return True
with patch.object(self.app, 'providers', new={'mock': MockProvider()}):
with patch.object(self.app, "providers", new={"mock": MockProvider()}):
self.assertTrue(self.app.continuum_is_enabled())
def test_model(self):
@ -250,7 +274,7 @@ app_title = WuttaTest
except ImportError:
pytest.skip("test not relevant without sqlalchemy")
self.assertNotIn('model', self.app.__dict__)
self.assertNotIn("model", self.app.__dict__)
self.assertIs(self.app.model, model)
def test_get_model(self):
@ -262,20 +286,20 @@ app_title = WuttaTest
self.assertIs(self.app.get_model(), model)
def test_get_title(self):
self.assertEqual(self.app.get_title(), 'WuttJamaican')
self.assertEqual(self.app.get_title(), "WuttJamaican")
def test_get_node_title(self):
# default
self.assertEqual(self.app.get_node_title(), 'WuttJamaican')
self.assertEqual(self.app.get_node_title(), "WuttJamaican")
# will fallback to app title
self.config.setdefault('wuttatest.app_title', "WuttaTest")
self.assertEqual(self.app.get_node_title(), 'WuttaTest')
self.config.setdefault("wuttatest.app_title", "WuttaTest")
self.assertEqual(self.app.get_node_title(), "WuttaTest")
# will read from config
self.config.setdefault('wuttatest.node_title', "WuttaNode")
self.assertEqual(self.app.get_node_title(), 'WuttaNode')
self.config.setdefault("wuttatest.node_title", "WuttaNode")
self.assertEqual(self.app.get_node_title(), "WuttaNode")
def test_get_node_type(self):
@ -283,8 +307,8 @@ app_title = WuttaTest
self.assertIsNone(self.app.get_node_type())
# will read from config
self.config.setdefault('wuttatest.node_type', 'warehouse')
self.assertEqual(self.app.get_node_type(), 'warehouse')
self.config.setdefault("wuttatest.node_type", "warehouse")
self.assertEqual(self.app.get_node_type(), "warehouse")
def test_get_distribution(self):
@ -296,16 +320,16 @@ app_title = WuttaTest
# works with "non-native" objects
query = Query({})
dist = self.app.get_distribution(query)
self.assertEqual(dist, 'SQLAlchemy')
self.assertEqual(dist, "SQLAlchemy")
# can override dist via config
self.config.setdefault('wuttatest.app_dist', 'importlib_metadata')
self.config.setdefault("wuttatest.app_dist", "importlib_metadata")
dist = self.app.get_distribution()
self.assertEqual(dist, 'importlib_metadata')
self.assertEqual(dist, "importlib_metadata")
# but the provided object takes precedence
dist = self.app.get_distribution(query)
self.assertEqual(dist, 'SQLAlchemy')
self.assertEqual(dist, "SQLAlchemy")
def test_get_distribution_pre_python_3_10(self):
@ -318,30 +342,32 @@ app_title = WuttaTest
importlib_metadata = MagicMock()
importlib_metadata.packages_distributions = MagicMock(
return_value={
'wuttjamaican': ['WuttJamaican'],
'config': ['python-configuration'],
})
"wuttjamaican": ["WuttJamaican"],
"config": ["python-configuration"],
}
)
orig_import = __import__
def mock_import(name, *args, **kwargs):
if name == 'importlib.metadata':
if name == "importlib.metadata":
raise ImportError
if name == 'importlib_metadata':
if name == "importlib_metadata":
return importlib_metadata
return orig_import(name, *args, **kwargs)
with patch('builtins.__import__', side_effect=mock_import):
with patch("builtins.__import__", side_effect=mock_import):
# default should always be WuttJamaican (right..?)
dist = self.app.get_distribution()
self.assertEqual(dist, 'WuttJamaican')
self.assertEqual(dist, "WuttJamaican")
# also works with "non-native" objects
from config import Configuration
config = Configuration({})
dist = self.app.get_distribution(config)
self.assertEqual(dist, 'python-configuration')
self.assertEqual(dist, "python-configuration")
# hacky sort of test, just in case we can't deduce the
# package dist based on the obj - easy enough since we
@ -350,17 +376,17 @@ app_title = WuttaTest
self.assertIsNone(dist)
# can override dist via config
self.config.setdefault('wuttatest.app_dist', 'importlib_metadata')
self.config.setdefault("wuttatest.app_dist", "importlib_metadata")
dist = self.app.get_distribution()
self.assertEqual(dist, 'importlib_metadata')
self.assertEqual(dist, "importlib_metadata")
# but the provided object takes precedence
dist = self.app.get_distribution(config)
self.assertEqual(dist, 'python-configuration')
self.assertEqual(dist, "python-configuration")
# hacky test again, this time config override should win
dist = self.app.get_distribution(42)
self.assertEqual(dist, 'importlib_metadata')
self.assertEqual(dist, "importlib_metadata")
def test_get_version(self):
from importlib.metadata import version
@ -373,31 +399,31 @@ app_title = WuttaTest
# works with "non-native" objects
query = Query({})
ver = self.app.get_version(obj=query)
self.assertEqual(ver, version('SQLAlchemy'))
self.assertEqual(ver, version("SQLAlchemy"))
# random object will not yield a dist nor version
ver = self.app.get_version(obj=42)
self.assertIsNone(ver)
# can override dist via config
self.config.setdefault('wuttatest.app_dist', 'python-configuration')
self.config.setdefault("wuttatest.app_dist", "python-configuration")
ver = self.app.get_version()
self.assertEqual(ver, version('python-configuration'))
self.assertEqual(ver, version("python-configuration"))
# but the provided object takes precedence
ver = self.app.get_version(obj=query)
self.assertEqual(ver, version('SQLAlchemy'))
self.assertEqual(ver, version("SQLAlchemy"))
# can also specify the dist
ver = self.app.get_version(dist='passlib')
self.assertEqual(ver, version('passlib'))
ver = self.app.get_version(dist="passlib")
self.assertEqual(ver, version("passlib"))
def test_make_title(self):
text = self.app.make_title('foo_bar')
text = self.app.make_title("foo_bar")
self.assertEqual(text, "Foo Bar")
def test_make_full_name(self):
name = self.app.make_full_name('Fred', '', 'Flintstone', '')
name = self.app.make_full_name("Fred", "", "Flintstone", "")
self.assertEqual(name, "Fred Flintstone")
def test_make_uuid(self):
@ -414,12 +440,10 @@ app_title = WuttaTest
pass
# with progress
self.app.progress_loop(act, [1, 2, 3], ProgressBase,
message="whatever")
self.app.progress_loop(act, [1, 2, 3], ProgressBase, message="whatever")
# without progress
self.app.progress_loop(act, [1, 2, 3], None,
message="whatever")
self.app.progress_loop(act, [1, 2, 3], None, message="whatever")
def test_get_session(self):
try:
@ -433,7 +457,7 @@ app_title = WuttaTest
self.assertIsNone(self.app.get_session(user))
Session = orm.sessionmaker()
engine = sa.create_engine('sqlite://')
engine = sa.create_engine("sqlite://")
mysession = Session(bind=engine)
mysession.add(user)
session = self.app.get_session(user)
@ -453,39 +477,39 @@ app_title = WuttaTest
def test_render_currency(self):
# null
self.assertEqual(self.app.render_currency(None), '')
self.assertEqual(self.app.render_currency(None), "")
# basic decimal example
value = decimal.Decimal('42.00')
self.assertEqual(self.app.render_currency(value), '$42.00')
value = decimal.Decimal("42.00")
self.assertEqual(self.app.render_currency(value), "$42.00")
# basic float example
value = 42.00
self.assertEqual(self.app.render_currency(value), '$42.00')
self.assertEqual(self.app.render_currency(value), "$42.00")
# decimal places will be rounded
value = decimal.Decimal('42.12345')
self.assertEqual(self.app.render_currency(value), '$42.12')
value = decimal.Decimal("42.12345")
self.assertEqual(self.app.render_currency(value), "$42.12")
# but we can declare the scale
value = decimal.Decimal('42.12345')
self.assertEqual(self.app.render_currency(value, scale=4), '$42.1234')
value = decimal.Decimal("42.12345")
self.assertEqual(self.app.render_currency(value, scale=4), "$42.1234")
# negative numbers get parens
value = decimal.Decimal('-42.42')
self.assertEqual(self.app.render_currency(value), '($42.42)')
value = decimal.Decimal("-42.42")
self.assertEqual(self.app.render_currency(value), "($42.42)")
def test_render_date(self):
self.assertEqual(self.app.render_date(None), '')
self.assertEqual(self.app.render_date(None), "")
dt = datetime.date(2024, 12, 11)
self.assertEqual(self.app.render_date(dt), '2024-12-11')
self.assertEqual(self.app.render_date(dt), "2024-12-11")
def test_render_datetime(self):
self.assertEqual(self.app.render_datetime(None), '')
self.assertEqual(self.app.render_datetime(None), "")
dt = datetime.datetime(2024, 12, 11, 8, 30, tzinfo=datetime.timezone.utc)
self.assertEqual(self.app.render_datetime(dt), '2024-12-11 08:30+0000')
self.assertEqual(self.app.render_datetime(dt), "2024-12-11 08:30+0000")
def test_render_error(self):
@ -509,15 +533,15 @@ app_title = WuttaTest
self.assertEqual(self.app.render_percent(None), "")
# typical
self.assertEqual(self.app.render_percent(12.3419), '12.34 %')
self.assertEqual(self.app.render_percent(12.3419), "12.34 %")
# more decimal places
self.assertEqual(self.app.render_percent(12.3419, decimals=3), '12.342 %')
self.assertEqual(self.app.render_percent(12.3419, decimals=4), '12.3419 %')
self.assertEqual(self.app.render_percent(12.3419, decimals=3), "12.342 %")
self.assertEqual(self.app.render_percent(12.3419, decimals=4), "12.3419 %")
# negative
self.assertEqual(self.app.render_percent(-12.3419), '(12.34 %)')
self.assertEqual(self.app.render_percent(-12.3419, decimals=3), '(12.342 %)')
self.assertEqual(self.app.render_percent(-12.3419), "(12.34 %)")
self.assertEqual(self.app.render_percent(-12.3419, decimals=3), "(12.342 %)")
def test_render_quantity(self):
@ -525,11 +549,11 @@ app_title = WuttaTest
self.assertEqual(self.app.render_quantity(None), "")
# integer decimals become integers
value = decimal.Decimal('1.000')
value = decimal.Decimal("1.000")
self.assertEqual(self.app.render_quantity(value), "1")
# but decimal places are preserved
value = decimal.Decimal('1.234')
value = decimal.Decimal("1.234")
self.assertEqual(self.app.render_quantity(value), "1.234")
# zero can be empty string
@ -537,20 +561,20 @@ app_title = WuttaTest
self.assertEqual(self.app.render_quantity(0, empty_zero=True), "")
def test_render_time_ago(self):
with patch.object(mod, 'humanize') as humanize:
humanize.naturaltime.return_value = 'now'
with patch.object(mod, "humanize") as humanize:
humanize.naturaltime.return_value = "now"
now = datetime.datetime.now()
result = self.app.render_time_ago(now)
self.assertEqual(result, 'now')
self.assertEqual(result, "now")
humanize.naturaltime.assert_called_once_with(now)
def test_get_person(self):
people = self.app.get_people_handler()
with patch.object(people, 'get_person') as get_person:
get_person.return_value = 'foo'
person = self.app.get_person('bar')
get_person.assert_called_once_with('bar')
self.assertEqual(person, 'foo')
with patch.object(people, "get_person") as get_person:
get_person.return_value = "foo"
person = self.app.get_person("bar")
get_person.assert_called_once_with("bar")
self.assertEqual(person, "foo")
def test_get_auth_handler(self):
from wuttjamaican.auth import AuthHandler
@ -561,55 +585,80 @@ app_title = WuttaTest
def test_get_batch_handler(self):
# error if handler not found
self.assertRaises(KeyError, self.app.get_batch_handler, 'CannotFindMe!')
self.assertRaises(KeyError, self.app.get_batch_handler, "CannotFindMe!")
# caller can specify default
handler = self.app.get_batch_handler('foo', default='wuttjamaican.batch:BatchHandler')
handler = self.app.get_batch_handler(
"foo", default="wuttjamaican.batch:BatchHandler"
)
self.assertIsInstance(handler, BatchHandler)
# default can be configured
self.config.setdefault('wuttatest.batch.foo.handler.default_spec',
'wuttjamaican.batch:BatchHandler')
handler = self.app.get_batch_handler('foo')
self.config.setdefault(
"wuttatest.batch.foo.handler.default_spec",
"wuttjamaican.batch:BatchHandler",
)
handler = self.app.get_batch_handler("foo")
self.assertIsInstance(handler, BatchHandler)
# preference can be configured
self.config.setdefault('wuttatest.batch.foo.handler.spec',
'tests.test_app:MockBatchHandler')
handler = self.app.get_batch_handler('foo')
self.config.setdefault(
"wuttatest.batch.foo.handler.spec", "tests.test_app:MockBatchHandler"
)
handler = self.app.get_batch_handler("foo")
self.assertIsInstance(handler, MockBatchHandler)
def test_get_batch_handler_specs(self):
# empty by default
specs = self.app.get_batch_handler_specs('foo')
specs = self.app.get_batch_handler_specs("foo")
self.assertEqual(specs, [])
# caller can specify default as string
specs = self.app.get_batch_handler_specs('foo', default='wuttjamaican.batch:BatchHandler')
self.assertEqual(specs, ['wuttjamaican.batch:BatchHandler'])
specs = self.app.get_batch_handler_specs(
"foo", default="wuttjamaican.batch:BatchHandler"
)
self.assertEqual(specs, ["wuttjamaican.batch:BatchHandler"])
# caller can specify default as list
specs = self.app.get_batch_handler_specs('foo', default=['wuttjamaican.batch:BatchHandler',
'tests.test_app:MockBatchHandler'])
self.assertEqual(specs, ['wuttjamaican.batch:BatchHandler',
'tests.test_app:MockBatchHandler'])
specs = self.app.get_batch_handler_specs(
"foo",
default=[
"wuttjamaican.batch:BatchHandler",
"tests.test_app:MockBatchHandler",
],
)
self.assertEqual(
specs,
["wuttjamaican.batch:BatchHandler", "tests.test_app:MockBatchHandler"],
)
# default can be configured
self.config.setdefault('wuttatest.batch.foo.handler.default_spec',
'wuttjamaican.batch:BatchHandler')
specs = self.app.get_batch_handler_specs('foo')
self.assertEqual(specs, ['wuttjamaican.batch:BatchHandler'])
self.config.setdefault(
"wuttatest.batch.foo.handler.default_spec",
"wuttjamaican.batch:BatchHandler",
)
specs = self.app.get_batch_handler_specs("foo")
self.assertEqual(specs, ["wuttjamaican.batch:BatchHandler"])
# the rest come from entry points
with patch.object(mod, 'load_entry_points', return_value={
'mock': MockBatchHandler,
'another': AnotherBatchHandler,
}):
specs = self.app.get_batch_handler_specs('foo')
self.assertEqual(specs, ['wuttjamaican.batch:BatchHandler',
'tests.test_app:AnotherBatchHandler',
'tests.test_app:MockBatchHandler'])
with patch.object(
mod,
"load_entry_points",
return_value={
"mock": MockBatchHandler,
"another": AnotherBatchHandler,
},
):
specs = self.app.get_batch_handler_specs("foo")
self.assertEqual(
specs,
[
"wuttjamaican.batch:BatchHandler",
"tests.test_app:AnotherBatchHandler",
"tests.test_app:MockBatchHandler",
],
)
def test_get_db_handler(self):
try:
@ -653,15 +702,15 @@ app_title = WuttaTest
def test_send_email(self):
from wuttjamaican.email import EmailHandler
with patch.object(EmailHandler, 'send_email') as send_email:
self.app.send_email('foo')
send_email.assert_called_once_with('foo')
with patch.object(EmailHandler, "send_email") as send_email:
self.app.send_email("foo")
send_email.assert_called_once_with("foo")
class TestAppProvider(TestCase):
def setUp(self):
self.config = WuttaConfig(appname='wuttatest')
self.config = WuttaConfig(appname="wuttatest")
self.app = mod.AppHandler(self.config)
self.config._app = self.app
@ -671,11 +720,11 @@ class TestAppProvider(TestCase):
provider = mod.AppProvider(self.config)
self.assertIs(provider.config, self.config)
self.assertIs(provider.app, self.app)
self.assertEqual(provider.appname, 'wuttatest')
self.assertEqual(provider.appname, "wuttatest")
# but can pass app handler instead
with warnings.catch_warnings():
warnings.filterwarnings('ignore', category=DeprecationWarning)
warnings.filterwarnings("ignore", category=DeprecationWarning)
provider = mod.AppProvider(self.app)
self.assertIs(provider.config, self.config)
self.assertIs(provider.app, self.app)
@ -686,17 +735,17 @@ class TestAppProvider(TestCase):
pass
# nb. we specify *classes* here
fake_providers = {'fake': FakeProvider}
fake_providers = {"fake": FakeProvider}
with patch('wuttjamaican.app.load_entry_points') as load_entry_points:
with patch("wuttjamaican.app.load_entry_points") as load_entry_points:
load_entry_points.return_value = fake_providers
# sanity check, we get *instances* back from this
providers = self.app.get_all_providers()
load_entry_points.assert_called_once_with('wutta.app.providers')
load_entry_points.assert_called_once_with("wutta.app.providers")
self.assertEqual(len(providers), 1)
self.assertIn('fake', providers)
self.assertIsInstance(providers['fake'], FakeProvider)
self.assertIn("fake", providers)
self.assertIsInstance(providers["fake"], FakeProvider)
def test_hasattr(self):
@ -704,15 +753,15 @@ class TestAppProvider(TestCase):
def fake_foo(self):
pass
self.app.providers = {'fake': FakeProvider(self.config)}
self.app.providers = {"fake": FakeProvider(self.config)}
self.assertTrue(hasattr(self.app, 'fake_foo'))
self.assertFalse(hasattr(self.app, 'fake_method_does_not_exist'))
self.assertTrue(hasattr(self.app, "fake_foo"))
self.assertFalse(hasattr(self.app, "fake_method_does_not_exist"))
def test_getattr(self):
# enum
self.assertNotIn('enum', self.app.__dict__)
self.assertNotIn("enum", self.app.__dict__)
self.assertIs(self.app.enum, wuttjamaican.enum)
# now we test that providers are loaded...
@ -722,12 +771,12 @@ class TestAppProvider(TestCase):
return 42
# nb. using instances here
fake_providers = {'fake': FakeProvider(self.config)}
fake_providers = {"fake": FakeProvider(self.config)}
with patch.object(self.app, 'get_all_providers') as get_all_providers:
with patch.object(self.app, "get_all_providers") as get_all_providers:
get_all_providers.return_value = fake_providers
self.assertNotIn('providers', self.app.__dict__)
self.assertNotIn("providers", self.app.__dict__)
self.assertIs(self.app.providers, fake_providers)
get_all_providers.assert_called_once_with()
@ -738,27 +787,27 @@ class TestAppProvider(TestCase):
pytest.skip("test not relevant without sqlalchemy")
# model
self.assertNotIn('model', self.app.__dict__)
self.assertNotIn("model", self.app.__dict__)
self.assertIs(self.app.model, wuttjamaican.db.model)
def test_getattr_providers(self):
# collection of providers is loaded on demand
self.assertNotIn('providers', self.app.__dict__)
self.assertNotIn("providers", self.app.__dict__)
self.assertIsNotNone(self.app.providers)
# custom attr does not exist yet
self.assertRaises(AttributeError, getattr, self.app, 'foo_value')
self.assertRaises(AttributeError, getattr, self.app, "foo_value")
# but provider can supply the attr
self.app.providers['mytest'] = MagicMock(foo_value='bar')
self.assertEqual(self.app.foo_value, 'bar')
self.app.providers["mytest"] = MagicMock(foo_value="bar")
self.assertEqual(self.app.foo_value, "bar")
class TestGenericHandler(ConfigTestCase):
def make_config(self, **kw):
kw.setdefault('appname', 'wuttatest')
kw.setdefault("appname", "wuttatest")
return super().make_config(**kw)
def make_handler(self, **kwargs):
@ -768,34 +817,36 @@ class TestGenericHandler(ConfigTestCase):
handler = mod.GenericHandler(self.config)
self.assertIs(handler.config, self.config)
self.assertIs(handler.app, self.app)
self.assertEqual(handler.appname, 'wuttatest')
self.assertEqual(handler.appname, "wuttatest")
def test_get_spec(self):
self.assertEqual(mod.GenericHandler.get_spec(), 'wuttjamaican.app:GenericHandler')
self.assertEqual(
mod.GenericHandler.get_spec(), "wuttjamaican.app:GenericHandler"
)
def test_get_provider_modules(self):
# no providers, no email modules
with patch.object(self.app, 'providers', new={}):
with patch.object(self.app, "providers", new={}):
handler = self.make_handler()
self.assertEqual(handler.get_provider_modules('email'), [])
self.assertEqual(handler.get_provider_modules("email"), [])
# provider may specify modules as list
providers = {
'wuttatest': MagicMock(email_modules=['wuttjamaican.app']),
"wuttatest": MagicMock(email_modules=["wuttjamaican.app"]),
}
with patch.object(self.app, 'providers', new=providers):
with patch.object(self.app, "providers", new=providers):
handler = self.make_handler()
modules = handler.get_provider_modules('email')
modules = handler.get_provider_modules("email")
self.assertEqual(len(modules), 1)
self.assertIs(modules[0], mod)
# provider may specify modules as string
providers = {
'wuttatest': MagicMock(email_modules='wuttjamaican.app'),
"wuttatest": MagicMock(email_modules="wuttjamaican.app"),
}
with patch.object(self.app, 'providers', new=providers):
with patch.object(self.app, "providers", new=providers):
handler = self.make_handler()
modules = handler.get_provider_modules('email')
modules = handler.get_provider_modules("email")
self.assertEqual(len(modules), 1)
self.assertIs(modules[0], mod)