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 = {} |