[nft] tests: py: add ct timeout tests

Message ID 20180708113620.20098-1-harshasharmaiitr@gmail.com
State Changes Requested
Delegated to: Pablo Neira
Headers show
Series
  • [nft] tests: py: add ct timeout tests
Related show

Commit Message

Harsha Sharma July 8, 2018, 11:36 a.m.
Add test for adding ct timeout objects and assigning it via rule.

Signed-off-by: Harsha Sharma <harshasharmaiitr@gmail.com>
---
 tests/py/ip/objects.t         | 6 ++++++
 tests/py/ip/objects.t.payload | 4 ++++
 tests/py/nft-test.py          | 4 ++++
 3 files changed, 14 insertions(+)

Patch

diff --git a/tests/py/ip/objects.t b/tests/py/ip/objects.t
index 5e8c763..82638f2 100644
--- a/tests/py/ip/objects.t
+++ b/tests/py/ip/objects.t
@@ -32,3 +32,9 @@  ct helper set tcp dport map {21 : "cthelp1", 2121 : "cthelp1" };ok
 ip saddr 192.168.1.3 limit name "lim1";ok
 ip saddr 192.168.1.3 limit name "lim3";fail
 limit name tcp dport map {443 : "lim1", 80 : "lim2", 22 : "lim1"};ok
+
+# ct timeout
+%cttime1 type ct timeout { protocol:tcp, established:122 ;};ok
+%cttime2 type ct timeout { protocol:udp, syn_sent:122 ;};fail
+
+ct timeout set "cttime1";ok
diff --git a/tests/py/ip/objects.t.payload b/tests/py/ip/objects.t.payload
index 193d797..4ab5b73 100644
--- a/tests/py/ip/objects.t.payload
+++ b/tests/py/ip/objects.t.payload
@@ -59,3 +59,7 @@  ip test-ip4 output
   [ cmp eq reg 1 0x00000006 ]
   [ payload load 2b @ transport header + 2 => reg 1 ]
   [ objref sreg 1 set __objmap%d ]
+
+# ct timeout set "cttime1"
+ip test-ip4 output
+  [ objref type 6 name cttime1 ]
diff --git a/tests/py/nft-test.py b/tests/py/nft-test.py
index e4367ad..5e9e2e3 100755
--- a/tests/py/nft-test.py
+++ b/tests/py/nft-test.py
@@ -1082,6 +1082,10 @@  def obj_process(obj_line, filename, lineno):
        obj_type = "ct helper"
        tokens[3] = ""
 
+    if obj_type == "ct" and tokens[3] == "timeout":
+       obj_type = "ct timeout"
+       tokens[3] = ""
+
     if len(tokens) > 3:
         obj_spcf = " ".join(tokens[3:])