Function test_parallel_after_finalize

ciphercore-base/src/graphs.rs:5787–5834

()

    #[tokio::test(flavor = "multi_thread", worker_threads = 50)]
    async fn test_parallel_after_finalize() -> Result<()> {
        let context = create_context()?;
        let graph = context.create_graph()?;
        let input1 = graph.input(scalar_type(INT32))?;
        let input2 = graph.input(scalar_type(INT32))?;
        let output = graph.add(input1.clone(), input2.clone())?;
        graph.set_output_node(output.clone())?;
        graph.finalize()?;
        context.set_main_graph(graph.clone())?;

        context.set_node_name(input1.clone(), "one")?;
        context.set_node_name(input2.clone(), "two")?;

        context.finalize()?;

        assert!(output.clone().get_type().is_ok());

        const PAR_ITERS: usize = 2001;

        let mut get_type_futures = vec![];
        for _ in 0..PAR_ITERS {
            get_type_futures.push(parallel_get_type(output.clone()));
        }
        futures::future::try_join_all(get_type_futures).await?;

        let mut get_random_evaluate_futures = vec![];
        for _ in 0..PAR_ITERS {
            get_random_evaluate_futures
                .push(parallel_random_evaluate(graph.clone(), context.clone()))
        }
        futures::future::try_join_all(get_random_evaluate_futures).await?;

        let input_parties = vec![IOStatus::Party(0), IOStatus::Party(1)];
        let output_parties = vec![IOStatus::Party(0)];

        let mut get_mpc_eval_futures = vec![];
        for _ in 0..PAR_ITERS {
            get_mpc_eval_futures.push(parallel_prepare_for_mpc_evaluation(
                context.clone(),
                vec![input_parties.clone()],
                vec![output_parties.clone()],
                InlineConfig::default(),
            ));
        }
        futures::future::try_join_all(get_mpc_eval_futures).await?;

        Ok(())
    }
}