Fix unit test coverage issues for wrappertask

Copy and paste fail.

Change-Id: Ib3e5144d81fda7598e41e92f425b489d2cfc9a46

diff --git a/heat/tests/test_scheduler.py b/heat/tests/test_scheduler.py
index 684e4eec495034793d6aad4669325da19ac3ec21..206f24f78414896b1a460a8a07a7ff5d4c34c789 100644 (file)
--- a/heat/tests/test_scheduler.py
+++ b/heat/tests/test_scheduler.py
@@ -529,8 +529,6 @@ class WrapperTaskTest(mox.MoxTestBase):
 
     def test_cancel(self):
         def child_task():
-            yield
-
             try:
                 yield
             except GeneratorExit:
@@ -541,7 +539,7 @@ class WrapperTaskTest(mox.MoxTestBase):
         @scheduler.wrappertask
         def parent_task():
             try:
-                yield DummyTask()()
+                yield child_task()
             except GeneratorExit:
                 raise
             else:
@@ -553,8 +551,6 @@ class WrapperTaskTest(mox.MoxTestBase):
 
     def test_cancel_exit(self):
         def child_task():
-            yield
-
             try:
                 yield
             except GeneratorExit:
@@ -565,7 +561,7 @@ class WrapperTaskTest(mox.MoxTestBase):
         @scheduler.wrappertask
         def parent_task():
             try:
-                yield DummyTask()()
+                yield child_task()
             except GeneratorExit:
                 raise
             else:
@@ -577,8 +573,6 @@ class WrapperTaskTest(mox.MoxTestBase):
 
     def test_cancel_parent_exit(self):
         def child_task():
-            yield
-
             try:
                 yield
             except GeneratorExit:
@@ -589,7 +583,7 @@ class WrapperTaskTest(mox.MoxTestBase):
         @scheduler.wrappertask
         def parent_task():
             try:
-                yield DummyTask()()
+                yield child_task()
             except GeneratorExit:
                 return
             else: