Soam Vasani
828fa5f88c
Mutable mux test occasionally panics because the server is shut down before the client goroutine. Fix this by shutting down the client goroutines before shutting down the server.
118 lines
2.6 KiB
Go
118 lines
2.6 KiB
Go
/*
|
|
Copyright 2016 The Fission Authors.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package router
|
|
|
|
import (
|
|
"testing"
|
|
"net/http"
|
|
"github.com/gorilla/mux"
|
|
"log"
|
|
"io/ioutil"
|
|
"time"
|
|
)
|
|
|
|
func OldHandler(responseWriter http.ResponseWriter, request *http.Request) {
|
|
responseWriter.Write([]byte("old handler"))
|
|
}
|
|
func NewHandler(responseWriter http.ResponseWriter, request *http.Request) {
|
|
responseWriter.Write([]byte("new handler"))
|
|
}
|
|
|
|
func verifyRequest(expectedResponse string) {
|
|
resp, err := http.Get("http://localhost:3333")
|
|
if (err != nil) {
|
|
log.Panic("failed make get request")
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|
if (err != nil) {
|
|
log.Panic("failed to read response")
|
|
}
|
|
|
|
bodyStr := string(body)
|
|
log.Printf("Server responded with %v", bodyStr)
|
|
if (bodyStr != expectedResponse) {
|
|
log.Panic("Unexpected response")
|
|
}
|
|
}
|
|
|
|
func startServer(mr *mutableRouter) {
|
|
http.ListenAndServe(":3333", mr)
|
|
}
|
|
|
|
|
|
func spamServer(quit chan bool) {
|
|
i := 0
|
|
for {
|
|
select {
|
|
case <- quit:
|
|
break
|
|
default:
|
|
i = i + 1
|
|
resp, err := http.Get("http://localhost:3333")
|
|
if (err != nil) {
|
|
log.Panicf("failed make get request %v: %v", i, err)
|
|
}
|
|
resp.Body.Close()
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestMutableMux(t *testing.T) {
|
|
// make a simple mutable router
|
|
log.Print("Create mutable router")
|
|
muxRouter := mux.NewRouter()
|
|
muxRouter.HandleFunc("/", OldHandler)
|
|
mr := NewMutableRouter(muxRouter)
|
|
|
|
// start http server
|
|
log.Print("Start http server")
|
|
go startServer(mr)
|
|
|
|
// continuously make requests, panic if any fails
|
|
time.Sleep(100 * time.Millisecond)
|
|
q1 := make(chan bool)
|
|
go spamServer(q1)
|
|
q2 := make(chan bool)
|
|
go spamServer(q2)
|
|
q3 := make(chan bool)
|
|
go spamServer(q3)
|
|
|
|
time.Sleep(5 * time.Millisecond)
|
|
|
|
// connect and verify old handler
|
|
log.Print("Verify old handler")
|
|
verifyRequest("old handler")
|
|
|
|
// change the muxer
|
|
log.Print("Change mux router")
|
|
newMuxRouter := mux.NewRouter()
|
|
newMuxRouter.HandleFunc("/", NewHandler)
|
|
mr.UpdateRouter(newMuxRouter)
|
|
|
|
// connect and verify the new handler
|
|
log.Print("Verify new handler")
|
|
verifyRequest("new handler")
|
|
|
|
time.Sleep(5 * time.Millisecond)
|
|
q1 <- true
|
|
q2 <- true
|
|
q3 <- true
|
|
time.Sleep(100 * time.Millisecond)
|
|
}
|