def test_repair(self, connection, close):
url = 'test-url'
node = Mock(address='test')
# test
reader = Reader(node, url)
reader.repair()
# validation
close.assert_called_once_with()
connection.return_value.close.assert_called_once_with()
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_repair(self, connection, close):
url = 'test-url'
node = Mock(address='test')
# test
reader = Reader(node, url)
reader.repair()
# validation
close.assert_called_once_with()
connection.return_value.close.assert_called_once_with()
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)
