algorithms-and-data-structures/AlgorithmsAndDataStructures.Tests/DataStructures/Concurrency/SimpleTokenBucketTests.cs at master · arst/algorithms-and-data-structures · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
using System;
using System.Threading;
using AlgorithmsAndDataStructures.DataStructures.Concurrency;
using Xunit;

namespace AlgorithmsAndDataStructures.Tests.DataStructures.Concurrency;

public class SimpleTokenBucketTests
{
    [Fact]
    public void CanSendCorrectPacket()
    {
        using var sut = new SimpleTokenBucket(101, 100, 2000);
        var producer = new Thread(Send);
        var result = false;
        producer.Start();
        producer.Join();

        void Send()
        {
            result = sut.TrySend(100);
        }

        Assert.True(result);
    }

    [Fact]
    public void PacketBiggerThenBucketIsDropped()
    {
        using var sut = new SimpleTokenBucket(101, 100, 2000);
        var producer = new Thread(Send);
        var result = true;
        producer.Start();
        producer.Join();


        void Send()
        {
            result = sut.TrySend(102);
        }

        Assert.False(result);
    }

    [Fact]
    public void DropPacketsWhenThereAreNotEnoughTokens()
    {
        using var sut = new SimpleTokenBucket(100, 100, 20000);
        var producer = new Thread(Send);
        var result1 = false;
        var result2 = true;
        producer.Start();
        producer.Join();


        void Send()
        {
            result1 = sut.TrySend(100);
            result2 = sut.TrySend(100);
        }

        Assert.True(result1);
        Assert.False(result2);
    }

    [Fact]
    public void DropPacketsWhenQueueIsFull()
    {
        using var sut = new SimpleTokenBucket(200, 100, 20000);
        var producer = new Thread(Send);
        var result1 = false;
        var result2 = false;
        var result3 = true;
        producer.Start();
        producer.Join();

        void Send()
        {
            result1 = sut.TrySend(100);
            result2 = sut.TrySend(100);
            result3 = sut.TrySend(100);
        }

        Assert.True(result1);
        Assert.True(result2);
        Assert.False(result3);
    }

    [Fact]
    public void LeaksWithConstantRate()
    {
        using var sut = new SimpleTokenBucket(200, 100, 20000);
        var producer = new Thread(Send);
        var result1 = false;
        var result2 = false;
        var result3 = true;
        var result4 = false;
        producer.Start();
        producer.Join();

        void Send()
        {
            result1 = sut.TrySend(100);
            result2 = sut.TrySend(100);
            result3 = sut.TrySend(100);
            Thread.Sleep(TimeSpan.FromMilliseconds(30000));
            result4 = sut.TrySend(100);
        }

        Assert.True(result1);
        Assert.True(result2);
        Assert.False(result3);
        Assert.True(result4);
    }
}