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