Index: net/data/verify_certificate_chain_unittest/rebase-errors.py |
diff --git a/net/data/verify_certificate_chain_unittest/rebase-errors.py b/net/data/verify_certificate_chain_unittest/rebase-errors.py |
index 8640e18fc3bdc93901f8a9523e3298a8ee9c0b54..a99d7db365c93920404005e8143255e084d12ff5 100755 |
--- a/net/data/verify_certificate_chain_unittest/rebase-errors.py |
+++ b/net/data/verify_certificate_chain_unittest/rebase-errors.py |
@@ -107,12 +107,12 @@ def fixup_py_file(path, actual_errors): |
contents = read_file_to_string(path) |
# This assumes that the errors variable uses triple quotes. |
- prog = re.compile(r'^errors = """(.*?)"""', re.MULTILINE | re.DOTALL) |
+ prog = re.compile(r'^errors = (""".*?"""|None)', re.MULTILINE | re.DOTALL) |
result = prog.search(contents) |
# Replace the stuff in between the triple quotes with the actual errors. |
contents = replace_string(contents, result.start(1), result.end(1), |
- actual_errors) |
+ '"""' + actual_errors + '"""') |
# Update the file. |
write_string_to_file(contents, path) |