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  }