← Back to branch summary

~free.ekanayaka/landscape-client/lucid-1.5.0-0ubuntu0.10.04.0

« back to all changes in this revision

Viewing changes to landscape/monitor/tests/test_rebootrequired.py

  • Committer: Bazaar Package Importer
  • Author(s): Free Ekanayaka
  • Date: 2009-12-16 10:50:05 UTC
  • mto: This revision was merged to the branch mainline in revision 25.
  • Revision ID: james.westby@ubuntu.com-20091216105005-9lgmwl5zklvvp6e2
Tags: upstream-1.4.0
Import upstream version 1.4.0

Show diffs side-by-side

added added

removed removed

Lines of Context:
landscape/monitor/tests/test_rebootrequired.py
2
2
 
3
3
from landscape.monitor.rebootrequired import RebootRequired
4
4
from landscape.tests.helpers import LandscapeIsolatedTest
5
 
from landscape.tests.helpers import (
6
 
    MakePathHelper, MonitorHelper, LogKeeperHelper)
 
5
from landscape.tests.helpers import MonitorHelper, LogKeeperHelper
 
6
from landscape.tests.mocker import ANY
7
7
 
8
8
 
9
9
class RebootRequiredTest(LandscapeIsolatedTest):
10
10
 
11
 
    helpers = [MakePathHelper, MonitorHelper, LogKeeperHelper]
 
11
    helpers = [MonitorHelper, LogKeeperHelper]
12
12
 
13
13
    def setUp(self):
14
14
        super(RebootRequiredTest, self).setUp()
15
 
        self.reboot_required_filename = self.make_path("")
 
15
        self.reboot_required_filename = self.makeFile("")
16
16
        self.plugin = RebootRequired(self.reboot_required_filename)
17
17
        self.monitor.add(self.plugin)
18
18
        self.mstore.set_accepted_types(["reboot-required"])
53
53
        If the server can accept them, the plugin should send
54
54
        C{reboot-required} messages.
55
55
        """
56
 
        mock_plugin = self.mocker.patch(self.plugin)
57
 
        mock_plugin.send_message()
58
 
        self.mocker.count(1)
 
56
        broker_mock = self.mocker.replace(self.remote)
 
57
        broker_mock.send_message(ANY, urgent=True)
59
58
        self.mocker.replay()
60
59
        self.plugin.run()
61
60
        self.mstore.set_accepted_types([])