FilesExpand file tree

 master

/

util.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

26 lines (19 loc) · 711 Bytes

 master

/

util.py

Top

File metadata and controls

• Code
• Blame

26 lines (19 loc) · 711 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

# Copyright (c) Microsoft Corporation. All rights reserved.

# 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)