forked from fluentcassandra/fluentcassandra
/
RoundRobinServerManagerTests.cs
82 lines (68 loc) · 2.37 KB
/
RoundRobinServerManagerTests.cs
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
using System;
using System.Collections.Generic;
using Xunit;
namespace FluentCassandra.Connections.Tests
{
public class RoundRobinServerManagerTests
{
[Fact]
public void CanBlackListAndCleanQueueTest()
{
RoundRobinServerManager target = new RoundRobinServerManager(new ConnectionBuilder("Server=unit-test-1,unit-test-2,unit-test-3"));
Server srv = new Server("unit-test-4");
target.Add(srv);
bool gotServer4 = false;
for (int i = 0; i < 4; i++)
{
Server server = target.Next();
if (server.ToString().Equals(srv.ToString(), StringComparison.OrdinalIgnoreCase))
{
gotServer4 = true;
break;
}
}
Assert.True(gotServer4);
target.BlackList(srv);
gotServer4 = false;
for (int i = 0; i < 4; i++)
{
Server server = target.Next();
if (server.Equals(srv))
{
gotServer4 = true;
break;
}
}
Assert.False(gotServer4);
}
[Fact]
public void HasNextWithMoreThanHalfBlacklistedTest()
{
RoundRobinServerManager target = new RoundRobinServerManager(new ConnectionBuilder("Server=unit-test-1"));
Server srv1 = null;
Server srv2 = new Server("unit-test-2");
Server srv3 = new Server("unit-test-3");
Server srv4 = new Server("unit-test-4");
target.Add(srv2);
target.Add(srv3);
target.Add(srv4);
List<Server> servers = new List<Server> { new Server("unit-test-1"), srv2, srv3, srv4 };
for (int i = 0; i < 4; i++)
{
Server srv = target.Next();
Assert.True(servers[i].ToString().Equals(srv.ToString(), StringComparison.OrdinalIgnoreCase));
if(i == 0)
{
srv1 = srv;
}
}
target.BlackList(srv2);
target.BlackList(srv3);
Assert.True(target.HasNext);
target.BlackList(srv1);
Assert.True(target.HasNext);
target.BlackList(srv4);
Assert.False(target.HasNext);
}
}
}