Merge "Add DynamicLoopCall timers to ThreadGroups"

1 files changed, 50 insertions, 0 deletions

diff --git a/tests/unit/test_threadgroup.py b/tests/unit/test_threadgroup.py
new file mode 100644
index 0000000..f627215
--- /dev/null
+++ b/tests/unit/test_threadgroup.py
@@ -0,0 +1,50 @@
+# vim: tabstop=4 shiftwidth=4 softtabstop=4
+
+# Copyright (c) 2012 Rackspace Hosting
+# All Rights Reserved.
+#
+#    Licensed under the Apache License, Version 2.0 (the "License"); you may
+#    not use this file except in compliance with the License. You may obtain
+#    a copy of the License at
+#
+#         http://www.apache.org/licenses/LICENSE-2.0
+#
+#    Unless required by applicable law or agreed to in writing, software
+#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+#    License for the specific language governing permissions and limitations
+#    under the License.
+
+"""
+Unit Tests for thread groups
+"""
+
+from openstack.common import log as logging
+from openstack.common import threadgroup
+from tests import utils
+
+LOG = logging.getLogger(__name__)
+
+
+class ThreadGroupTestCase(utils.BaseTestCase):
+    """Test cases for thread group"""
+    def setUp(self):
+        super(ThreadGroupTestCase, self).setUp()
+        self.tg = threadgroup.ThreadGroup()
+        self.addCleanup(self.tg.stop)
+
+    def test_add_dynamic_timer(self):
+
+        def foo(*args, **kwargs):
+            pass
+        initial_delay = 1
+        periodic_interval_max = 2
+        self.tg.add_dynamic_timer(foo, initial_delay, periodic_interval_max,
+                                  'arg', kwarg='kwarg')
+
+        self.assertEqual(1, len(self.tg.timers))
+
+        timer = self.tg.timers[0]
+        self.assertTrue(timer._running)
+        self.assertEqual(('arg',), timer.args)
+        self.assertEqual({'kwarg': 'kwarg'}, timer.kw)