Skip to content

Commit b4c1815

Browse files
committed
Minor fixes.
1 parent 11a1755 commit b4c1815

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

umock.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -315,7 +315,7 @@ def __init__(self, target, new=None):
315315
If no new object is provided, a new Mock object is created.
316316
"""
317317
self.target = target
318-
self.new = new or Mock()
318+
self.new = new
319319

320320
def __call__(self, func, *args, **kwargs):
321321
"""
@@ -332,8 +332,8 @@ def __enter__(self, target, new):
332332
"""
333333
Replace the target with new.
334334
"""
335-
self.target = resolve_target(self.target)
336-
self.new = new
335+
self.target = resolve_target(target)
336+
self.new = new or Mock()
337337
self._old = getattr(self.target, self.new.__name__, None)
338338
setattr(self.target, self.new.__name__, self.new)
339339
return self.new
@@ -376,7 +376,7 @@ def resolve_target(target):
376376
module = __import__(module_name)
377377
while parts:
378378
# Traverse the parts of the target to find the package boundary.
379-
p = parts[0]]
379+
p = parts[0]
380380
new_module_name = f"{module_name}.{p}"
381381
try:
382382
module = __import__(new_module_name)

0 commit comments

Comments
 (0)