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")