micro/network/router/default_table_test.go
2019-07-01 23:38:48 +01:00

190 lines
4.7 KiB
Go

package router
import "testing"
func testSetup() (Table, Route) {
table := NewTable()
route := Route{
Destination: "dest.svc",
Gateway: "dest.gw",
Router: "dest.router",
Network: "dest.network",
Metric: 10,
}
return table, route
}
func TestAdd(t *testing.T) {
table, route := testSetup()
if err := table.Add(route); err != nil {
t.Errorf("error adding route: %s", err)
}
if table.Size() != 1 {
t.Errorf("invalid number of routes. expected: 1, given: %d", table.Size())
}
// adds new route for the original destination
route.Gateway = "dest.gw2"
if err := table.Add(route); err != nil {
t.Errorf("error adding route: %s", err)
}
if table.Size() != 2 {
t.Errorf("invalid number of routes. expected: 2, given: %d", table.Size())
}
// overrides an existing route: the size of the table does not change
route.Metric = 100
route.Policy = OverrideIfExists
if err := table.Add(route); err != nil {
t.Errorf("error adding route: %s", err)
}
if table.Size() != 2 {
t.Errorf("invalid number of routes. expected: 2, given: %d", table.Size())
}
// dont add new route if it already exists
route.Policy = IgnoreIfExists
if err := table.Add(route); err != nil {
t.Errorf("error adding route: %s", err)
}
if table.Size() != 2 {
t.Errorf("invalid number of routes. expected: 2, given: %d", table.Size())
}
// adding the same route under this policy should error
route.Policy = AddIfNotExists
if err := table.Add(route); err != ErrDuplicateRoute {
t.Errorf("error adding route. Expected error: %s, Given: %s", ErrDuplicateRoute, err)
}
}
func TestDelete(t *testing.T) {
table, route := testSetup()
if err := table.Add(route); err != nil {
t.Errorf("error adding route: %s", err)
}
if table.Size() != 1 {
t.Errorf("invalid number of routes. expected: 1, given: %d", table.Size())
}
// should fail to delete non-existant route
oldDest := route.Destination
route.Destination = "randDest"
if err := table.Delete(route); err != ErrRouteNotFound {
t.Errorf("error deleting route. Expected error: %s, given: %s", ErrRouteNotFound, err)
}
if table.Size() != 1 {
t.Errorf("invalid number of routes. expected: %d, given: %d", 1, table.Size())
}
// we should be able to delete the routes now
route.Destination = oldDest
if err := table.Delete(route); err != nil {
t.Errorf("error deleting route: %s", err)
}
if table.Size() != 0 {
t.Errorf("invalid number of routes. expected: %d, given: %d", 0, table.Size())
}
}
func TestUpdate(t *testing.T) {
table, route := testSetup()
if err := table.Add(route); err != nil {
t.Errorf("error adding route: %s", err)
}
if table.Size() != 1 {
t.Errorf("invalid number of routes. expected: 1, given: %d", table.Size())
}
route.Metric = 200
if err := table.Update(route); err != nil {
t.Errorf("error updating route: %s", err)
}
if table.Size() != 1 {
t.Errorf("invalid number of routes. expected: 1, given: %d", table.Size())
}
// this should add a new route; we are hashing routes on <destination, gateway, network>
route.Destination = "new.dest"
if err := table.Update(route); err != nil {
t.Errorf("error updating route: %s", err)
}
// NOTE: default policy is AddIfNotExists so the new route will be added here
if table.Size() != 2 {
t.Errorf("invalid number of routes. expected: 2, given: %d", table.Size())
}
// this should add a new route; we are hashing routes on <destination, gateway, network>
route.Gateway = "new.gw"
if err := table.Update(route); err != nil {
t.Errorf("error updating route: %s", err)
}
// NOTE: default policy is AddIfNotExists so the new route will be added here
if table.Size() != 3 {
t.Errorf("invalid number of routes. expected: 3, given: %d", table.Size())
}
// this should NOTE add a new route; we are setting the policy to IgnoreIfExists
route.Destination = "rand.dest"
route.Policy = IgnoreIfExists
if err := table.Update(route); err != ErrRouteNotFound {
t.Errorf("error updating route. Expected error: %s, given: %s", ErrRouteNotFound, err)
}
if table.Size() != 3 {
t.Errorf("invalid number of routes. expected: 3, given: %d", table.Size())
}
}
func TestList(t *testing.T) {
table, route := testSetup()
dest := []string{"one.svc", "two.svc", "three.svc"}
for i := 0; i < len(dest); i++ {
route.Destination = dest[i]
if err := table.Add(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(dest) {
t.Errorf("incorrect number of routes listed. Expected: %d, Given: %d", len(dest), len(routes))
}
if len(routes) != table.Size() {
t.Errorf("mismatch number of routes and table size. Routes: %d, Size: %d", len(routes), table.Size())
}
}