mirror of
https://github.com/cubefs/cubefs.git
synced 2026-08-02 10:06:14 +00:00
101 lines
2.7 KiB
Go
101 lines
2.7 KiB
Go
package master
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/cubefs/cubefs/proto"
|
|
"github.com/cubefs/cubefs/util"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestDataNode(t *testing.T) {
|
|
// /dataNode/add and /dataNode/response processed by mock data server
|
|
var err error
|
|
addr := "127.0.0.1:9096"
|
|
func() {
|
|
mockServerLock.Lock()
|
|
defer mockServerLock.Unlock()
|
|
mockDataServers = append(mockDataServers, addDataServer(addr, "test-add-zone", defaultMediaType))
|
|
}()
|
|
server.cluster.checkDataNodeHeartbeat()
|
|
time.Sleep(5 * time.Second)
|
|
getDataNodeInfo(addr, t)
|
|
updateDisks(addr, t)
|
|
decommissionDataNode(addr, t)
|
|
for i := 0; i < 10; i++ { // decommission is async process
|
|
_, err = server.cluster.dataNode(addr)
|
|
if err == nil {
|
|
time.Sleep(time.Second)
|
|
continue
|
|
}
|
|
break
|
|
}
|
|
if err != nil {
|
|
t.Errorf("decommission datanode [%v] failed", addr)
|
|
}
|
|
server.cluster.dataNodes.Delete(addr)
|
|
}
|
|
|
|
func getDataNodeInfo(addr string, t *testing.T) {
|
|
reqURL := fmt.Sprintf("%v%v?addr=%v", hostAddr, proto.GetDataNode, addr)
|
|
process(reqURL, t)
|
|
}
|
|
|
|
func decommissionDataNode(addr string, t *testing.T) {
|
|
reqURL := fmt.Sprintf("%v%v?addr=%v", hostAddr, proto.DecommissionDataNode, addr)
|
|
process(reqURL, t)
|
|
}
|
|
|
|
func updateDisks(addr string, t *testing.T) {
|
|
dn, err := server.cluster.dataNode(addr)
|
|
require.NoError(t, err)
|
|
|
|
dn.AllDisks = []string{"/data1"}
|
|
allDisk := []string{"/data1", "/data2", "/data3"}
|
|
badDisk := []string{"/data1"}
|
|
updated, _ := dn.updateDisks(allDisk, badDisk)
|
|
require.Equal(t, updated, true)
|
|
require.Equal(t, allDisk, dn.AllDisks)
|
|
require.Equal(t, badDisk, dn.BadDisks)
|
|
}
|
|
|
|
func TestCalculateDpLimitByDiskCapacity(t *testing.T) {
|
|
t.Run("SSD", func(t *testing.T) {
|
|
cfg := newClusterConfig()
|
|
cfg.DpLimitSsdBaseCount = 150
|
|
cfg.DpLimitSsdFactor = 50 // 5.0 in tenths
|
|
cluster := &Cluster{cfg: cfg}
|
|
|
|
dn := &DataNode{
|
|
AllDisks: []string{"/data1", "/data2"},
|
|
Total: 4096 * util.GB, // 4TB in bytes
|
|
MediaType: proto.MediaType_SSD,
|
|
}
|
|
|
|
got := dn.calculateDpLimitByDiskCapacity(cluster)
|
|
// expected = base*diskCount + (totalGB*factor)/(120*10) where factor is tenths
|
|
want := uint64(150*2 + (4096*50)/(120*10))
|
|
require.Equal(t, want, got)
|
|
})
|
|
|
|
t.Run("HDD", func(t *testing.T) {
|
|
cfg := newClusterConfig()
|
|
cfg.DpLimitHddBaseCount = 100
|
|
cfg.DpLimitHddFactor = 20 // 2.0 in tenths
|
|
cluster := &Cluster{cfg: cfg}
|
|
|
|
dn := &DataNode{
|
|
AllDisks: []string{"/data1"},
|
|
Total: 14336 * util.GB, // 14TB in bytes
|
|
MediaType: proto.MediaType_HDD,
|
|
}
|
|
|
|
got := dn.calculateDpLimitByDiskCapacity(cluster)
|
|
// expected = base*diskCount + (totalGB*factor)/(120*10)
|
|
want := uint64(100*1 + (14336*20)/(120*10))
|
|
require.Equal(t, want, got)
|
|
})
|
|
}
|
