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
|
// Copyright (C) MongoDB, Inc. 2017-present.
//
// Licensed under the Apache License, Version 2.0 (the "License"); you may
// not use this file except in compliance with the License. You may obtain
// a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
package topology
import (
"testing"
"github.com/stretchr/testify/assert"
"go.mongodb.org/mongo-driver/mongo/description"
)
func TestDiffHostList(t *testing.T) {
h1 := "1.0.0.0:27017"
h2 := "2.0.0.0:27017"
h3 := "3.0.0.0:27017"
h4 := "4.0.0.0:27017"
h5 := "5.0.0.0:27017"
h6 := "6.0.0.0:27017"
s1 := description.Server{Addr: "1.0.0.0:27017"}
s2 := description.Server{Addr: "2.0.0.0:27017"}
s3 := description.Server{Addr: "3.0.0.0:27017"}
s6 := description.Server{Addr: "6.0.0.0:27017"}
topo := description.Topology{
Servers: []description.Server{s6, s1, s3, s2},
}
hostlist := []string{h2, h4, h3, h5}
diff := diffHostList(topo, hostlist)
assert.ElementsMatch(t, []string{h4, h5}, diff.Added)
assert.ElementsMatch(t, []string{h1, h6}, diff.Removed)
// Ensure that original topology servers and hostlist were not reordered.
assert.EqualValues(t, []description.Server{s6, s1, s3, s2}, topo.Servers)
assert.EqualValues(t, []string{h2, h4, h3, h5}, hostlist)
}
func TestDiffTopology(t *testing.T) {
s1 := description.Server{Addr: "1.0.0.0:27017"}
s2 := description.Server{Addr: "2.0.0.0:27017"}
s3 := description.Server{Addr: "3.0.0.0:27017"}
s4 := description.Server{Addr: "4.0.0.0:27017"}
s5 := description.Server{Addr: "5.0.0.0:27017"}
s6 := description.Server{Addr: "6.0.0.0:27017"}
t1 := description.Topology{
Servers: []description.Server{s6, s1, s3, s2},
}
t2 := description.Topology{
Servers: []description.Server{s2, s4, s3, s5},
}
diff := diffTopology(t1, t2)
assert.ElementsMatch(t, []description.Server{s4, s5}, diff.Added)
assert.ElementsMatch(t, []description.Server{s1, s6}, diff.Removed)
// Ensure that original topology servers were not reordered.
assert.EqualValues(t, []description.Server{s6, s1, s3, s2}, t1.Servers)
assert.EqualValues(t, []description.Server{s2, s4, s3, s5}, t2.Servers)
}
|
