| Index: tools/telemetry/telemetry/unittest_util/system_stub.py
|
| diff --git a/tools/telemetry/telemetry/unittest_util/system_stub.py b/tools/telemetry/telemetry/unittest_util/system_stub.py
|
| index 31f71f81d67d8218110062a3d3c54bbb78118757..634431e3c545b21823c18246f62f39601d52069f 100644
|
| --- a/tools/telemetry/telemetry/unittest_util/system_stub.py
|
| +++ b/tools/telemetry/telemetry/unittest_util/system_stub.py
|
| @@ -49,7 +49,13 @@ class Override(object):
|
|
|
| def Restore(self):
|
| for module_name, original_module in self._overrides.iteritems():
|
| - setattr(self._base_module, module_name, original_module)
|
| + if original_module is None:
|
| + # This will happen when we override built-in functions, like open.
|
| + # If we don't delete the attribute, we will shadow the built-in
|
| + # function with an attribute set to None.
|
| + delattr(self._base_module, module_name)
|
| + else:
|
| + setattr(self._base_module, module_name, original_module)
|
| self._overrides = {}
|
|
|
|
|
|
|