Index: testing_support/auto_stub.py |
diff --git a/testing_support/auto_stub.py b/testing_support/auto_stub.py |
index e4fa647877dd047c648ec72dc3ebf4f45f52057f..f2ebd4b406725b821e82dd281612efd7f7b87ce6 100644 |
--- a/testing_support/auto_stub.py |
+++ b/testing_support/auto_stub.py |
@@ -30,21 +30,24 @@ class AutoStubMixIn(object): |
setattr(obj, member, previous_value) |
+def format_call(caller_name, *args, **kwargs): |
+ """Returns expectation suitable for passing to `SimpleMock.check_calls`.""" |
+ str_args = ', '.join(repr(arg) for arg in args) |
+ str_kwargs = ', '.join('%s=%r' % (k, v) |
+ for k, v in sorted(kwargs.items())) |
Sergiy Byelozyorov
2016/07/28 13:01:36
iteritems?
tandrii(chromium)
2016/07/28 13:04:06
Done.
|
+ return '%s(%s)' % ( |
+ caller_name, ', '.join(x for x in [str_args, str_kwargs] if x)) |
+ |
class SimpleMock(object): |
"""Really simple manual class mock.""" |
- def __init__(self, unit_test, sorted_kwargs=False): |
+ def __init__(self, unit_test): |
"""Do not call __init__ if you want to use the global call list to detect |
ordering across different instances. |
Args: |
unit_test (unittest.TestCase): instance of a test class. |
- sorted_kwargs (bool): if True, kwargs in expectations will always be |
- sorted by key. |
""" |
- # TODO(tandrii): sorted_kwargs MUST BE REALLY TRUE by default, but i don't |
- # want to fix all the tests, so keeping backwards compatability. |
self.calls = [] |
- self.sorted_kwargs = sorted_kwargs |
self.unit_test = unit_test |
self.assertEqual = unit_test.assertEqual |
@@ -63,14 +66,7 @@ class SimpleMock(object): |
def _register_call(self, *args, **kwargs): |
"""Registers the name of the caller function.""" |
caller_name = kwargs.pop('caller_name', None) or inspect.stack()[1][3] |
- str_args = ', '.join(repr(arg) for arg in args) |
- |
- kwargs_items = kwargs.items() |
- if self.sorted_kwargs: |
- kwargs_items.sort() |
- str_kwargs = ', '.join('%s=%r' % (k, v) for k, v in kwargs_items) |
- self.calls.append('%s(%s)' % ( |
- caller_name, ', '.join(filter(None, [str_args, str_kwargs])))) |
+ self.calls.append(format_call(caller_name, *args, **kwargs)) |
class TestCase(thread_watcher.TestCase, AutoStubMixIn): |