[tor-commits] [ooni-probe/master] Add ProcessTest template and example of it's usage

[art@xxxxxxxxxxxxxx]

commit e6994225574fb947256e529dfaeecec7a3bdfb10
Author: Arturo Filastò <art@xxxxxxxxx>
Date:   Thu Mar 26 13:35:33 2015 +0100

    Add ProcessTest template and example of it's usage
---
 data/nettests/examples/example_process.py |    9 +++
 ooni/templates/process.py                 |   85 +++++++++++++++++++++++++++++
 2 files changed, 94 insertions(+)

diff --git a/data/nettests/examples/example_process.py b/data/nettests/examples/example_process.py
new file mode 100644
index 0000000..32e52e1
--- /dev/null
+++ b/data/nettests/examples/example_process.py
@@ -0,0 +1,9 @@
+from twisted.internet import defer
+
+from ooni.templates import process
+
+
+class TestProcessExample(process.ProcessTest):
+    @defer.inlineCallbacks
+    def test_http_and_dns(self):
+        yield self.run(["echo", "Hello world!"])
diff --git a/ooni/templates/process.py b/ooni/templates/process.py
new file mode 100644
index 0000000..736dbcd
--- /dev/null
+++ b/ooni/templates/process.py
@@ -0,0 +1,85 @@
+from twisted.internet import protocol, defer, reactor
+
+from ooni.nettest import NetTestCase
+from ooni.errors import failureToString
+from ooni.utils import log
+
+
+class ProcessDirector(protocol.ProcessProtocol):
+    def __init__(self, d, finished=None):
+        self.d = d
+        self.stderr = ""
+        self.stdout = ""
+        self.finished = finished
+
+    def data(self):
+        return
+
+    def finish(self, reason=None):
+        exit_reason = failureToString(reason)
+        data = {
+            "stderr": self.stderr,
+            "stdout": self.stdout,
+            "exit_reason": exit_reason
+        }
+        self.d.callback(data)
+
+    def shouldClose(self):
+        if self.finished is None:
+            return False
+        return self.finished(self.stdout, self.stderr)
+
+    def connectionMade(self):
+        self.transport.write("")
+        self.transport.closeStdin()
+
+    def outReceived(self, data):
+        log.debug("STDOUT: %s" % data)
+        self.stdout += data
+        if self.shouldClose():
+            self.transport.loseConnection()
+
+    def errReceived(self, data):
+        log.debug("STDERR: %s" % data)
+        self.stderr += data
+
+    def inConnectionLost(self):
+        log.debug("inConnectionLost")
+        # self.d.callback(self.data())
+
+    def outConnectionLost(self):
+        log.debug("outConnectionLost")
+
+    def errConnectionLost(self):
+        log.debug("errConnectionLost")
+
+    def processExited(self, reason):
+        log.debug("Exited %s" % reason)
+
+    def processEnded(self, reason):
+        log.debug("Ended %s" % reason)
+        self.finish(reason)
+
+
+class ProcessTest(NetTestCase):
+    name = "Base Process Test"
+    version = "0.1"
+
+    requiresRoot = False
+    timeout = 5
+    address = None
+    port = None
+
+    def _setUp(self):
+        super(ProcessTest, self)._setUp()
+
+    def processEnded(self, result):
+        self.report.update(result)
+        return result
+
+    def run(self, command):
+        d = defer.Deferred()
+        d.addCallback(self.processEnded)
+        processDirector = ProcessDirector(d)
+        reactor.spawnProcess(processDirector, command[0], command)
+        return d



_______________________________________________
tor-commits mailing list
tor-commits@xxxxxxxxxxxxxxxxxxxx
https://lists.torproject.org/cgi-bin/mailman/listinfo/tor-commits