summaryrefslogtreecommitdiff
path: root/service/health/service_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'service/health/service_test.go')
-rw-r--r--service/health/service_test.go279
1 files changed, 279 insertions, 0 deletions
diff --git a/service/health/service_test.go b/service/health/service_test.go
new file mode 100644
index 00000000..3e86b0bd
--- /dev/null
+++ b/service/health/service_test.go
@@ -0,0 +1,279 @@
+package health
+
+import (
+ "encoding/json"
+ "io/ioutil"
+ "net/http"
+ "testing"
+ "time"
+
+ "github.com/sirupsen/logrus"
+ "github.com/sirupsen/logrus/hooks/test"
+ "github.com/spiral/roadrunner/service"
+ rrhttp "github.com/spiral/roadrunner/service/http"
+ "github.com/stretchr/testify/assert"
+)
+
+type testCfg struct {
+ healthCfg string
+ httpCfg string
+ target string
+}
+
+func (cfg *testCfg) Get(name string) service.Config {
+ if name == ID {
+ return &testCfg{target: cfg.healthCfg}
+ }
+
+ if name == rrhttp.ID {
+ return &testCfg{target: cfg.httpCfg}
+ }
+
+ return nil
+}
+
+func (cfg *testCfg) Unmarshal(out interface{}) error {
+ err := json.Unmarshal([]byte(cfg.target), out)
+ return err
+}
+
+func TestService_Serve(t *testing.T) {
+ logger, _ := test.NewNullLogger()
+ logger.SetLevel(logrus.DebugLevel)
+
+ c := service.NewContainer(logger)
+ c.Register(rrhttp.ID, &rrhttp.Service{})
+ c.Register(ID, &Service{})
+
+ assert.NoError(t, c.Init(&testCfg{
+ healthCfg: `{
+ "address": "localhost:2116"
+ }`,
+ httpCfg: `{
+ "address": "localhost:2115",
+ "workers":{
+ "command": "php ../../tests/http/client.php echo pipes",
+ "pool": {"numWorkers": 1}
+ }
+ }`,
+ }))
+
+ s, status := c.Get(ID)
+ assert.NotNil(t, s)
+ assert.Equal(t, service.StatusOK, status)
+
+ hS, httpStatus := c.Get(rrhttp.ID)
+ assert.NotNil(t, hS)
+ assert.Equal(t, service.StatusOK, httpStatus)
+
+ go func() { c.Serve() }()
+ time.Sleep(time.Millisecond * 500)
+ defer c.Stop()
+
+ _, res, err := get("http://localhost:2116/")
+ assert.NoError(t, err)
+ assert.Equal(t, http.StatusOK, res.StatusCode)
+
+}
+
+func TestService_Serve_DeadWorker(t *testing.T) {
+ logger, _ := test.NewNullLogger()
+ logger.SetLevel(logrus.DebugLevel)
+
+ c := service.NewContainer(logger)
+ c.Register(rrhttp.ID, &rrhttp.Service{})
+ c.Register(ID, &Service{})
+
+ assert.NoError(t, c.Init(&testCfg{
+ healthCfg: `{
+ "address": "localhost:2116"
+ }`,
+ httpCfg: `{
+ "address": "localhost:2115",
+ "workers":{
+ "command": "php ../../tests/http/client.php echo pipes",
+ "pool": {"numWorkers": 1}
+ }
+ }`,
+ }))
+
+ s, status := c.Get(ID)
+ assert.NotNil(t, s)
+ assert.Equal(t, service.StatusOK, status)
+
+ hS, httpStatus := c.Get(rrhttp.ID)
+ assert.NotNil(t, hS)
+ assert.Equal(t, service.StatusOK, httpStatus)
+
+ go func() { c.Serve() }()
+ time.Sleep(time.Millisecond * 500)
+ defer c.Stop()
+
+ // Kill the worker
+ httpSvc := hS.(*rrhttp.Service)
+ httpSvc.Server().Workers()[0].Kill()
+
+ // Check health check
+ _, res, err := get("http://localhost:2116/")
+ assert.NoError(t, err)
+ assert.Equal(t, http.StatusInternalServerError, res.StatusCode)
+}
+
+func TestService_Serve_DeadWorkerStillHealthy(t *testing.T) {
+ logger, _ := test.NewNullLogger()
+ logger.SetLevel(logrus.DebugLevel)
+
+ c := service.NewContainer(logger)
+ c.Register(rrhttp.ID, &rrhttp.Service{})
+ c.Register(ID, &Service{})
+
+ assert.NoError(t, c.Init(&testCfg{
+ healthCfg: `{
+ "address": "localhost:2116"
+ }`,
+ httpCfg: `{
+ "address": "localhost:2115",
+ "workers":{
+ "command": "php ../../tests/http/client.php echo pipes",
+ "pool": {"numWorkers": 3}
+ }
+ }`,
+ }))
+
+ s, status := c.Get(ID)
+ assert.NotNil(t, s)
+ assert.Equal(t, service.StatusOK, status)
+
+ hS, httpStatus := c.Get(rrhttp.ID)
+ assert.NotNil(t, hS)
+ assert.Equal(t, service.StatusOK, httpStatus)
+
+ go func() { c.Serve() }()
+ time.Sleep(time.Second * 1)
+ defer c.Stop()
+
+ // Kill one of the workers
+ httpSvc := hS.(*rrhttp.Service)
+ httpSvc.Server().Workers()[0].Kill()
+
+ // Check health check
+ _, res, err := get("http://localhost:2116/")
+ assert.NoError(t, err)
+ assert.Equal(t, http.StatusOK, res.StatusCode)
+}
+
+func TestService_Serve_NoHTTPService(t *testing.T) {
+ logger, _ := test.NewNullLogger()
+ logger.SetLevel(logrus.DebugLevel)
+
+ c := service.NewContainer(logger)
+ c.Register(ID, &Service{})
+
+ assert.NoError(t, c.Init(&testCfg{
+ healthCfg: `{
+ "address": "localhost:2116"
+ }`,
+ }))
+
+ s, status := c.Get(ID)
+ assert.NotNil(t, s)
+ assert.Equal(t, service.StatusInactive, status)
+}
+
+func TestService_Serve_NoServer(t *testing.T) {
+ logger, _ := test.NewNullLogger()
+ logger.SetLevel(logrus.DebugLevel)
+
+ healthSvc := &Service{}
+
+ c := service.NewContainer(logger)
+ c.Register(rrhttp.ID, &rrhttp.Service{})
+ c.Register(ID, healthSvc)
+
+ assert.NoError(t, c.Init(&testCfg{
+ healthCfg: `{
+ "address": "localhost:2116"
+ }`,
+ httpCfg: `{
+ "address": "localhost:2115",
+ "workers":{
+ "command": "php ../../tests/http/client.php echo pipes",
+ "pool": {"numWorkers": 1}
+ }
+ }`,
+ }))
+
+ s, status := c.Get(ID)
+ assert.NotNil(t, s)
+ assert.Equal(t, service.StatusOK, status)
+
+ hS, httpStatus := c.Get(rrhttp.ID)
+ assert.NotNil(t, hS)
+ assert.Equal(t, service.StatusOK, httpStatus)
+
+ go func() { c.Serve() }()
+ time.Sleep(time.Millisecond * 500)
+ defer c.Stop()
+
+ // Set the httpService to nil
+ healthSvc.httpService = nil
+
+ _, res, err := get("http://localhost:2116/")
+ assert.NoError(t, err)
+ assert.Equal(t, http.StatusInternalServerError, res.StatusCode)
+}
+
+func TestService_Serve_NoPool(t *testing.T) {
+ logger, _ := test.NewNullLogger()
+ logger.SetLevel(logrus.DebugLevel)
+
+ httpSvc := &rrhttp.Service{}
+
+ c := service.NewContainer(logger)
+ c.Register(rrhttp.ID, httpSvc)
+ c.Register(ID, &Service{})
+
+ assert.NoError(t, c.Init(&testCfg{
+ healthCfg: `{
+ "address": "localhost:2116"
+ }`,
+ httpCfg: `{
+ "address": "localhost:2115",
+ "workers":{
+ "command": "php ../../tests/http/client.php echo pipes",
+ "pool": {"numWorkers": 1}
+ }
+ }`,
+ }))
+
+ s, status := c.Get(ID)
+ assert.NotNil(t, s)
+ assert.Equal(t, service.StatusOK, status)
+
+ hS, httpStatus := c.Get(rrhttp.ID)
+ assert.NotNil(t, hS)
+ assert.Equal(t, service.StatusOK, httpStatus)
+
+ go func() { c.Serve() }()
+ time.Sleep(time.Millisecond * 500)
+ defer c.Stop()
+
+ // Stop the pool
+ httpSvc.Server().Stop()
+
+ _, res, err := get("http://localhost:2116/")
+ assert.NoError(t, err)
+ assert.Equal(t, http.StatusInternalServerError, res.StatusCode)
+}
+
+// get request and return body
+func get(url string) (string, *http.Response, error) {
+ r, err := http.Get(url)
+ if err != nil {
+ return "", nil, err
+ }
+ defer r.Body.Close()
+
+ b, err := ioutil.ReadAll(r.Body)
+ return string(b), r, err
+}
generated by cgit v1.2.3 (git 2.25.1) at 2025-01-29 20:18:49 +0000