def test_open_already(self, receiver): url = 'test-url' queue = Mock(name='test-queue') # test reader = Reader(queue, url) reader.is_open = Mock(return_value=True) reader.open() # validation self.assertFalse(reader.connection.open.called) self.assertFalse(receiver.called)
def test_open(self, receiver, connection): url = 'test-url' queue = Queue('test-queue') receiver.return_value.open.return_value = receiver.return_value # test reader = Reader(queue, url) reader.is_open = Mock(return_value=False) reader.open() # validation connection.return_value.open.assert_called_once_with() connection.return_value.channel.assert_called_once_with() receiver.assert_called_once_with(reader) self.assertEqual(reader.channel, connection.return_value.channel.return_value) self.assertEqual(reader.receiver, receiver.return_value)