mirror of
https://github.com/cubefs/cubefs.git
synced 2026-08-02 02:00:56 +00:00
62 lines
1.5 KiB
Go
62 lines
1.5 KiB
Go
package master
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/cubefs/cubefs/proto"
|
|
"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, DefaultZoneName, 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)
|
|
}
|
