diff options
| author | Chris Lu <chris.lu@gmail.com> | 2018-06-25 00:01:53 -0700 |
|---|---|---|
| committer | Chris Lu <chris.lu@gmail.com> | 2018-06-25 00:01:53 -0700 |
| commit | 0d83c1b91eba8edb64b247984a2a02d3e774c9ce (patch) | |
| tree | d5840a400aae31943a24e2dac180e584162b2d0b /weed/topology/topology_test.go | |
| parent | 018a9a20be26b704abb27b19a22a4377624a9d5c (diff) | |
| download | seaweedfs-0d83c1b91eba8edb64b247984a2a02d3e774c9ce.tar.xz seaweedfs-0d83c1b91eba8edb64b247984a2a02d3e774c9ce.zip | |
refactoring
Diffstat (limited to 'weed/topology/topology_test.go')
| -rw-r--r-- | weed/topology/topology_test.go | 30 |
1 files changed, 2 insertions, 28 deletions
diff --git a/weed/topology/topology_test.go b/weed/topology/topology_test.go index 06f506c37..36aa07157 100644 --- a/weed/topology/topology_test.go +++ b/weed/topology/topology_test.go @@ -4,7 +4,6 @@ import ( "testing" "github.com/chrislusf/seaweedfs/weed/sequence" "github.com/chrislusf/seaweedfs/weed/pb/master_pb" - "github.com/chrislusf/seaweedfs/weed/storage" ) func TestRemoveDataCenter(t *testing.T) { @@ -44,20 +43,8 @@ func TestHandlingVolumeServerHeartbeat(t *testing.T) { } volumeMessages = append(volumeMessages, volumeMessage) } - var volumeInfos []storage.VolumeInfo - for _, v := range volumeMessages { - if vi, err := storage.NewVolumeInfo(v); err == nil { - volumeInfos = append(volumeInfos, vi) - } - } - deletedVolumes := dn.UpdateVolumes(volumeInfos) - for _, v := range volumeInfos { - topo.RegisterVolumeLayout(v, dn) - } - for _, v := range deletedVolumes { - topo.UnRegisterVolumeLayout(v, dn) - } + topo.SyncDataNodeRegistration(volumeMessages, dn) assert(t, "activeVolumeCount1", topo.activeVolumeCount, volumeCount) assert(t, "volumeCount", topo.volumeCount, volumeCount) @@ -81,20 +68,7 @@ func TestHandlingVolumeServerHeartbeat(t *testing.T) { } volumeMessages = append(volumeMessages, volumeMessage) } - var volumeInfos []storage.VolumeInfo - for _, v := range volumeMessages { - if vi, err := storage.NewVolumeInfo(v); err == nil { - volumeInfos = append(volumeInfos, vi) - } - } - - deletedVolumes := dn.UpdateVolumes(volumeInfos) - for _, v := range volumeInfos { - topo.RegisterVolumeLayout(v, dn) - } - for _, v := range deletedVolumes { - topo.UnRegisterVolumeLayout(v, dn) - } + topo.SyncDataNodeRegistration(volumeMessages, dn) assert(t, "activeVolumeCount1", topo.activeVolumeCount, volumeCount) assert(t, "volumeCount", topo.volumeCount, volumeCount) |