mirror of
https://github.com/openziti/ziti.git
synced 2026-10-11 23:12:35 +00:00
Paul Lorenz
120 lines
2.9 KiB
Go
120 lines
2.9 KiB
Go
package network
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/michaelquigley/pfxlog"
|
|
"github.com/openziti/ziti/controller/db"
|
|
"github.com/openziti/ziti/controller/xt"
|
|
"github.com/openziti/ziti/controller/xt_smartrouting"
|
|
"github.com/openziti/ziti/common/ctrl_msg"
|
|
)
|
|
|
|
func TestRouteSender_DestroysTerminatorWhenInvalidOnHandleRouteSendAndWeControl(t *testing.T) {
|
|
ctx := db.NewTestContext(t)
|
|
defer ctx.Cleanup()
|
|
|
|
config := newTestConfig(ctx)
|
|
defer close(config.closeNotify)
|
|
|
|
network, err := NewNetwork(config)
|
|
ctx.NoError(err)
|
|
|
|
entityHelper := newTestEntityHelper(ctx, network)
|
|
logger := pfxlog.ChannelLogger("test")
|
|
|
|
router1 := entityHelper.addTestRouter()
|
|
router2 := entityHelper.addTestRouter()
|
|
path := &Path{
|
|
Nodes: []*Router{router1, router2},
|
|
}
|
|
|
|
svc := entityHelper.addTestService("svc")
|
|
|
|
instanceId := "instanceId"
|
|
|
|
term := entityHelper.addTestTerminator(svc.Id, router1.Id, instanceId, true)
|
|
term.Binding = "edge"
|
|
|
|
errCode := byte(ctrl_msg.ErrorTypeInvalidTerminator)
|
|
|
|
rs := routeSender{
|
|
serviceCounters: network,
|
|
terminators: network.Terminators,
|
|
attendance: make(map[string]bool),
|
|
}
|
|
|
|
status := &RouteStatus{
|
|
Router: router1,
|
|
ErrorCode: &errCode,
|
|
Success: false,
|
|
Attempt: 1,
|
|
Err: "THIS IS A TEST",
|
|
}
|
|
|
|
peerData, cleanup, err := rs.handleRouteSend(1, path, xt_smartrouting.NewFactory().NewStrategy(), status, term, logger)
|
|
ctx.Error(err)
|
|
ctx.ErrorContains(err, status.Err)
|
|
ctx.Nil(peerData)
|
|
ctx.Empty(cleanup)
|
|
|
|
newTerm, err := network.Terminators.Read(term.Id)
|
|
ctx.Error(err)
|
|
ctx.Nil(newTerm)
|
|
}
|
|
|
|
func TestRouteSender_SetPrecidenceToNilTerminatorWhenInvalidOnHandleRouteSendAndWeDontControl(t *testing.T) {
|
|
ctx := db.NewTestContext(t)
|
|
defer ctx.Cleanup()
|
|
|
|
config := newTestConfig(ctx)
|
|
defer close(config.closeNotify)
|
|
|
|
network, err := NewNetwork(config)
|
|
ctx.NoError(err)
|
|
|
|
entityHelper := newTestEntityHelper(ctx, network)
|
|
logger := pfxlog.ChannelLogger("test")
|
|
|
|
router1 := entityHelper.addTestRouter()
|
|
router2 := entityHelper.addTestRouter()
|
|
path := &Path{
|
|
Nodes: []*Router{router1, router2},
|
|
}
|
|
|
|
svc := entityHelper.addTestService("svc")
|
|
|
|
identity := "identity"
|
|
|
|
term := entityHelper.addTestTerminator(svc.Id, router1.Id, identity, true)
|
|
term.Binding = "DNE"
|
|
|
|
errCode := byte(ctrl_msg.ErrorTypeInvalidTerminator)
|
|
|
|
rs := routeSender{
|
|
serviceCounters: network,
|
|
terminators: network.Terminators,
|
|
attendance: make(map[string]bool),
|
|
}
|
|
|
|
status := &RouteStatus{
|
|
Router: router1,
|
|
ErrorCode: &errCode,
|
|
Success: false,
|
|
Attempt: 1,
|
|
Err: "THIS IS A TEST",
|
|
}
|
|
|
|
peerData, cleanup, err := rs.handleRouteSend(1, path, xt_smartrouting.NewFactory().NewStrategy(), status, term, logger)
|
|
ctx.Error(err)
|
|
ctx.ErrorContains(err, status.Err)
|
|
ctx.Nil(peerData)
|
|
ctx.Empty(cleanup)
|
|
|
|
newTerm, err := network.Terminators.Read(term.Id)
|
|
ctx.NoError(err)
|
|
ctx.NotNil(newTerm)
|
|
|
|
ctx.Equal(xt.Precedences.Failed, newTerm.GetPrecedence())
|
|
}
|