X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/47b2e4988e9bfb69ef00de1c900a20f714af5f2f..98d6c8c5743e0fd6be85af3b9f30286a358bd1d4:/services/ws/server_test.go

diff --git a/services/ws/server_test.go b/services/ws/server_test.go
index da7440d0e0..b1f943857a 100644
--- a/services/ws/server_test.go
+++ b/services/ws/server_test.go
@@ -1,6 +1,11 @@
+// Copyright (C) The Arvados Authors. All rights reserved.
+//
+// SPDX-License-Identifier: AGPL-3.0
+
 package main
 
 import (
+	"encoding/json"
 	"io/ioutil"
 	"net/http"
 	"sync"
@@ -65,6 +70,7 @@ func (s *serverSuite) TestBadDB(c *check.C) {
 
 func (s *serverSuite) TestHealth(c *check.C) {
 	go s.srv.Run()
+	defer s.srv.Close()
 	s.srv.WaitReady()
 	for _, token := range []string{"", "foo", s.cfg.ManagementToken} {
 		req, err := http.NewRequest("GET", "http://"+s.srv.listener.Addr().String()+"/_health/ping", nil)
@@ -84,3 +90,35 @@ func (s *serverSuite) TestHealth(c *check.C) {
 		}
 	}
 }
+
+func (s *serverSuite) TestStatus(c *check.C) {
+	go s.srv.Run()
+	defer s.srv.Close()
+	s.srv.WaitReady()
+	req, err := http.NewRequest("GET", "http://"+s.srv.listener.Addr().String()+"/status.json", nil)
+	c.Assert(err, check.IsNil)
+	resp, err := http.DefaultClient.Do(req)
+	c.Check(err, check.IsNil)
+	c.Check(resp.StatusCode, check.Equals, http.StatusOK)
+	var status map[string]interface{}
+	err = json.NewDecoder(resp.Body).Decode(&status)
+	c.Check(err, check.IsNil)
+	c.Check(status["Version"], check.Not(check.Equals), "")
+}
+
+func (s *serverSuite) TestHealthDisabled(c *check.C) {
+	s.cfg.ManagementToken = ""
+
+	go s.srv.Run()
+	defer s.srv.Close()
+	s.srv.WaitReady()
+
+	for _, token := range []string{"", "foo", arvadostest.ManagementToken} {
+		req, err := http.NewRequest("GET", "http://"+s.srv.listener.Addr().String()+"/_health/ping", nil)
+		c.Assert(err, check.IsNil)
+		req.Header.Add("Authorization", "Bearer "+token)
+		resp, err := http.DefaultClient.Do(req)
+		c.Check(err, check.IsNil)
+		c.Check(resp.StatusCode, check.Equals, http.StatusNotFound)
+	}
+}