package registry import ( "fmt" "testing" "github.com/unistack-org/micro/v3/router" ) func testSetup() (*table, router.Route) { routr := NewRouter().(*rtr) table := newTable(routr.lookup) route := router.Route{ Service: "dest.svc", Address: "dest.addr", Gateway: "dest.gw", Network: "dest.network", Router: "src.router", Link: "det.link", Metric: 10, } return table, route } func TestCreate(t *testing.T) { table, route := testSetup() if err := table.Create(route); err != nil { t.Fatalf("error adding route: %s", err) } // adds new route for the original destination route.Gateway = "dest.gw2" if err := table.Create(route); err != nil { t.Fatalf("error adding route: %s", err) } // adding the same route under Insert policy must error if err := table.Create(route); err != router.ErrDuplicateRoute { t.Fatalf("error adding route. Expected error: %s, found: %s", router.ErrDuplicateRoute, err) } } func TestDelete(t *testing.T) { table, route := testSetup() if err := table.Create(route); err != nil { t.Fatalf("error adding route: %s", err) } // should fail to delete non-existant route prevSvc := route.Service route.Service = "randDest" if err := table.Delete(route); err != router.ErrRouteNotFound { t.Fatalf("error deleting route. Expected: %s, found: %s", router.ErrRouteNotFound, err) } // we should be able to delete the existing route route.Service = prevSvc if err := table.Delete(route); err != nil { t.Fatalf("error deleting route: %s", err) } } func TestUpdate(t *testing.T) { table, route := testSetup() if err := table.Create(route); err != nil { t.Fatalf("error adding route: %s", err) } // change the metric of the original route route.Metric = 200 if err := table.Update(route); err != nil { t.Fatalf("error updating route: %s", err) } // this should add a new route route.Service = "rand.dest" if err := table.Update(route); err != nil { t.Fatalf("error updating route: %s", err) } } func TestList(t *testing.T) { table, route := testSetup() svc := []string{"one.svc", "two.svc", "three.svc"} for i := 0; i < len(svc); i++ { route.Service = svc[i] if err := table.Create(route); err != nil { t.Fatalf("error adding route: %s", err) } } routes, err := table.List() if err != nil { t.Fatalf("error listing routes: %s", err) } if len(routes) != len(svc) { t.Fatalf("incorrect number of routes listed. Expected: %d, found: %d", len(svc), len(routes)) } } func TestQuery(t *testing.T) { table, route := testSetup() svc := []string{"svc1", "svc2", "svc3", "svc1"} net := []string{"net1", "net2", "net1", "net3"} gw := []string{"gw1", "gw2", "gw3", "gw3"} rtr := []string{"rtr1", "rt2", "rt3", "rtr3"} for i := 0; i < len(svc); i++ { route.Service = svc[i] route.Network = net[i] route.Gateway = gw[i] route.Router = rtr[i] route.Link = router.DefaultLink if err := table.Create(route); err != nil { t.Fatalf("error adding route: %s", err) } } // return all routes routes, err := table.Query() if err != nil { t.Fatalf("error looking up routes: %s", err) } else if len(routes) == 0 { t.Fatalf("error looking up routes: not found") } // query routes particular network network := "net1" routes, err = table.Query(router.QueryNetwork(network)) if err != nil { t.Fatalf("error looking up routes: %s", err) } if len(routes) != 2 { t.Fatalf("incorrect number of routes returned. Expected: %d, found: %d", 2, len(routes)) } for _, route := range routes { if route.Network != network { t.Fatalf("incorrect route returned. Expected network: %s, found: %s", network, route.Network) } } // query routes for particular gateway gateway := "gw1" routes, err = table.Query(router.QueryGateway(gateway)) if err != nil { t.Fatalf("error looking up routes: %s", err) } if len(routes) != 1 { t.Fatalf("incorrect number of routes returned. Expected: %d, found: %d", 1, len(routes)) } if routes[0].Gateway != gateway { t.Fatalf("incorrect route returned. Expected gateway: %s, found: %s", gateway, routes[0].Gateway) } // query routes for particular router rt := "rtr1" routes, err = table.Query(router.QueryRouter(rt)) if err != nil { t.Fatalf("error looking up routes: %s", err) } if len(routes) != 1 { t.Fatalf("incorrect number of routes returned. Expected: %d, found: %d", 1, len(routes)) } if routes[0].Router != rt { t.Fatalf("incorrect route returned. Expected router: %s, found: %s", rt, routes[0].Router) } // query particular gateway and network query := []router.QueryOption{ router.QueryGateway(gateway), router.QueryNetwork(network), router.QueryRouter(rt), } routes, err = table.Query(query...) if err != nil { t.Fatalf("error looking up routes: %s", err) } if len(routes) != 1 { t.Fatalf("incorrect number of routes returned. Expected: %d, found: %d", 1, len(routes)) } if routes[0].Gateway != gateway { t.Fatalf("incorrect route returned. Expected gateway: %s, found: %s", gateway, routes[0].Gateway) } if routes[0].Network != network { t.Fatalf("incorrect network returned. Expected network: %s, found: %s", network, routes[0].Network) } if routes[0].Router != rt { t.Fatalf("incorrect route returned. Expected router: %s, found: %s", rt, routes[0].Router) } // non-existen route query routes, err = table.Query(router.QueryService("foobar")) if err != router.ErrRouteNotFound { t.Fatalf("error looking up routes. Expected: %s, found: %s", router.ErrRouteNotFound, err) } if len(routes) != 0 { t.Fatalf("incorrect number of routes returned. Expected: %d, found: %d", 0, len(routes)) } // query NO routes query = []router.QueryOption{ router.QueryGateway(gateway), router.QueryNetwork(network), router.QueryLink("network"), } routes, err = table.Query(query...) if err != nil { t.Fatalf("error looking up routes: %s", err) } if len(routes) > 0 { t.Fatalf("incorrect number of routes returned. Expected: %d, found: %d", 0, len(routes)) } // insert local routes to query for i := 0; i < 2; i++ { route.Link = "foobar" route.Address = fmt.Sprintf("local.route.address-%d", i) if err := table.Create(route); err != nil { t.Fatalf("error adding route: %s", err) } } // query local routes query = []router.QueryOption{ router.QueryGateway("*"), router.QueryNetwork("*"), router.QueryLink("foobar"), } routes, err = table.Query(query...) if err != nil { t.Fatalf("error looking up routes: %s", err) } if len(routes) != 2 { t.Fatalf("incorrect number of routes returned. Expected: %d, found: %d", 2, len(routes)) } // add two different routes for svcX with different metric for i := 0; i < 2; i++ { route.Service = "svcX" route.Address = fmt.Sprintf("svcX.route.address-%d", i) route.Metric = int64(100 + i) route.Link = router.DefaultLink if err := table.Create(route); err != nil { t.Fatalf("error adding route: %s", err) } } query = []router.QueryOption{ router.QueryService("svcX"), } routes, err = table.Query(query...) if err != nil { t.Fatalf("error looking up routes: %s", err) } if len(routes) != 2 { t.Fatalf("incorrect number of routes returned. Expected: %d, found: %d", 1, len(routes)) } } func TestFallback(t *testing.T) { r := &rtr{ options: router.DefaultOptions(), } route := router.Route{ Service: "go.micro.service.foo", Router: r.options.Id, Link: router.DefaultLink, Metric: router.DefaultLocalMetric, } r.table = newTable(func(s string) ([]router.Route, error) { return []router.Route{route}, nil }) r.start() rts, err := r.Lookup(router.QueryService("go.micro.service.foo")) if err != nil { t.Fatalf("error looking up service %s", err) } if len(rts) != 1 { t.Fatalf("incorrect number of routes returned %d", len(rts)) } // deleting from the table but the next query should invoke the fallback that we passed during new table creation if err := r.table.Delete(route); err != nil { t.Fatalf("error deleting route %s", err) } rts, err = r.Lookup(router.QueryService("go.micro.service.foo")) if err != nil { t.Fatalf("error looking up service %s", err) } if len(rts) != 1 { t.Fatalf("incorrect number of routes returned %d", len(rts)) } } func TestFallbackError(t *testing.T) { r := &rtr{ options: router.DefaultOptions(), } r.table = newTable(func(s string) ([]router.Route, error) { return nil, fmt.Errorf("ERROR") }) r.start() _, err := r.Lookup(router.QueryService("go.micro.service.foo")) if err == nil { t.Fatalf("expected error looking up service but none returned") } }