NPO-Accounting/conservancy_beancount
6
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

test_rtutil: Implement our own nullcontext.

Browse source 
contextlib.nullcontext does not exist in Python 3.6.
This commit is contained in:
Brett Smith 2020-05-13 10:12:08 -04:00
parent 031b660cc6
commit 30e386f645
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/test_rtutil.py
View file

@ -55,11 +55,15 @@ def new_client():
TICKET_DATA = testutil.RTClient.TICKET_DATA.copy() TICKET_DATA = testutil.RTClient.TICKET_DATA.copy()
return RTClient() return RTClient()
@contextlib.contextmanager
def nullcontext(thing):
yield thing
def new_cache(database=':memory:'): def new_cache(database=':memory:'):
db = rtutil.RTLinkCache.setup(database) db = rtutil.RTLinkCache.setup(database)
if db is None: if db is None:
print("NOTE: did not set up database cache at {}".format(database)) print("NOTE: did not set up database cache at {}".format(database))
return contextlib.nullcontext(db) return nullcontext(db)
else: else:
return contextlib.closing(db) return contextlib.closing(db)