Add internal/server/user/server{,_test}.go

Co-authored-by: Roberto Abdelkader Martínez Pérez <robertomartinezp@gmail.com>
This commit is contained in:
pancho horrillo
2019-10-07 19:05:59 +02:00
parent 10da324a57
commit 10586bae95
3 changed files with 155 additions and 1 deletions
+1 -1
View File
@@ -4,7 +4,7 @@ go 1.12
require ( require (
github.com/google/uuid v1.1.1 // indirect github.com/google/uuid v1.1.1 // indirect
github.com/gorilla/mux v1.7.3 // indirect github.com/gorilla/mux v1.7.3
github.com/spf13/cobra v0.0.5 github.com/spf13/cobra v0.0.5
gopkg.in/h2non/gock.v1 v1.0.15 gopkg.in/h2non/gock.v1 v1.0.15
) )
+31
View File
@@ -0,0 +1,31 @@
package user
import (
"net/http"
"sync"
"github.com/gorilla/mux"
)
type swappableMux struct {
m sync.RWMutex
root mux.Router
}
func (sm *swappableMux) get() mux.Router {
sm.m.RLock()
defer sm.m.RUnlock()
return sm.root
}
func (sm *swappableMux) set(mux mux.Router) {
sm.m.Lock()
sm.root = mux
sm.m.Unlock()
}
var Server http.Server
func Run() {
}
+123
View File
@@ -0,0 +1,123 @@
package user
import (
"reflect"
"testing"
"time"
"github.com/gorilla/mux"
)
// TODO TestRunRunsAnHTTPServer(t *testing.T) {}
func TestSwappableMuxGetReturnsTheCurrentMux(t *testing.T) {
sm := swappableMux{}
mux := sm.get()
if !reflect.DeepEqual(mux, sm.root) {
t.Errorf("Returned mux is not the same %#v", mux)
}
}
func TestSwappableMuxGetReturnsADifferentInstance(t *testing.T) {
sm := swappableMux{}
mux := sm.get()
if &mux == &sm.root {
t.Error("Returned mux is the same instance")
}
}
func TestSwappableMuxGetWaitsForTheMutexToBeReleased(t *testing.T) {
sm := swappableMux{}
sm.m.Lock()
defer sm.m.Unlock()
c := make(chan mux.Router)
go func() { c <- sm.get() }()
time.Sleep(10 * time.Millisecond)
select {
case <-c:
t.Error("Didn't acquire the mutex")
default:
}
}
func TestSwappableMuxGetIsAbleToReadWhileOthersAreReading(t *testing.T) {
sm := swappableMux{}
sm.m.RLock()
defer sm.m.RUnlock()
c := make(chan mux.Router)
go func() { c <- sm.get() }()
time.Sleep(10 * time.Millisecond)
select {
case <-c:
default:
t.Error("The mutex cannot be acquired")
}
}
func TestSwappableMuxSetSetsTheGivenMux(t *testing.T) {
sm := swappableMux{}
mux := mux.Router{
KeepContext: true,
}
sm.set(mux)
if !sm.root.KeepContext {
t.Error("mux not set")
}
}
func TestSwappableMuxSetSetsADifferentInstance(t *testing.T) {
sm := swappableMux{}
mux := mux.Router{}
sm.set(mux)
if &mux == &sm.root {
t.Error("Set mux is the same instance")
}
}
func TestSwappableMuxSetWaitsForWriterToReleaseMutex(t *testing.T) {
sm := swappableMux{}
sm.m.Lock()
defer sm.m.Unlock()
c := make(chan bool)
go func() { sm.set(mux.Router{}); c <- true }()
time.Sleep(10 * time.Millisecond)
select {
case <-c:
t.Error("Didn't acquire the mutex")
default:
}
}
func TestSwappableMuxSetWaitsForReadersToReleaseMutex(t *testing.T) {
sm := swappableMux{}
sm.m.RLock()
defer sm.m.RUnlock()
c := make(chan bool)
go func() { sm.set(mux.Router{}); c <- true }()
time.Sleep(10 * time.Millisecond)
select {
case <-c:
t.Error("Didn't acquire the mutex")
default:
}
}