| // Copyright 2015 The Vanadium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style |
| // license that can be found in the LICENSE file. |
| |
| package vsync |
| |
| import ( |
| "fmt" |
| "reflect" |
| "testing" |
| "time" |
| |
| "v.io/v23/discovery" |
| wire "v.io/v23/services/syncbase" |
| _ "v.io/x/ref/runtime/factories/generic" |
| ) |
| |
| func TestPeerDiscovery(t *testing.T) { |
| // Set a large value to prevent the syncer and the peer manager from |
| // running because this test adds fake neighborhood information. |
| peerSyncInterval = 1 * time.Hour |
| peerManagementInterval = 1 * time.Hour |
| |
| svc := createService(t) |
| defer destroyService(t, svc) |
| s := svc.sync |
| |
| checkPeers := func(peers map[string]uint32, want map[string]*discovery.Advertisement) { |
| got := s.filterDiscoveryPeers(peers) |
| if !reflect.DeepEqual(got, want) { |
| t.Errorf("filterDiscoveryPeers: wrong data: got %v, want %v", got, want) |
| } |
| } |
| |
| peers := map[string]uint32{"a": 0, "b": 1, "c": 2} |
| |
| checkPeers(peers, nil) |
| |
| // Add peer neighbors. |
| svcA := &discovery.Advertisement{ |
| Attributes: discovery.Attributes{discoveryAttrPeer: "a"}, |
| Addresses: []string{"aa", "aaa"}, |
| } |
| svcB := &discovery.Advertisement{ |
| Attributes: discovery.Attributes{discoveryAttrPeer: "b"}, |
| Addresses: []string{"bb", "bbb"}, |
| } |
| |
| s.updateDiscoveryInfo("a", svcA) |
| s.updateDiscoveryInfo("b", svcB) |
| |
| checkPeers(peers, map[string]*discovery.Advertisement{"a": svcA, "b": svcB}) |
| checkPeers(map[string]uint32{"x": 0, "y": 1, "z": 2}, map[string]*discovery.Advertisement{}) |
| |
| // Remove a neighbor. |
| s.updateDiscoveryInfo("a", nil) |
| |
| checkPeers(peers, map[string]*discovery.Advertisement{"b": svcB}) |
| |
| // Remove the other neighbor. |
| s.updateDiscoveryInfo("b", nil) |
| |
| checkPeers(peers, map[string]*discovery.Advertisement{}) |
| } |
| |
| func TestSyncgroupDiscovery(t *testing.T) { |
| // Set a large value to prevent the syncer and the peer manager from |
| // running because this test adds fake neighborhood information. |
| peerSyncInterval = 1 * time.Hour |
| peerManagementInterval = 1 * time.Hour |
| |
| svc := createService(t) |
| defer destroyService(t, svc) |
| s := svc.sync |
| |
| checkSyncgroupAdmins := func(sgName string, want []*discovery.Advertisement) { |
| dbId := wire.Id{Name: fmt.Sprintf("d%s", sgName), Blessing: "dblessing"} |
| sgId := wire.Id{Name: sgName, Blessing: "blessing"} |
| got := s.filterSyncgroupAdmins(dbId, sgId) |
| |
| g := make(map[*discovery.Advertisement]bool) |
| for _, e := range got { |
| g[e] = true |
| } |
| w := make(map[*discovery.Advertisement]bool) |
| for _, e := range want { |
| w[e] = true |
| } |
| if !reflect.DeepEqual(g, w) { |
| t.Errorf("checkSyncgroupAdmins: wrong data: got %v, want %v", got, want) |
| } |
| } |
| |
| checkSyncgroupAdmins("foo", nil) |
| |
| // Add syncgroup admin neighbors. |
| svcA := &discovery.Advertisement{ |
| Attributes: discovery.Attributes{ |
| discoveryAttrDatabaseName: "dfoo", |
| discoveryAttrDatabaseBlessing: "dblessing", |
| discoveryAttrSyncgroupName: "foo", |
| discoveryAttrSyncgroupBlessing: "blessing"}, |
| Addresses: []string{"aa", "aaa"}, |
| } |
| svcB := &discovery.Advertisement{ |
| Attributes: discovery.Attributes{ |
| discoveryAttrDatabaseName: "dfoo", |
| discoveryAttrDatabaseBlessing: "dblessing", |
| discoveryAttrSyncgroupName: "foo", |
| discoveryAttrSyncgroupBlessing: "blessing"}, |
| Addresses: []string{"bb", "bbb"}, |
| } |
| svcC := &discovery.Advertisement{ |
| Attributes: discovery.Attributes{ |
| discoveryAttrDatabaseName: "dbar", |
| discoveryAttrDatabaseBlessing: "dblessing", |
| discoveryAttrSyncgroupName: "bar", |
| discoveryAttrSyncgroupBlessing: "blessing"}, |
| Addresses: []string{"cc", "ccc"}, |
| } |
| |
| s.updateDiscoveryInfo("foo_a", svcA) |
| s.updateDiscoveryInfo("foo_b", svcB) |
| s.updateDiscoveryInfo("bar_c", svcC) |
| |
| checkSyncgroupAdmins("foo", []*discovery.Advertisement{svcA, svcB}) |
| checkSyncgroupAdmins("bar", []*discovery.Advertisement{svcC}) |
| checkSyncgroupAdmins("haha", nil) |
| |
| // Remove an admin from the "foo" syncgroup. |
| s.updateDiscoveryInfo("foo_a", nil) |
| |
| checkSyncgroupAdmins("foo", []*discovery.Advertisement{svcB}) |
| checkSyncgroupAdmins("bar", []*discovery.Advertisement{svcC}) |
| |
| // Remove the other "foo" admin. |
| s.updateDiscoveryInfo("foo_b", nil) |
| |
| checkSyncgroupAdmins("foo", nil) |
| checkSyncgroupAdmins("bar", []*discovery.Advertisement{svcC}) |
| |
| // Remove the other "bar" admin. |
| s.updateDiscoveryInfo("bar_c", nil) |
| |
| checkSyncgroupAdmins("bar", nil) |
| } |