diff options
Diffstat (limited to 'binary_search_tool/test/switch_tmp.py')
-rwxr-xr-x | binary_search_tool/test/switch_tmp.py | 34 |
1 files changed, 34 insertions, 0 deletions
diff --git a/binary_search_tool/test/switch_tmp.py b/binary_search_tool/test/switch_tmp.py new file mode 100755 index 00000000..165004ed --- /dev/null +++ b/binary_search_tool/test/switch_tmp.py @@ -0,0 +1,34 @@ +#!/usr/bin/python2 +"""Change portions of the object files to good. + +This file is a test switch script. Used only for the test test_tmp_cleanup. +The "portion" is defined by the file (which is passed as the only argument to +this script) content. Every line in the file is an object index, which will be +set to good (mark as 42). +""" + +from __future__ import print_function + +import sys + +import common + + +def Main(argv): + working_set = common.ReadWorkingSet() + object_index = common.ReadObjectIndex(argv[1]) + + # Random number so the results can be checked + for oi in object_index: + working_set[int(oi)] = 42 + + common.WriteWorkingSet(working_set) + with open('tmp_file', 'w') as f: + f.write(argv[1]) + + return 0 + + +if __name__ == '__main__': + retval = Main(sys.argv) + sys.exit(retval) |