vscode-python/pythonFiles/tests/util.py at master · SamB/vscode-python · GitHub

27 lines (19 loc) · 712 Bytes

# Licensed under the MIT License.

class Stub(object):

def __init__(self):

self.calls = []

def add_call(self, name, args=None, kwargs=None):

self.calls.append((name, args, kwargs))

class StubProxy(object):

def __init__(self, stub=None, name=None):

self.name = name

self.stub = stub if stub is not None else Stub()

@property

def calls(self):

return self.stub.calls

def add_call(self, funcname, *args, **kwargs):

callname = funcname

if self.name:

callname = '{}.{}'.format(self.name, funcname)

return self.stub.add_call(callname, *args, **kwargs)

Expand file tree