← Back to branch summary

~ubuntu-branches/ubuntu/lucid/landscape-client/lucid-updates

« back to all changes in this revision

Viewing changes to landscape/manager/tests/test_hardwareinfo.py

  • Committer: Package Import Robot
  • Author(s): Andreas Hasenack
  • Date: 2012-04-10 14:28:48 UTC
  • mfrom: (1.1.27)
  • mto: This revision was merged to the branch mainline in revision 35.
  • Revision ID: package-import@ubuntu.com-20120410142848-7xsy4g2xii7y7ntc
Import upstream version 12.04.3

Show diffs side-by-side

added added

removed removed

Lines of Context:
landscape/manager/tests/test_hardwareinfo.py
 
1
from landscape.tests.helpers import LandscapeTest, ManagerHelper
 
2
 
 
3
from landscape.manager.hardwareinfo import HardwareInfo
 
4
 
 
5
 
 
6
class HardwareInfoTests(LandscapeTest):
 
7
    helpers = [ManagerHelper]
 
8
 
 
9
    def setUp(self):
 
10
        super(HardwareInfoTests, self).setUp()
 
11
        self.info = HardwareInfo()
 
12
        self.info.command = "/bin/echo"
 
13
        self.manager.add(self.info)
 
14
 
 
15
        service = self.broker_service
 
16
        service.message_store.set_accepted_types(["hardware-info"])
 
17
 
 
18
    def test_message(self):
 
19
        """
 
20
        L{HardwareInfo} sends the output of its command when running.
 
21
        """
 
22
        deferred = self.info.send_message()
 
23
 
 
24
        def check(ignored):
 
25
            self.assertMessages(
 
26
                self.broker_service.message_store.get_pending_messages(),
 
27
                [{"data": u"-xml -quiet\n", "type": "hardware-info"}])
 
28
 
 
29
        return deferred.addCallback(check)
 
30
 
 
31
    def test_run_upgraded_system(self):
 
32
        """
 
33
        L{HardwareInfo} sends the output of its command when running on
 
34
        a system that has been upgraded to include this plugin, i.e.
 
35
        where the client already knows that it can send the
 
36
        hardware-info message.
 
37
        """
 
38
        deferred = self.info.run()
 
39
 
 
40
        def check(ignored):
 
41
            self.assertMessages(
 
42
                self.broker_service.message_store.get_pending_messages(),
 
43
                [{"data": u"-xml -quiet\n", "type": "hardware-info"}])
 
44
 
 
45
        return deferred.addCallback(check)