Grcov report - test

1

//! Test helpers.

2

3

#[cfg(test)]

4

use super::{CongestionWindow, rtt::RoundtripTimeEstimator};

5

6

// Make a new RTT estimator.

7

#[cfg(test)]

8

10

pub(crate) fn new_rtt_estimator() -> RoundtripTimeEstimator {

9

10

    RoundtripTimeEstimator::new(&params::build_rtt_params())

10

11

12

// Make a new congestion window.

13

#[cfg(test)]

14

28

pub(crate) fn new_cwnd() -> CongestionWindow {

15

28

    CongestionWindow::new(params::build_cwnd_params())

16

28

17

18

#[cfg(test)]

19

pub(crate) mod params {

20

    use tor_units::Percentage;

21

22

    use crate::ccparams::{

23

        Algorithm, CongestionControlParams, CongestionControlParamsBuilder, CongestionWindowParams,

24

        CongestionWindowParamsBuilder, FixedWindowParams, FixedWindowParamsBuilder,

25

        RoundTripEstimatorParams, RoundTripEstimatorParamsBuilder, VegasParamsBuilder,

26

};

27

28

388

    fn build_fixed_params() -> FixedWindowParams {

29

388

        FixedWindowParamsBuilder::default()

30

388

            .circ_window_start(1000)

31

388

            .circ_window_min(100)

32

388

            .circ_window_max(1000)

33

388

            .build()

34

388

            .expect("Unable to build fixed window params")

35

388

36

37

66

    pub(crate) fn build_cc_vegas_params() -> CongestionControlParams {

38

        // Following values are the default from the proposal. They likely differ from what the

39

        // consensus uses today.

40

66

        let params = VegasParamsBuilder::default()

41

66

            .cell_in_queue_params((186, 248, 310, 186, 600).into())

42

66

            .ss_cwnd_max(5000)

43

66

            .cwnd_full_gap(4)

44

66

            .cwnd_full_min_pct(Percentage::new(25))

45

66

            .cwnd_full_per_cwnd(1)

46

66

            .build()

47

66

            .expect("Unable to build Vegas params");

48

66

        CongestionControlParamsBuilder::default()

49

66

            .rtt_params(build_rtt_params())

50

66

            .cwnd_params(build_cwnd_params())

51

66

            .alg(Algorithm::Vegas(params))

52

66

            .fixed_window_params(build_fixed_params())

53

66

            .build()

54

66

            .expect("Unable to build CC params")

55

66

56

57

322

    pub(crate) fn build_cc_fixed_params() -> CongestionControlParams {

58

322

        let params = build_fixed_params();

59

322

        CongestionControlParamsBuilder::default()

60

322

            .rtt_params(build_rtt_params())

61

322

            .cwnd_params(build_cwnd_params())

62

322

            .alg(Algorithm::FixedWindow(params))

63

322

            .fixed_window_params(params)

64

322

            .build()

65

322

            .expect("Unable to build CC params")

66

322

67

68

    // Build the round trip estimator parameters. with good enough values for tests.

69

398

    pub(crate) fn build_rtt_params() -> RoundTripEstimatorParams {

70

398

        RoundTripEstimatorParamsBuilder::default()

71

398

            .ewma_cwnd_pct(Percentage::new(50))

72

398

            .ewma_max(10)

73

398

            .ewma_ss_max(2)

74

398

            .rtt_reset_pct(Percentage::new(100))

75

398

            .build()

76

398

            .expect("Unable to build RTT parameters")

77

398

78

79

    // Build the congestion window parameters. with good enough values for tests.

80

416

    pub(crate) fn build_cwnd_params() -> CongestionWindowParams {

81

        // Values taken from the prop324.

82

416

        CongestionWindowParamsBuilder::default()

83

416

            .cwnd_init(124)

84

416

            .cwnd_inc_pct_ss(Percentage::new(100))

85

416

            .cwnd_inc(1)

86

416

            .cwnd_inc_rate(31)

87

416

            .cwnd_min(124)

88

416

            .cwnd_max(u32::MAX)

89

416

            .sendme_inc(31)

90

416

            .build()

91

416

            .expect("Unable to build congestion window parameters")

92

416

93