vbatts/pkg
1
0
Fork 0
Code Pull requests Releases Activity

Merge pull request #23151 from AkihiroSuda/fix23012PkgAuth

Browse source 
Fix racy tests in pkg/authorization
This commit is contained in:
Brian Goff 2016-06-01 20:45:54 -04:00
commit 83be58f17e
1 changed files with 13 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
authorization/authz_unix_test.go
View file

@ -28,7 +28,7 @@ const pluginAddress = "authzplugin.sock"
func TestAuthZRequestPluginError(t *testing.T) {
server := authZPluginTestServer{t: t}
go server.start()
server.start()
defer server.stop()
authZPlugin := createTestPlugin(t)
@ -59,7 +59,7 @@ func TestAuthZRequestPluginError(t *testing.T) {
func TestAuthZRequestPlugin(t *testing.T) {
server := authZPluginTestServer{t: t}
go server.start()
server.start()
defer server.stop()
authZPlugin := createTestPlugin(t)
@ -91,7 +91,7 @@ func TestAuthZRequestPlugin(t *testing.T) {
func TestAuthZResponsePlugin(t *testing.T) {
server := authZPluginTestServer{t: t}
go server.start()
server.start()
defer server.stop()
authZPlugin := createTestPlugin(t)
@ -223,6 +223,7 @@ type authZPluginTestServer struct {
recordedRequest Request
// response stores the response sent from the plugin to the daemon
replayResponse Response
server *httptest.Server
}
// start starts the test server that implements the plugin
@ -234,12 +235,19 @@ func (t *authZPluginTestServer) start() {
r.HandleFunc("/Plugin.Activate", t.activate)
r.HandleFunc("/"+AuthZApiRequest, t.auth)
r.HandleFunc("/"+AuthZApiResponse, t.auth)
server := http.Server{Handler: r, Addr: pluginAddress}
server.Serve(l)
t.server = &httptest.Server{
Listener: l,
Config: &http.Server{
Handler: r,
Addr: pluginAddress,
},
}
t.server.Start()
}
// stop stops the test server that implements the plugin
func (t *authZPluginTestServer) stop() {
t.server.Close()
os.Remove(pluginAddress)
if t.listener != nil {
t.listener.Close()