mirror of
https://git.anonymousland.org/anonymousland/synapse.git
synced 2024-09-08 05:51:57 +00:00
remove unused DeferredMockCallable
This commit is contained in:
parent
c834f1d67a
commit
b751624ff8
@ -20,13 +20,12 @@ import os
|
|||||||
import time
|
import time
|
||||||
import uuid
|
import uuid
|
||||||
import warnings
|
import warnings
|
||||||
from inspect import getcallargs
|
|
||||||
from typing import Type
|
from typing import Type
|
||||||
from urllib import parse as urlparse
|
from urllib import parse as urlparse
|
||||||
|
|
||||||
from mock import Mock, patch
|
from mock import Mock, patch
|
||||||
|
|
||||||
from twisted.internet import defer, reactor
|
from twisted.internet import defer
|
||||||
|
|
||||||
from synapse.api.constants import EventTypes
|
from synapse.api.constants import EventTypes
|
||||||
from synapse.api.errors import CodeMessageException, cs_error
|
from synapse.api.errors import CodeMessageException, cs_error
|
||||||
@ -343,14 +342,6 @@ def setup_test_homeserver(
|
|||||||
return hs
|
return hs
|
||||||
|
|
||||||
|
|
||||||
def get_mock_call_args(pattern_func, mock_func):
|
|
||||||
""" Return the arguments the mock function was called with interpreted
|
|
||||||
by the pattern functions argument list.
|
|
||||||
"""
|
|
||||||
invoked_args, invoked_kargs = mock_func.call_args
|
|
||||||
return getcallargs(pattern_func, *invoked_args, **invoked_kargs)
|
|
||||||
|
|
||||||
|
|
||||||
def mock_getRawHeaders(headers=None):
|
def mock_getRawHeaders(headers=None):
|
||||||
headers = headers if headers is not None else {}
|
headers = headers if headers is not None else {}
|
||||||
|
|
||||||
@ -536,86 +527,6 @@ class MockClock:
|
|||||||
return d
|
return d
|
||||||
|
|
||||||
|
|
||||||
def _format_call(args, kwargs):
|
|
||||||
return ", ".join(
|
|
||||||
["%r" % (a) for a in args] + ["%s=%r" % (k, v) for k, v in kwargs.items()]
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
class DeferredMockCallable:
|
|
||||||
"""A callable instance that stores a set of pending call expectations and
|
|
||||||
return values for them. It allows a unit test to assert that the given set
|
|
||||||
of function calls are eventually made, by awaiting on them to be called.
|
|
||||||
"""
|
|
||||||
|
|
||||||
def __init__(self):
|
|
||||||
self.expectations = []
|
|
||||||
self.calls = []
|
|
||||||
|
|
||||||
def __call__(self, *args, **kwargs):
|
|
||||||
self.calls.append((args, kwargs))
|
|
||||||
|
|
||||||
if not self.expectations:
|
|
||||||
raise ValueError(
|
|
||||||
"%r has no pending calls to handle call(%s)"
|
|
||||||
% (self, _format_call(args, kwargs))
|
|
||||||
)
|
|
||||||
|
|
||||||
for (call, result, d) in self.expectations:
|
|
||||||
if args == call[1] and kwargs == call[2]:
|
|
||||||
d.callback(None)
|
|
||||||
return result
|
|
||||||
|
|
||||||
failure = AssertionError(
|
|
||||||
"Was not expecting call(%s)" % (_format_call(args, kwargs))
|
|
||||||
)
|
|
||||||
|
|
||||||
for _, _, d in self.expectations:
|
|
||||||
try:
|
|
||||||
d.errback(failure)
|
|
||||||
except Exception:
|
|
||||||
pass
|
|
||||||
|
|
||||||
raise failure
|
|
||||||
|
|
||||||
def expect_call_and_return(self, call, result):
|
|
||||||
self.expectations.append((call, result, defer.Deferred()))
|
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
|
||||||
def await_calls(self, timeout=1000):
|
|
||||||
deferred = defer.DeferredList(
|
|
||||||
[d for _, _, d in self.expectations], fireOnOneErrback=True
|
|
||||||
)
|
|
||||||
|
|
||||||
timer = reactor.callLater(
|
|
||||||
timeout / 1000,
|
|
||||||
deferred.errback,
|
|
||||||
AssertionError(
|
|
||||||
"%d pending calls left: %s"
|
|
||||||
% (
|
|
||||||
len([e for e in self.expectations if not e[2].called]),
|
|
||||||
[e for e in self.expectations if not e[2].called],
|
|
||||||
)
|
|
||||||
),
|
|
||||||
)
|
|
||||||
|
|
||||||
yield deferred
|
|
||||||
|
|
||||||
timer.cancel()
|
|
||||||
|
|
||||||
self.calls = []
|
|
||||||
|
|
||||||
def assert_had_no_calls(self):
|
|
||||||
if self.calls:
|
|
||||||
calls = self.calls
|
|
||||||
self.calls = []
|
|
||||||
|
|
||||||
raise AssertionError(
|
|
||||||
"Expected not to received any calls, got:\n"
|
|
||||||
+ "\n".join(["call(%s)" % _format_call(c[0], c[1]) for c in calls])
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
async def create_room(hs, room_id: str, creator_id: str):
|
async def create_room(hs, room_id: str, creator_id: str):
|
||||||
"""Creates and persist a creation event for the given room
|
"""Creates and persist a creation event for the given room
|
||||||
"""
|
"""
|
||||||
|
Loading…
Reference in New Issue
Block a user