← 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_fakepackagemanager.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_fakepackagemanager.py
 
1
from landscape.manager.plugin import SUCCEEDED
 
2
 
 
3
from landscape.manager.fakepackagemanager import FakePackageManager
 
4
from landscape.tests.helpers import LandscapeTest, ManagerHelper
 
5
 
 
6
 
 
7
class FakePackageManagerTest(LandscapeTest):
 
8
    """Tests for the fake package manager plugin."""
 
9
 
 
10
    helpers = [ManagerHelper]
 
11
 
 
12
    def setUp(self):
 
13
        super(FakePackageManagerTest, self).setUp()
 
14
        self.package_manager = FakePackageManager()
 
15
        self.package_manager.randint = lambda x, y: 0
 
16
 
 
17
    def test_handle_change_packages(self):
 
18
        """
 
19
        L{FakePackageManager} is able to handle a C{change-packages} message,
 
20
        creating a C{change-packages-result} in response.
 
21
        """
 
22
        self.manager.add(self.package_manager)
 
23
        service = self.broker_service
 
24
        service.message_store.set_accepted_types(["change-packages-result"])
 
25
        message = {"type": "change-packages", "operation-id": 1}
 
26
        self.manager.dispatch_message(message)
 
27
        self.manager.reactor.advance(1)
 
28
 
 
29
        self.assertMessages(service.message_store.get_pending_messages(),
 
30
                            [{"type": "change-packages-result",
 
31
                              "result-text": "OK done.",
 
32
                              "result-code": 1, "operation-id": 1}])
 
33
 
 
34
    def test_handle_change_package_locks(self):
 
35
        """
 
36
        L{FakePackageManager} is able to handle a C{change-package-locks}
 
37
        message, creating a C{operation-result} in response.
 
38
        """
 
39
        self.manager.add(self.package_manager)
 
40
        service = self.broker_service
 
41
        service.message_store.set_accepted_types(["operation-result"])
 
42
        message = {"type": "change-package-locks", "operation-id": 1}
 
43
        self.manager.dispatch_message(message)
 
44
        self.manager.reactor.advance(1)
 
45
 
 
46
        self.assertMessages(service.message_store.get_pending_messages(),
 
47
                            [{"type": "operation-result",
 
48
                              "result-text":
 
49
                                  "Package locks successfully changed.",
 
50
                              "result-code": 0, "status": SUCCEEDED,
 
51
                              "operation-id": 1}])
 
52
 
 
53
    def test_handle_release_upgrade(self):
 
54
        """
 
55
        L{FakePackageManager} is able to handle a C{release-upgrade} message,
 
56
        creating a C{operation-result} in response.
 
57
        """
 
58
        self.manager.add(self.package_manager)
 
59
        service = self.broker_service
 
60
        service.message_store.set_accepted_types(["operation-result"])
 
61
        message = {"type": "release-upgrade", "operation-id": 1}
 
62
        self.manager.dispatch_message(message)
 
63
        self.manager.reactor.advance(1)
 
64
 
 
65
        self.assertMessages(service.message_store.get_pending_messages(),
 
66
                            [{"type": "operation-result",
 
67
                              "result-text":
 
68
                                  "Successful release upgrade.",
 
69
                              "result-code": 0, "status": SUCCEEDED,
 
70
                              "operation-id": 1}])