github.com/cloudreve/Cloudreve/v3@v3.0.0-20240224133659-3edb00a6484c/pkg/cluster/node_test.go (about) 1 package cluster 2 3 import ( 4 model "github.com/cloudreve/Cloudreve/v3/models" 5 "github.com/stretchr/testify/assert" 6 "testing" 7 ) 8 9 func TestNewNodeFromDBModel(t *testing.T) { 10 a := assert.New(t) 11 a.IsType(&SlaveNode{}, NewNodeFromDBModel(&model.Node{ 12 Type: model.SlaveNodeType, 13 })) 14 a.IsType(&MasterNode{}, NewNodeFromDBModel(&model.Node{ 15 Type: model.MasterNodeType, 16 })) 17 }