def test_broadcast_futures_doesnt_set_deadline_header_for_l5(
self, mock_gen_request, mock_get_address, mock_create_task,
mock_dto):
fake_session = MagicMock()
fake_session.post = MagicMock(return_value="session_request")
broadcast_processor.make_broadcast_futures(fake_session, "block_id", 5,
{"chain_id"})
fake_session.post.assert_called_once_with(
url="addr/v1/enqueue",
data=b"some data",
headers={"header": "thing"},
timeout=broadcast_processor.HTTP_REQUEST_TIMEOUT)
def test_broadcast_futures_doesnt_return_future_for_exception_with_a_chain(
self, mock_gen_req, mock_get_address, patch_get_broadcast):
fake_session = MagicMock()
fake_session.post = MagicMock(return_value="session_request")
self.assertEqual(
broadcast_processor.make_broadcast_futures(fake_session,
"block_id", 2,
{"chain_id"}), set())
def test_broadcast_futures_returns_set_of_futures_from_session_posts(
self, mock_gen_request, mock_get_address, mock_create_task,
patch_get_broadcast):
fake_session = MagicMock()
fake_session.post = MagicMock(return_value="session_request")
self.assertEqual(
broadcast_processor.make_broadcast_futures(fake_session,
"block_id", 2,
{"chain_id"}), {"task"})
mock_get_address.assert_called_once_with("chain_id")
mock_create_task.assert_called_once_with("session_request")
mock_gen_request.assert_called_once_with("POST", "chain_id",
"/v1/enqueue", "dto")
fake_session.post.assert_called_once_with(
url="addr/v1/enqueue",
data=b"some data",
headers={
"header": "thing",
"deadline":
str(broadcast_processor.BROADCAST_RECEIPT_WAIT_TIME)
},
timeout=broadcast_processor.HTTP_REQUEST_TIMEOUT,
)
def test_broadcast_futures_returns_none_on_get_broadcast_dto_failure(
self, mock_increment_error, patch_get_broadcast):
self.assertIsNone(
broadcast_processor.make_broadcast_futures(None, "block_id", 2,
{"chain_id"}))
mock_increment_error.assert_called_once_with("block_id", 2)
def test_broadcast_futures_gets_broadcast_dto_for_block_id(
self, patch_get_broadcast):
broadcast_processor.make_broadcast_futures(None, "id", 3, set())
patch_get_broadcast.assert_called_once_with(3, "id")
