def test_close(self):
connection = Mock()
receiver = Mock()
receiver.close.side_effect = KeyError
node = Mock(address='test')
# test
reader = Reader(node, '')
reader.connection = connection
reader.receiver = receiver
reader.is_open = Mock(return_value=True)
reader.close()
# validation
receiver.close.assert_called_once_with()
self.assertFalse(connection.close.called)
def test_close(self):
connection = Mock()
receiver = Mock()
receiver.close.side_effect = KeyError
node = Mock(address='test')
# test
reader = Reader(node, '')
reader.connection = connection
reader.receiver = receiver
reader.is_open = Mock(return_value=True)
reader.close()
# validation
receiver.close.assert_called_once_with()
self.assertFalse(connection.close.called)
def test_repair(self, connection):
url = 'test-url'
node = Mock(address='test')
# test
reader = Reader(node, url)
reader.close = Mock()
reader.repair()
# validation
reader.close.assert_called_once_with()
reader.connection.close.assert_called_once_with()
connection.return_value.open.assert_called_once_with()
connection.return_value.receiver.assert_called_once_with(node.address)
self.assertEqual(reader.receiver, reader.connection.receiver.return_value)
def test_repair(self, connection):
url = 'test-url'
node = Mock(address='test')
# test
reader = Reader(node, url)
reader.close = Mock()
reader.repair()
# validation
reader.close.assert_called_once_with()
reader.connection.close.assert_called_once_with()
connection.return_value.open.assert_called_once_with()
connection.return_value.receiver.assert_called_once_with(node.address)
self.assertEqual(reader.receiver,
reader.connection.receiver.return_value)
