41 lines
988 B
Go
41 lines
988 B
Go
|
package router
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"git.hpds.cc/Component/network/frame"
|
||
|
|
"git.hpds.cc/Component/network/metadata"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestRouter(t *testing.T) {
|
||
|
|
router := Default([]string{"sfn-1"})
|
||
|
|
|
||
|
|
m := &metadata.Default{}
|
||
|
|
|
||
|
|
route := router.Route(m)
|
||
|
|
|
||
|
|
err := route.Add("conn-1", "sfn-1", []frame.Tag{frame.Tag(1)})
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
ids := route.GetForwardRoutes(frame.Tag(1))
|
||
|
|
assert.Equal(t, []string{"conn-1"}, ids)
|
||
|
|
|
||
|
|
err = route.Add("conn-2", "sfn-2", []frame.Tag{frame.Tag(2)})
|
||
|
|
assert.EqualError(t, err, "SFN[sfn-2] does not exist in config functions")
|
||
|
|
|
||
|
|
err = route.Add("conn-3", "sfn-1", []frame.Tag{frame.Tag(1)})
|
||
|
|
assert.EqualError(t, err, "SFN[sfn-1] is already linked to another connection")
|
||
|
|
|
||
|
|
err = route.Remove("conn-1")
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
ids = route.GetForwardRoutes(frame.Tag(1))
|
||
|
|
assert.Equal(t, []string{"conn-3"}, ids)
|
||
|
|
|
||
|
|
router.Clean()
|
||
|
|
|
||
|
|
ids = route.GetForwardRoutes(frame.Tag(1))
|
||
|
|
assert.Equal(t, []string(nil), ids)
|
||
|
|
}
|