Implemented user server state aka route list
This commit is contained in:
@@ -0,0 +1,37 @@
|
|||||||
|
package user
|
||||||
|
|
||||||
|
import (
|
||||||
|
"sync"
|
||||||
|
|
||||||
|
"github.com/BBVA/kapow/internal/server/model"
|
||||||
|
)
|
||||||
|
|
||||||
|
type safeRouteList struct {
|
||||||
|
rs []model.Route
|
||||||
|
m sync.RWMutex
|
||||||
|
}
|
||||||
|
|
||||||
|
var Routes = New()
|
||||||
|
|
||||||
|
func New() safeRouteList {
|
||||||
|
return safeRouteList{}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (srl *safeRouteList) Append(r model.Route) {
|
||||||
|
srl.m.Lock()
|
||||||
|
srl.rs = append(srl.rs, r)
|
||||||
|
srl.m.Unlock()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (srl *safeRouteList) Snapshot() []model.Route {
|
||||||
|
srl.m.RLock()
|
||||||
|
defer srl.m.RUnlock()
|
||||||
|
|
||||||
|
if srl.rs == nil {
|
||||||
|
return nil
|
||||||
|
} else {
|
||||||
|
rs := make([]model.Route, len(srl.rs))
|
||||||
|
copy(rs, srl.rs)
|
||||||
|
return rs
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,136 @@
|
|||||||
|
package user
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/BBVA/kapow/internal/server/model"
|
||||||
|
"reflect"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestNewReturnAnEmptyStruct(t *testing.T) {
|
||||||
|
srl := New()
|
||||||
|
|
||||||
|
if len(srl.rs) != 0 {
|
||||||
|
t.Error("Unexpected member in slice")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestPackageHaveASingletonEmptyRouteList(t *testing.T) {
|
||||||
|
if !reflect.DeepEqual(Routes, New()) {
|
||||||
|
t.Error("Routes is not an empty safeRouteList")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAppendAppendsANewRouteToTheList(t *testing.T) {
|
||||||
|
srl := New()
|
||||||
|
|
||||||
|
srl.Append(model.Route{})
|
||||||
|
|
||||||
|
if len(srl.rs) == 0 {
|
||||||
|
t.Error("Route not added to the list")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAppendAdquiresMutexBeforeAdding(t *testing.T) {
|
||||||
|
srl := New()
|
||||||
|
|
||||||
|
srl.m.Lock()
|
||||||
|
defer srl.m.Unlock()
|
||||||
|
go srl.Append(model.Route{})
|
||||||
|
|
||||||
|
time.Sleep(10 * time.Millisecond)
|
||||||
|
|
||||||
|
if len(srl.rs) != 0 {
|
||||||
|
t.Error("Route added while mutex was adquired")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAppendAddsRouteAfterMutexIsReleased(t *testing.T) {
|
||||||
|
srl := New()
|
||||||
|
|
||||||
|
srl.m.Lock()
|
||||||
|
go srl.Append(model.Route{})
|
||||||
|
srl.m.Unlock()
|
||||||
|
|
||||||
|
time.Sleep(10 * time.Millisecond)
|
||||||
|
|
||||||
|
if len(srl.rs) != 1 {
|
||||||
|
t.Error("Route not added after mutex release")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSnapshotReturnTheCurrentListOfRoutes(t *testing.T) {
|
||||||
|
srl := New()
|
||||||
|
srl.Append(model.Route{Id: "FOO"})
|
||||||
|
|
||||||
|
rs := srl.Snapshot()
|
||||||
|
|
||||||
|
if !reflect.DeepEqual(srl.rs, rs) {
|
||||||
|
t.Error("Route list returned is not the current one")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSnapshotReturnADeepCopyOfTheListWhenEmpty(t *testing.T) {
|
||||||
|
srl := New()
|
||||||
|
|
||||||
|
rs := srl.Snapshot()
|
||||||
|
|
||||||
|
if rs != nil {
|
||||||
|
t.Fatal("Route list copy is not empty")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSnapshotReturnADeepCopyOfTheListWhenNonEmpty(t *testing.T) {
|
||||||
|
srl := New()
|
||||||
|
srl.Append(model.Route{Id: "FOO"})
|
||||||
|
|
||||||
|
rs := srl.Snapshot()
|
||||||
|
|
||||||
|
if &rs == &srl.rs {
|
||||||
|
t.Fatal("Route list is not a copy")
|
||||||
|
}
|
||||||
|
|
||||||
|
for i := 0; i < len(rs); i++ {
|
||||||
|
if &rs[i] == &srl.rs[i] {
|
||||||
|
t.Errorf("Route %q is not a copy", i)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSnapshotWaitsForTheWriterToFinish(t *testing.T) {
|
||||||
|
srl := New()
|
||||||
|
srl.Append(model.Route{Id: "FOO"})
|
||||||
|
|
||||||
|
srl.m.Lock()
|
||||||
|
defer srl.m.Unlock()
|
||||||
|
|
||||||
|
c := make(chan []model.Route)
|
||||||
|
go func() { c <- srl.Snapshot() }()
|
||||||
|
|
||||||
|
time.Sleep(10 * time.Millisecond)
|
||||||
|
|
||||||
|
select {
|
||||||
|
case <-c:
|
||||||
|
t.Error("Route list readed while mutex was adquired")
|
||||||
|
default: // This default prevents the select from being blocking
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSnapshotNonBlockingReadWithOtherReaders(t *testing.T) {
|
||||||
|
srl := New()
|
||||||
|
srl.Append(model.Route{Id: "FOO"})
|
||||||
|
|
||||||
|
srl.m.RLock()
|
||||||
|
defer srl.m.RUnlock()
|
||||||
|
|
||||||
|
c := make(chan []model.Route)
|
||||||
|
go func() { c <- srl.Snapshot() }()
|
||||||
|
|
||||||
|
time.Sleep(10 * time.Millisecond)
|
||||||
|
|
||||||
|
select {
|
||||||
|
case <-c:
|
||||||
|
default: // This default prevents the select from being blocking
|
||||||
|
t.Error("Route list couldn't be readed while mutex was adquired for read")
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user