← Back to branch summary

~lutostag/ubuntu/trusty/maas/1.5.4

« back to all changes in this revision

Viewing changes to src/maasserver/rpc/tests/test_regionservice.py

  • Committer: Package Import Robot
  • Author(s): Andres Rodriguez
  • Date: 2014-02-15 12:08:23 UTC
  • mto: This revision was merged to the branch mainline in revision 48.
  • Revision ID: package-import@ubuntu.com-20140215120823-u7dkitfy0h8tbruh
Tags: upstream-1.5+bzr1948
Import upstream version 1.5+bzr1948

Show diffs side-by-side

added added

removed removed

Lines of Context:
src/maasserver/rpc/tests/test_regionservice.py
20
20
from crochet import wait_for_reactor
21
21
from maasserver.rpc.regionservice import (
22
22
    Region,
23
 
    RegionFactory,
24
23
    RegionService,
25
24
    )
26
25
from maastesting.factory import factory
31
30
from provisioningserver.rpc.testing import call_responder
32
31
from testtools.matchers import (
33
32
    AfterPreprocessing,
 
33
    Equals,
34
34
    Is,
35
35
    IsInstance,
36
36
    MatchesListwise,
45
45
    fail,
46
46
    )
47
47
from twisted.internet.interfaces import IStreamServerEndpoint
 
48
from twisted.internet.protocol import Factory
48
49
from twisted.python import log
49
50
 
50
51
 
118
119
        service = RegionService(reactor)
119
120
        self.assertThat(service, IsInstance(Service))
120
121
        self.assertThat(service.endpoint, Provides(IStreamServerEndpoint))
121
 
        self.assertThat(service.factory, IsInstance(RegionFactory))
 
122
        self.assertThat(service.factory, IsInstance(Factory))
 
123
        self.assertThat(service.factory.protocol, Equals(Region))
122
124
 
123
125
    @wait_for_reactor
124
126
    def test_starting_and_stopping_the_service(self):
129
131
 
130
132
        def check_started(port):
131
133
            self.assertThat(port, IsInstance(tcp.Port))
132
 
            self.assertThat(port.factory, IsInstance(RegionFactory))
133
 
            # The port is saved on the service as an instance var.
134
 
            self.assertThat(service.port, Is(port))
 
134
            self.assertThat(port.factory, IsInstance(Factory))
 
135
            self.assertThat(port.factory.protocol, Equals(Region))
 
136
            # The port is saved as a private instance var.
 
137
            self.assertThat(service._port, Is(port))
135
138
            return service.stopService()
136
139
 
137
140
        service.starting.addCallback(check_started)
138
141
 
139
142
        def check_stopped(ignore, service=service):
140
 
            self.assertTrue(service.port.disconnected)
 
143
            self.assertTrue(service._port.disconnected)
141
144
 
142
145
        service.starting.addCallback(check_stopped)
143
146
 
157
160
 
158
161
        def check(port):
159
162
            self.assertThat(port, Is(None))
160
 
            self.assertThat(service.port, Is(None))
 
163
            self.assertThat(service._port, Is(None))
161
164
            return service.stopService()
162
165
 
163
166
        return service.starting.addCallback(check)
195
198
        def check(port):
196
199
            # The CancelledError is suppressed.
197
200
            self.assertThat(port, Is(None))
198
 
            self.assertThat(service.port, Is(None))
 
201
            self.assertThat(service._port, Is(None))
199
202
 
200
203
        return service.starting.addCallback(check)
201
204