graphql-python/graphene
1
1
Fork 0
mirror of https://github.com/graphql-python/graphene.git synced 2025-07-18 03:52:24 +03:00
Code Issues Packages Projects Releases Wiki Activity

Added pragma: no cover to untested bits of DataLoader (minor stuff)

Browse Source
This commit is contained in:
Cadu 2022-09-06 15:29:09 -03:00
parent 32ab8791f7
commit 1e7bfb7cf4
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
graphene/utils/dataloader.py
View File

@ -44,7 +44,7 @@ class DataLoader(object):
), "batch_load_fn must be coroutine. Received: {}".format(self.batch_load_fn) ), "batch_load_fn must be coroutine. Received: {}".format(self.batch_load_fn)
if not callable(self.batch_load_fn): if not callable(self.batch_load_fn):
raise TypeError( raise TypeError( # pragma: no cover
( (
"DataLoader must be have a batch_load_fn which accepts " "DataLoader must be have a batch_load_fn which accepts "
"Iterable<key> and returns Future<Iterable<value>>, but got: {}." "Iterable<key> and returns Future<Iterable<value>>, but got: {}."
@ -52,13 +52,13 @@ class DataLoader(object):
) )
if batch is not None: if batch is not None:
self.batch = batch self.batch = batch # pragma: no cover
if max_batch_size is not None: if max_batch_size is not None:
self.max_batch_size = max_batch_size self.max_batch_size = max_batch_size
if cache is not None: if cache is not None:
self.cache = cache self.cache = cache # pragma: no cover
self.get_cache_key = get_cache_key or (lambda x: x) self.get_cache_key = get_cache_key or (lambda x: x)
@ -77,7 +77,7 @@ class DataLoader(object):
Loads a key, returning a `Future` for the value represented by that key. Loads a key, returning a `Future` for the value represented by that key.
""" """
if key is None: if key is None:
raise TypeError( raise TypeError( # pragma: no cover
( (
"The loader.load() function must be called with a value, " "The loader.load() function must be called with a value, "
"but got: {}." "but got: {}."
@ -113,7 +113,7 @@ class DataLoader(object):
enqueue_post_future_job(self.loop, self) enqueue_post_future_job(self.loop, self)
else: else:
# Otherwise dispatch the (queue of one) immediately. # Otherwise dispatch the (queue of one) immediately.
dispatch_queue(self) dispatch_queue(self) # pragma: no cover
def load_many(self, keys): def load_many(self, keys):
""" """
@ -129,7 +129,7 @@ class DataLoader(object):
>>> ) >>> )
""" """
if not isinstance(keys, Iterable): if not isinstance(keys, Iterable):
raise TypeError( raise TypeError( # pragma: no cover
( (
"The loader.load_many() function must be called with Iterable<key> " "The loader.load_many() function must be called with Iterable<key> "
"but got: {}." "but got: {}."
@ -223,7 +223,7 @@ async def dispatch_queue_batch(loader, queue):
# Assert the expected response from batch_load_fn # Assert the expected response from batch_load_fn
if not batch_future or not iscoroutine(batch_future): if not batch_future or not iscoroutine(batch_future):
return failed_dispatch( return failed_dispatch( # pragma: no cover
loader, loader,
queue, queue,
TypeError( TypeError(
@ -238,7 +238,7 @@ async def dispatch_queue_batch(loader, queue):
try: try:
values = await batch_future values = await batch_future
if not isinstance(values, Iterable): if not isinstance(values, Iterable):
raise TypeError( raise TypeError( # pragma: no cover
( (
"DataLoader must be constructed with a function which accepts " "DataLoader must be constructed with a function which accepts "
"Iterable<key> and returns Future<Iterable<value>>, but the function did " "Iterable<key> and returns Future<Iterable<value>>, but the function did "
@ -248,7 +248,7 @@ async def dispatch_queue_batch(loader, queue):
values = list(values) values = list(values)
if len(values) != len(keys): if len(values) != len(keys):
raise TypeError( raise TypeError( # pragma: no cover
( (
"DataLoader must be constructed with a function which accepts " "DataLoader must be constructed with a function which accepts "
"Iterable<key> and returns Future<Iterable<value>>, but the function did " "Iterable<key> and returns Future<Iterable<value>>, but the function did "