package router import "testing" func testSetup() (*table, Route) { table := newTable() route := Route{ Service: "dest.svc", Gateway: "dest.gw", Network: "dest.network", Link: "det.link", Metric: 10, } return table, route } func TestCreate(t *testing.T) { table, route := testSetup() if err := table.Create(route); err != nil { t.Errorf("error adding route: %s", err) } // adds new route for the original destination route.Gateway = "dest.gw2" if err := table.Create(route); err != nil { t.Errorf("error adding route: %s", err) } // adding the same route under Insert policy must error if err := table.Create(route); err != ErrDuplicateRoute { t.Errorf("error adding route. Expected error: %s, found: %s", ErrDuplicateRoute, err) } } func TestDelete(t *testing.T) { table, route := testSetup() if err := table.Create(route); err != nil { t.Errorf("error adding route: %s", err) } // should fail to delete non-existant route prevSvc := route.Service route.Service = "randDest" if err := table.Delete(route); err != ErrRouteNotFound { t.Errorf("error deleting route. Expected: %s, found: %s", ErrRouteNotFound, err) } // we should be able to delete the existing route route.Service = prevSvc if err := table.Delete(route); err != nil { t.Errorf("error deleting route: %s", err) } } func TestUpdate(t *testing.T) { table, route := testSetup() if err := table.Create(route); err != nil { t.Errorf("error adding route: %s", err) } // change the metric of the original route route.Metric = 200 if err := table.Update(route); err != nil { t.Errorf("error updating route: %s", err) } // this should add a new route route.Service = "rand.dest" if err := table.Update(route); err != nil { t.Errorf("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.Errorf("error adding route: %s", err) } } routes, err := table.List() if err != nil { t.Errorf("error listing routes: %s", err) } if len(routes) != len(svc) { t.Errorf("incorrect number of routes listed. Expected: %d, found: %d", len(svc), len(routes)) } } func TestLookup(t *testing.T) { table, route := testSetup() svc := []string{"svc1", "svc2", "svc3"} net := []string{"net1", "net2", "net1"} gw := []string{"gw1", "gw2", "gw3"} for i := 0; i < len(svc); i++ { route.Service = svc[i] route.Network = net[i] route.Gateway = gw[i] if err := table.Create(route); err != nil { t.Errorf("error adding route: %s", err) } } // return all routes query := NewQuery() routes, err := table.Lookup(query) if err != nil { t.Errorf("error looking up routes: %s", err) } // query particular net query = NewQuery(QueryNetwork("net1")) routes, err = table.Lookup(query) if err != nil { t.Errorf("error looking up routes: %s", err) } if len(routes) != 2 { t.Errorf("incorrect number of routes returned. Expected: %d, found: %d", 2, len(routes)) } // query particular gateway gateway := "gw1" query = NewQuery(QueryGateway(gateway)) routes, err = table.Lookup(query) if err != nil { t.Errorf("error looking up routes: %s", err) } if len(routes) != 1 { t.Errorf("incorrect number of routes returned. Expected: %d, found: %d", 1, len(routes)) } if routes[0].Gateway != gateway { t.Errorf("incorrect route returned. Expected gateway: %s, found: %s", gateway, routes[0].Gateway) } // query particular route network := "net1" query = NewQuery( QueryGateway(gateway), QueryNetwork(network), ) routes, err = table.Lookup(query) if err != nil { t.Errorf("error looking up routes: %s", err) } if len(routes) != 1 { t.Errorf("incorrect number of routes returned. Expected: %d, found: %d", 1, len(routes)) } if routes[0].Gateway != gateway { t.Errorf("incorrect route returned. Expected gateway: %s, found: %s", gateway, routes[0].Gateway) } if routes[0].Network != network { t.Errorf("incorrect network returned. Expected network: %s, found: %s", network, routes[0].Network) } // bullshit route query query = NewQuery(QueryService("foobar")) routes, err = table.Lookup(query) if err != ErrRouteNotFound { t.Errorf("error looking up routes. Expected: %s, found: %s", ErrRouteNotFound, err) } if len(routes) != 0 { t.Errorf("incorrect number of routes returned. Expected: %d, found: %d", 0, len(routes)) } }