From cec03b4c6da87bc1c139af8c239f743961f28bd1 Mon Sep 17 00:00:00 2001
From: tsurendra <tsurendra@juniper.net>
Date: Thu, 3 Sep 2015 22:42:30 -0700
Subject: [PATCH] Closes-Bug: #1470628 Server Manager: Use puppet labs reboot
 instead of reboot command

Fix:
    Notify the reboot resourse.

Change-Id: Id02fad51aeec7e35a35760416e68e68af9ba00fc
---
 .../modules/contrail/manifests/lib/upgrade_kernel.pp          | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/contrail/environment/modules/contrail/manifests/lib/upgrade_kernel.pp b/contrail/environment/modules/contrail/manifests/lib/upgrade_kernel.pp
index d6ce9e5f..ec7c5fa0 100644
--- a/contrail/environment/modules/contrail/manifests/lib/upgrade_kernel.pp
+++ b/contrail/environment/modules/contrail/manifests/lib/upgrade_kernel.pp
@@ -32,7 +32,7 @@
               apply => "immediately",
               timeout => 0,
               message => "Rebooting for kernel upgrade",
-              subscribe       => Package[$image_extra],
+              subscribe       => [Package[$image_extra], Package[$headers], Package[$headers_generic], Package[$image]],
             }
         } else {
             if ($contrail_kernel_version == "" ) {
@@ -55,7 +55,7 @@
             package { $image : ensure => present, notify => Reboot["after"], }
             ->
             reboot { 'after':
-              subscribe       => Package[$image],
+              subscribe       => [Package[$image_extra], Package[$headers], Package[$headers_generic], Package[$image]],
               apply => "immediately",
               message => "Rebooting for kernel upgrade",
               timeout => 0,