run_isolated: close storage

client/run_isolated.py

 #!/usr/bin/env python
 # Copyright 2012 The LUCI Authors. All rights reserved.
 # Use of this source code is governed by the Apache v2.0 license that can be
 # found in the LICENSE file.
 
 """Runs a command with optional isolated input/output.
 
 Despite name "run_isolated", can run a generic non-isolated command specified as
 args.
 
@@ skipping 585 matching lines @@
         '%s in args requires --isolate-server' % ISOLATED_OUTDIR_PARAMETER)
 
   if options.root_dir:
     options.root_dir = unicode(os.path.abspath(options.root_dir))
   if options.json:
     options.json = unicode(os.path.abspath(options.json))
 
   command = [] if options.isolated else args
   if options.isolate_server:
     storage = isolateserver.get_storage(
-      options.isolate_server, options.namespace)
+        options.isolate_server, options.namespace)
     # Hashing schemes used by |storage| and |cache| MUST match.
-    assert storage.hash_algo == cache.hash_algo
-    return run_tha_test(
-        command, options.isolated, storage, cache, options.leak_temp_dir,
-        options.json, options.root_dir, options.hard_timeout,
-        options.grace_period, args)
+    with storage:
+      assert storage.hash_algo == cache.hash_algo
+      return run_tha_test(
+          command, options.isolated, storage, cache, options.leak_temp_dir,
+          options.json, options.root_dir, options.hard_timeout,
+          options.grace_period, args)
   else:
     return run_tha_test(
         command, options.isolated, None, cache, options.leak_temp_dir,
         options.json, options.root_dir, options.hard_timeout,
         options.grace_period, args)
 
 
 if __name__ == '__main__':
   # Ensure that we are always running with the correct encoding.
   fix_encoding.fix_encoding()
   sys.exit(main(sys.argv[1:]))