~xianghui/ubuntu/trusty/oslo.messaging/icehouse-lp1521958

« back to all changes in this revision

Viewing changes to tests/test_rabbit.py

  • Committer: Package Import Robot
  • Author(s): Chuck Short
  • Date: 2014-02-07 14:32:19 UTC
  • mfrom: (1.1.2)
  • Revision ID: package-import@ubuntu.com-20140207143219-faeitvpdrypb1c9x
Tags: 1.3.0~a7-0ubuntu1
* New upstream release.
* debian/control:
  - Add python-yaml, python-babel, python-six, python-mox3,
    python-mock as build dependency.
  - Dropped python-d2to1 as a dependency.
* wrap and sort.

Show diffs side-by-side

added added

removed removed

Lines of Context:
49
49
 
50
50
    scenarios = [
51
51
        ('none', dict(url=None, expected=None)),
52
 
        ('empty', dict(url='rabbit:///', expected=None)),
 
52
        ('empty',
 
53
         dict(url='rabbit:///',
 
54
              expected=dict(virtual_host=''))),
53
55
        ('localhost',
54
56
         dict(url='rabbit://localhost/',
55
57
              expected=dict(hostname='localhost',
56
58
                            username='',
57
59
                            password='',
58
60
                            virtual_host=''))),
 
61
        ('virtual_host',
 
62
         dict(url='rabbit:///vhost',
 
63
              expected=dict(virtual_host='vhost'))),
59
64
        ('no_creds',
60
65
         dict(url='rabbit://host/virtual_host',
61
66
              expected=dict(hostname='host',
79
84
 
80
85
    def setUp(self):
81
86
        super(TestRabbitTransportURL, self).setUp()
 
87
 
82
88
        self.messaging_conf.transport_driver = 'rabbit'
83
89
        self.messaging_conf.in_memory = True
84
90
 
85
 
    def test_transport_url(self):
 
91
        self._server_params = []
86
92
        cnx_init = rabbit_driver.Connection.__init__
87
 
        passed_params = []
88
 
 
89
 
        def record_params(self, conf, server_params=None):
90
 
            passed_params.append(server_params)
91
 
            return cnx_init(self, conf, server_params)
 
93
 
 
94
        def record_params(cnx, conf, server_params=None):
 
95
            self._server_params.append(server_params)
 
96
            return cnx_init(cnx, conf, server_params)
 
97
 
 
98
        def dummy_send(cnx, topic, msg, timeout=None):
 
99
            pass
92
100
 
93
101
        self.stubs.Set(rabbit_driver.Connection, '__init__', record_params)
94
 
 
95
 
        transport = messaging.get_transport(self.conf, self.url)
96
 
 
97
 
        driver = transport._driver
98
 
 
99
 
        target = messaging.Target(topic='testtopic')
100
 
 
101
 
        driver.listen(target)
102
 
 
103
 
        self.assertEquals(passed_params[0], self.expected)
 
102
        self.stubs.Set(rabbit_driver.Connection, 'topic_send', dummy_send)
 
103
 
 
104
        self._driver = messaging.get_transport(self.conf, self.url)._driver
 
105
        self._target = messaging.Target(topic='testtopic')
 
106
 
 
107
    def test_transport_url_listen(self):
 
108
        self._driver.listen(self._target)
 
109
        self.assertEqual(self._server_params[0], self.expected)
 
110
 
 
111
    def test_transport_url_send(self):
 
112
        self._driver.send(self._target, {}, {})
 
113
        self.assertEqual(self._server_params[0], self.expected)
104
114
 
105
115
 
106
116
class TestSendReceive(test_utils.BaseTestCase):
197
207
            msgs.append(received)
198
208
 
199
209
        # reply in reverse, except reply to the first guy second from last
200
 
        order = range(len(senders)-1, -1, -1)
 
210
        order = list(range(len(senders)-1, -1, -1))
201
211
        if len(order) > 1:
202
212
            order[-1], order[-2] = order[-2], order[-1]
203
213