131 lines
2.5 KiB
Go
131 lines
2.5 KiB
Go
package router
|
|
|
|
import (
|
|
"fmt"
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/micro/go-micro/registry/memory"
|
|
"github.com/micro/go-micro/util/log"
|
|
)
|
|
|
|
func routerTestSetup() Router {
|
|
r := memory.NewRegistry()
|
|
return newRouter(Registry(r))
|
|
}
|
|
|
|
func TestRouterStartStop(t *testing.T) {
|
|
r := routerTestSetup()
|
|
|
|
if err := r.Start(); err != nil {
|
|
t.Errorf("failed to start router: %v", err)
|
|
}
|
|
|
|
_, err := r.Advertise()
|
|
if err != nil {
|
|
t.Errorf("failed to start advertising: %v", err)
|
|
}
|
|
|
|
if err := r.Stop(); err != nil {
|
|
t.Errorf("failed to stop router: %v", err)
|
|
}
|
|
log.Debugf("TestRouterStartStop STOPPED")
|
|
}
|
|
|
|
func TestRouterAdvertise(t *testing.T) {
|
|
r := routerTestSetup()
|
|
|
|
// lower the advertise interval
|
|
AdvertiseEventsTick = 500 * time.Millisecond
|
|
|
|
if err := r.Start(); err != nil {
|
|
t.Errorf("failed to start router: %v", err)
|
|
}
|
|
|
|
ch, err := r.Advertise()
|
|
if err != nil {
|
|
t.Errorf("failed to start advertising: %v", err)
|
|
}
|
|
|
|
// receive announce event
|
|
ann := <-ch
|
|
log.Debugf("received announce advert: %v", ann)
|
|
|
|
// Generate random unique routes
|
|
nrRoutes := 5
|
|
routes := make([]Route, nrRoutes)
|
|
route := Route{
|
|
Service: "dest.svc",
|
|
Address: "dest.addr",
|
|
Gateway: "dest.gw",
|
|
Network: "dest.network",
|
|
Router: "src.router",
|
|
Link: "local",
|
|
Metric: 10,
|
|
}
|
|
|
|
for i := 0; i < nrRoutes; i++ {
|
|
testRoute := route
|
|
testRoute.Service = fmt.Sprintf("%s-%d", route.Service, i)
|
|
routes[i] = testRoute
|
|
}
|
|
|
|
var advertErr error
|
|
|
|
createDone := make(chan bool)
|
|
errChan := make(chan error)
|
|
|
|
var wg sync.WaitGroup
|
|
wg.Add(1)
|
|
go func() {
|
|
wg.Done()
|
|
defer close(createDone)
|
|
for _, route := range routes {
|
|
log.Debugf("Creating route %v", route)
|
|
if err := r.Table().Create(route); err != nil {
|
|
log.Debugf("Failed to create route: %v", err)
|
|
errChan <- err
|
|
return
|
|
}
|
|
}
|
|
}()
|
|
|
|
var adverts int
|
|
readDone := make(chan bool)
|
|
|
|
wg.Add(1)
|
|
go func() {
|
|
defer func() {
|
|
wg.Done()
|
|
readDone <- true
|
|
}()
|
|
for advert := range ch {
|
|
select {
|
|
case advertErr = <-errChan:
|
|
t.Errorf("failed advertising events: %v", advertErr)
|
|
default:
|
|
// do nothing for now
|
|
log.Debugf("Router advert received: %v", advert)
|
|
adverts += len(advert.Events)
|
|
}
|
|
return
|
|
}
|
|
}()
|
|
|
|
// done adding routes to routing table
|
|
<-createDone
|
|
// done reading adverts from the routing table
|
|
<-readDone
|
|
|
|
if adverts != nrRoutes {
|
|
t.Errorf("Expected %d adverts, received: %d", nrRoutes, adverts)
|
|
}
|
|
|
|
wg.Wait()
|
|
|
|
if err := r.Stop(); err != nil {
|
|
t.Errorf("failed to stop router: %v", err)
|
|
}
|
|
}
|