From 5518ad3b80afd4990e0c5462c96256d9155b02b7 Mon Sep 17 00:00:00 2001
From: Brian Elliott <bdelliott@gmail.com>
Date: Thu, 16 May 2013 13:00:24 +0000
Subject: Add graceful service shutdown support to Launcher

Provides support for letting each running service finish in-progress
requests and cleanup gracefully before stopping.

blueprint graceful-shutdown

Change-Id: Idf137620b87417de89eba23f05daf08780dc9dff
---
 tests/unit/test_service.py | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)

(limited to 'tests')

diff --git a/tests/unit/test_service.py b/tests/unit/test_service.py
index 0f93830..20007de 100644
--- a/tests/unit/test_service.py
+++ b/tests/unit/test_service.py
@@ -31,6 +31,7 @@ import socket
 import time
 import traceback
 
+from eventlet import event
 from oslo.config import cfg
 
 from openstack.common import eventlet_backdoor
@@ -195,6 +196,20 @@ class ServiceLauncherTest(utils.BaseTestCase):
         self.assertEqual(os.WEXITSTATUS(status), 0)
 
 
+class _Service(service.Service):
+    def __init__(self):
+        super(_Service, self).__init__()
+        self.init = event.Event()
+        self.cleaned_up = False
+
+    def start(self):
+        self.init.send()
+
+    def stop(self):
+        self.cleaned_up = True
+        super(_Service, self).stop()
+
+
 class LauncherTest(utils.BaseTestCase):
 
     def test_backdoor_port(self):
@@ -252,3 +267,15 @@ class LauncherTest(utils.BaseTestCase):
         svc = service.Service()
         self.assertRaises(eventlet_backdoor.EventletBackdoorConfigValueError,
                           service.launch, svc)
+
+    def test_graceful_shutdown(self):
+        # test that services are given a chance to clean up:
+        svc = _Service()
+
+        launcher = service.launch(svc)
+        # wait on 'init' so we know the service had time to start:
+        svc.init.wait()
+
+        launcher.stop()
+        self.assertTrue(svc.cleaned_up)
+        self.assertTrue(svc._done.ready())
-- 
cgit