micro/network/router/default_table_test.go

183 lines
4.6 KiB
Go
Raw Normal View History

2019-07-02 00:57:27 +03:00
package router
import "testing"
2019-07-02 01:15:45 +03:00
func testSetup() (Table, Route) {
2019-07-02 00:57:27 +03:00
table := NewTable()
2019-07-02 01:15:45 +03:00
route := Route{
Destination: "dest.svc",
Gateway: "dest.gw",
Router: "dest.router",
Network: "dest.network",
Metric: 10,
2019-07-02 00:57:27 +03:00
}
2019-07-02 01:15:45 +03:00
return table, route
2019-07-02 00:57:27 +03:00
}
2019-07-02 01:15:45 +03:00
func TestAdd(t *testing.T) {
table, route := testSetup()
testTableSize := table.Size()
2019-07-02 00:57:27 +03:00
if err := table.Add(route); err != nil {
t.Errorf("error adding route: %s", err)
}
testTableSize += 1
2019-07-02 00:57:27 +03:00
// 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)
}
testTableSize += 1
2019-07-02 00:57:27 +03:00
// overrides an existing route
2019-07-02 00:57:27 +03:00
route.Metric = 100
route.Policy = Override
2019-07-02 00:57:27 +03:00
if err := table.Add(route); err != nil {
t.Errorf("error adding route: %s", err)
}
// the size of the table should not change when Override policy is used
if table.Size() != testTableSize {
t.Errorf("invalid number of routes. expected: %d, given: %d", testTableSize, table.Size())
2019-07-02 00:57:27 +03:00
}
// dont add new route if it already exists
route.Policy = Skip
2019-07-02 00:57:27 +03:00
if err := table.Add(route); err != nil {
t.Errorf("error adding route: %s", err)
}
// the size of the table should not change if Skip policy is used
if table.Size() != testTableSize {
t.Errorf("invalid number of routes. expected: %d, given: %d", testTableSize, table.Size())
2019-07-02 00:57:27 +03:00
}
// adding the same route under Insert policy must error
route.Policy = Insert
2019-07-02 00:57:27 +03:00
if err := table.Add(route); err != ErrDuplicateRoute {
t.Errorf("error adding route. Expected error: %s, Given: %s", ErrDuplicateRoute, err)
}
}
func TestDelete(t *testing.T) {
2019-07-02 01:15:45 +03:00
table, route := testSetup()
testTableSize := table.Size()
2019-07-02 00:57:27 +03:00
if err := table.Add(route); err != nil {
t.Errorf("error adding route: %s", err)
}
testTableSize += 1
2019-07-02 00:57:27 +03:00
// should fail to delete non-existant route
prevDest := route.Destination
2019-07-02 00:57:27 +03:00
route.Destination = "randDest"
if err := table.Delete(route); err != ErrRouteNotFound {
t.Errorf("error deleting route. Expected error: %s, given: %s", ErrRouteNotFound, err)
}
// we should be able to delete the existing route
route.Destination = prevDest
2019-07-02 00:57:27 +03:00
if err := table.Delete(route); err != nil {
t.Errorf("error deleting route: %s", err)
}
testTableSize -= 1
2019-07-02 00:57:27 +03:00
if table.Size() != testTableSize {
t.Errorf("invalid number of routes. expected: %d, given: %d", testTableSize, table.Size())
2019-07-02 00:57:27 +03:00
}
}
func TestUpdate(t *testing.T) {
2019-07-02 01:15:45 +03:00
table, route := testSetup()
testTableSize := table.Size()
2019-07-02 00:57:27 +03:00
if err := table.Add(route); err != nil {
t.Errorf("error adding route: %s", err)
}
testTableSize += 1
2019-07-02 00:57:27 +03:00
// change the metric of the original route
2019-07-02 00:57:27 +03:00
route.Metric = 200
if err := table.Update(route); err != nil {
t.Errorf("error updating route: %s", err)
}
// the size of the table should not change as we're only updating the metric of an existing route
if table.Size() != testTableSize {
t.Errorf("invalid number of routes. expected: %d, given: %d", testTableSize, table.Size())
2019-07-02 00:57:27 +03:00
}
// this should add a new route
2019-07-02 00:57:27 +03:00
route.Destination = "new.dest"
if err := table.Update(route); err != nil {
t.Errorf("error updating route: %s", err)
}
testTableSize += 1
2019-07-02 00:57:27 +03:00
// Default policy is Insert so the new route will be added here since the route does not exist
if table.Size() != testTableSize {
t.Errorf("invalid number of routes. expected: %d, given: %d", testTableSize, table.Size())
2019-07-02 00:57:27 +03:00
}
// this should add a new route
2019-07-02 00:57:27 +03:00
route.Gateway = "new.gw"
if err := table.Update(route); err != nil {
t.Errorf("error updating route: %s", err)
}
testTableSize += 1
2019-07-02 00:57:27 +03:00
if table.Size() != testTableSize {
t.Errorf("invalid number of routes. expected: %d, given: %d", testTableSize, table.Size())
2019-07-02 00:57:27 +03:00
}
// this should NOT add a new route as we are setting the policy to Skip
2019-07-02 00:57:27 +03:00
route.Destination = "rand.dest"
route.Policy = Skip
2019-07-02 00:57:27 +03:00
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: %d, given: %d", testTableSize, table.Size())
2019-07-02 00:57:27 +03:00
}
}
2019-07-02 01:15:45 +03:00
2019-07-02 01:19:35 +03:00
func TestList(t *testing.T) {
2019-07-02 01:15:45 +03:00
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())
}
}