[PATCH] Introducing TCPConnection test module

[Jan Tluka]

This is a python wrapper for tcp_conn test tool.

Example of use:

    <run module="TCPConnection" host="2"
bg_id="server">
        <options>
            <option name="mode" value="server"/>
            <option name="address" value="{ip(2,testiface)}"/>
            <option name="portrange" value="10000-10050"/>
            <option name="continuous" value="yes"/>
        </options>
    </run>
    <ctl_wait seconds="3"/>
    <run module="TCPConnection" host="1"
bg_id="client">
        <options>
            <option name="mode" value="client"/>
            <option name="address" value="{ip(2,testiface)}"/>
            <option name="portrange" value="10000-10050"/>
            <option name="continuous" value="yes"/>
        </options>
    </run>

Signed-off-by: Jan Tluka <jtluka(a)redhat.com&gt;
---
 test_modules/TCPConnection.py | 60 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 60 insertions(+)
 create mode 100644 test_modules/TCPConnection.py

diff --git a/test_modules/TCPConnection.py b/test_modules/TCPConnection.py
new file mode 100644
index 0000000..43b88a9
--- /dev/null
+++ b/test_modules/TCPConnection.py
@@ -0,0 +1,60 @@
+"""
+This module defines TCPConnection test,
+a python wrapper for LNST's tcp_conn test tool
+"""
+
+__author__ = """
+jtluka(a)redhat.com (Jan Tluka)
+"""
+
+import re
+import errno
+import logging
+from lnst.Common.TestsCommon import TestGeneric
+from lnst.Common.Utils import bool_it
+
+class TCPConnection(TestGeneric):
+    def run(self):
+        mode = self.get_mopt("mode")
+        address = self.get_mopt("address")
+        portrange = self.get_mopt("portrange")
+
+        continuous = self.get_opt("continuous")
+        debug = self.get_opt("debug")
+        ipv6 = self.get_opt("ipv6")
+
+        cmd = ""
+        if (mode == "server"):
+            logging.debug("TCPConnection: running as server")
+            cmd += "./tcp_listen"
+        elif (mode == "client"):
+            logging.debug("TCPConnection: running as client")
+            cmd += "./tcp_connect"
+        else:
+            raise Exception("Invalid mode value for TCPConnection test
module!")
+
+        cmd += " -a %s -p %s" % (address, portrange)
+
+        if continuous and bool_it(continuous):
+            cmd += " -c"
+
+        if debug and bool_it(debug):
+            cmd += " -d"
+
+        if ipv6 and bool_it(ipv6):
+            cmd += " -6"
+
+        output = self.exec_from("tcp_conn", cmd, die_on_err=False,
log_outputs=True)[0]
+
+        logging.debug("TCPConnection done, inspecting logs ...")
+
+        m = None
+        if (mode == "client"):
+            m = re.search("made [0-9]* connections", output)
+        elif (mode == "server"):
+            m = re.search("handled [0-9]* connections", output)
+
+        if m is None:
+            return self.set_fail({'msg': "Unexpected error"})
+        else:
+            return self.set_pass({'msg': m.group(0)})
-- 
1.8.1.4