Sl 396

tempesta_fw/t/functional/run_all_tests.sh

@@ -1,6 +1,6 @@
 #!/bin/bash
 #
-# 2014. Written by NatSys Lab. ([email protected]).
+# 2014-2016. Tempesta Technologies Inc. ([email protected]).
 
 function run() {
 	echo run: $1
@@ -18,5 +18,4 @@ echo ------------------------------------------------------------------
 echo Running functional tests...
 echo ------------------------------------------------------------------
 
-# Doesn't pass yet.
-# run fragmented_requests.py
+run tests.py 

tempesta_fw/t/functional/tests.py

@@ -0,0 +1,22 @@
+#!/usr/bin/env python
+
+__author__ = 'Tempesta Technologies Inc.'
+__copyright__ = 'Copyright (C) 2016 Tempesta Technologies Inc. ([email protected]).'
+__license__ = 'GPL2'
+
+import pkgutil
+import tests
+import subprocess
+import sys
+from os.path import dirname, realpath, sep
+
+sys.path.append((dirname(realpath(__file__))+ sep + "tests" + sep + "helpers"))
+
+for loader, name, ispkg in pkgutil.iter_modules(path = tests.__path__,
+						 prefix = ''):
+	if not ispkg:
+		print(name)
+		test = loader.find_module(name).load_module(name)
+		tclass = getattr(test, 'Test')
+		print("test:", tclass().get_name())
+		tclass().run()

tempesta_fw/t/functional/tests/__init__.py

@@ -0,0 +1,6 @@
+__author__ = 'Tempesta Technologies Inc.'
+__copyright__ = 'Copyright (C) 2016 Tempesta Technologies Inc. ([email protected]).'
+__license__ = 'GPL2'
+
+__all__ = [ "bomber", "fragmented_requests", "test_frang", "test_cache",
+	    "test_parser" ]

tempesta_fw/t/functional/tests/bomber.py

@@ -0,0 +1,25 @@
+#!/usr/bin/env python
+__author__ = 'Tempesta Technologies Inc.'
+__copyright__ = 'Copyright (C) 20115-2016 Tempesta Technologies Inc. ([email protected]).'
+__license__ = 'GPL2'
+
+import conf
+import tfw
+
+c = conf.Config("etc/tempesta_fw.conf")
+
+class Test:
+	def get_name(self):
+		return 'bomber'
+
+	def run(self):
+		c.add_option('cache', '0')
+		c.add_option('listen', '8081')
+		c.add_option('server', '127.0.0.1:80')
+		tfw.start()
+		print("tfw started\n")
+		tfw.start_bomber()
+		print("bomber started\n")
+		tfw.stop()
+		print("tfw stoped\n")
+

tempesta_fw/t/functional/tests/fragmented_requests.py

@@ -0,0 +1,39 @@
+#!/usr/bin/env python
+
+__author__ = 'Temesta Technologies Inc.'
+__copyright__ = 'Copyright (C) 2016 Tempesta Technologies Inc. ([email protected]).'
+__license__ = 'GPL2'
+
+import conf
+import tfw
+import socket
+class Test:
+	def fragmentize_str(self, s, frag_size):
+		"""
+		Split a string into a list of equal N-sized fragmen.
+		>>> fragmentize_str("foo12bar34baz", 3)
+		['foo', '12b', 'ar3', '4ba', 'z']
+		"""
+		return [s[i:i+frag_size]  for i in range(0, len(s), frag_size)]
+
+	def run(self):
+		c = conf.Config('etc/tempesta_fw.conf')
+		c.add_option('cache', '0')
+		c.add_option('listen', '8081')
+		c.add_option('server', '127.0.0.1:80')
+		vs_get = b"GET / HTTP/1.0\r\nhost: localhost\r\n\r\n"
+		tfw.start()
+		print("tfw start\n")
+		s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+		s.connect(("127.0.0.1",8081))
+		for fs in self.fragmentize_str(vs_get, 3):
+			s.sendall(fs)
+		data = s.recv(1024)
+		tfw.stop()
+		s.close()
+		if len(data) > 0:
+			print("Res:{}\n".format(True))
+
+	def get_name(self):
+		return 'fragmented request'
+

tempesta_fw/t/functional/tests/helpers/__init__.py

@@ -0,0 +1,5 @@
+__author__ = 'Tempesta Technologies Inc.'
+__copyright__ = 'Copyright (C) 2016 Tempesta Technologies. ([email protected]).'
+__license__ = 'GPL2'
+
+__all__ = [ 'be', 'cli', 'tfw','conf', 'apache', 'tfwparser' ]

tempesta_fw/t/functional/tests/helpers/apache.py

@@ -0,0 +1,10 @@
+#!/usr/bin/env python3
+
+__author__ = 'Tempesta Technologies Inc.'
+__copyright__ = 'Copyright (C) 2016 Tempesta Technologies. ([email protected]).'
+__license__ = 'GPL2'
+
+import subprocess
+
+def start():
+	subprocess.call("service apache2 start", shell = True)