Chromium Code Reviews| Index: testing/legion/client_rpc_methods.py |
| diff --git a/testing/legion/client_rpc_methods.py b/testing/legion/client_rpc_methods.py |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..2f99269e4a56087ee6195cc776e554f1aa858164 |
| --- /dev/null |
| +++ b/testing/legion/client_rpc_methods.py |
| @@ -0,0 +1,49 @@ |
| +# Copyright 2015 The Chromium Authors. All rights reserved. |
| +# Use of this source code is governed by a BSD-style license that can be |
| +# found in the LICENSE file. |
| +"""Client RPC Methods. |
|
Marc-Antoine Ruel (Google)
2015/01/30 21:58:39
This is not informative. "client_rpc_methods.py" p
Mike Meade
2015/02/03 01:18:08
Done.
|
| + |
| +The RPC methods made available on the client are all defined here in the |
| +RPCMethods class. This class is made separate for easy maintenance and |
| +method lookup. |
| +""" |
| + |
| +import logging |
| +import subprocess |
| +import threading |
| + |
| + |
| +class RPCMethods(object): |
| + """The class exposing RPC methods.""" |
| + |
| + def __init__(self, server): |
| + self.server = server |
| + |
| + def Echo(self, message): |
| + """Simple RPC method to print and return a message.""" |
| + logging.info('Echoing %s', message) |
| + return 'echo ' + message |
| + |
| + def Subprocess(self, cmd): |
| + """Run the commands in a subprocess. |
| + |
| + Note that stdout and stderr are passed to this process' stdout and stderr |
| + and only the return code is returned. |
| + |
| + Args: |
| + cmd: A command list to pass to subprocess.call. |
| + """ |
| + p = subprocess.Popen(cmd, stdout=subprocess.PIPE, |
| + stderr=subprocess.PIPE) |
| + p.wait() |
|
Marc-Antoine Ruel (Google)
2015/01/30 21:58:39
This will hang. You need to use .communicate()
Mike Meade
2015/02/03 01:18:08
Done.
|
| + return (p.returncode, p.stdout.read(), p.stderr.read()) |
| + |
| + def Quit(self): |
| + """Call server.shutdown in another thread. |
| + |
| + This is needed because server.shutdown waits for the server to actually |
| + quit. However the server cannot shutdown until it completes handling this |
| + call. Calling this in the same thread results in a deadlock. |
| + """ |
| + t = threading.Thread(target=self.server.shutdown) |
| + t.start() |