1 files changed, 95 insertions, 0 deletions
diff --git a/test/test_respawn.py b/test/test_respawn.py
new file mode 100644
index 00000000..e7eef004
--- /dev/null
+++ b/test/test_respawn.py
@@ -0,0 +1,95 @@
+import re
+import time
+import subprocess
+import unittest
+from unit.applications.lang.python import TestApplicationPython
+
+
+class TestRespawn(TestApplicationPython):
+    prerequisites = {'modules': {'python': 'any'}}
+
+    PATTERN_ROUTER = 'unit: router'
+    PATTERN_CONTROLLER = 'unit: controller'
+
+    def setUp(self):
+        super().setUp()
+
+        self.app_name = "app-" + self.testdir.split('/')[-1]
+
+        self.load('empty', self.app_name)
+
+        self.assertIn(
+            'success',
+            self.conf('1', 'applications/' + self.app_name + '/processes')
+        )
+
+    def pid_by_name(self, name):
+        output = subprocess.check_output(['ps', 'ax']).decode()
+        m = re.search('\s*(\d+).*' + name, output)
+        return m if m is None else m.group(1)
+
+    def kill_pids(self, *pids):
+        subprocess.call(['kill', '-9'] + list(pids))
+
+    def wait_for_process(self, process):
+        for i in range(50):
+            found = self.pid_by_name(process)
+
+            if found is not None:
+                break
+
+            time.sleep(0.1)
+
+        return found
+
+    def smoke_test(self):
+        for _ in range(5):
+            self.assertIn(
+                'success',
+                self.conf('1', 'applications/' + self.app_name + '/processes')
+            )
+            self.assertEqual(self.get()['status'], 200)
+
+        # Check if the only one router, controller,
+        # and application processes running.
+
+        output = subprocess.check_output(['ps', 'ax']).decode()
+        self.assertEqual(len(re.findall(self.PATTERN_ROUTER, output)), 1)
+        self.assertEqual(len(re.findall(self.PATTERN_CONTROLLER, output)), 1)
+        self.assertEqual(len(re.findall(self.app_name, output)), 1)
+
+    def test_respawn_router(self):
+        pid = self.pid_by_name(self.PATTERN_ROUTER)
+
+        self.kill_pids(pid)
+        self.skip_alerts.append(r'process %s exited on signal 9' % pid)
+
+        self.assertIsNotNone(self.wait_for_process(self.PATTERN_ROUTER))
+
+        self.smoke_test()
+
+    def test_respawn_controller(self):
+        pid = self.pid_by_name(self.PATTERN_CONTROLLER)
+
+        self.kill_pids(pid)
+        self.skip_alerts.append(r'process %s exited on signal 9' % pid)
+
+        self.assertIsNotNone(self.wait_for_process(self.PATTERN_CONTROLLER))
+
+        self.assertEqual(self.get()['status'], 200)
+
+        self.smoke_test()
+
+    def test_respawn_application(self):
+        pid = self.pid_by_name(self.app_name)
+
+        self.kill_pids(pid)
+        self.skip_alerts.append(r'process %s exited on signal 9' % pid)
+
+        self.assertIsNotNone(self.wait_for_process(self.app_name))
+
+        self.smoke_test()
+
+
+if __name__ == '__main__':
+    TestRespawn.main()