Add unit tests

2020-12-17 17:25:53 -08:00
parent 3702339f44
commit 6be4d69d02
705 changed files with 120529 additions and 150051 deletions
--- a/controller/consul_test.go
+++ b/controller/consul_test.go
@@ -0,0 +1,170 @@
+package controller
+
+import (
+	"bytes"
+	"io/ioutil"
+	"net/http"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+var mockConsulData = map[string]string{
+	"single-app": `{"Services": {
+		"test-app-1": {
+			"ID": "test-app-1",
+			"Service": "test-service",
+			"Tags": [
+				"enable_gocast", "gocast_vip=1.1.1.1/32", "gocast_monitor=consul"
+			]
+		}
+	}}`,
+	"single-app-no-match": `{"Services": {
+		"test-app-1": {
+			"ID": "test-app-1",
+			"Service": "test-service",
+			"Tags": [
+				"foo"
+			]
+		}
+	}}`,
+	"single-app-no-vip": `{"Services": {
+		"test-app-1": {
+			"ID": "test-app-1",
+			"Service": "test-service",
+			"Tags": [
+				"enable_gocast", "gocast_monitor=consul"
+			]
+		}
+	}}`,
+}
+
+var mockConsulCheckData = map[string]string{
+	"remote-pass": `[
+		{
+		"Node": "test-node1",
+		"Status": "passing",
+		"ServiceName": "test-service"
+		},
+		{
+		"Node": "test-node2",
+		"Status": "passing",
+		"ServiceName": "test-service"
+		}
+	]`,
+	"remote-fail": `[
+		{
+		"Node": "test-node1",
+		"Status": "failed",
+		"ServiceName": "test-service"
+		}
+	]`,
+	"local-pass": `{
+  		"service:test-service": {
+  		  "Node": "test-node1",
+  		  "Status": "passing",
+  		  "ServiceName": "test-service"
+  		}
+	}`,
+	"local-fail": `{
+  		"service:test-service": {
+  		  "Node": "test-node1",
+  		  "Status": "failed",
+  		  "ServiceName": "test-service"
+  		}
+	}`,
+}
+
+type MockClient struct {
+	get func(url string) (*http.Response, error)
+}
+
+func (c *MockClient) Get(url string) (*http.Response, error) {
+	if c.get != nil {
+		return c.get(url)
+	}
+	return nil, nil
+}
+
+func TestQueryServices(t *testing.T) {
+	a := assert.New(t)
+	client := &MockClient{}
+	cm := &ConsulMon{
+		addr: "foo", node: "test", client: client,
+	}
+
+	// test valid app
+	client.get = func(url string) (*http.Response, error) {
+		b := bytes.NewBuffer([]byte(mockConsulData["single-app"]))
+		return &http.Response{Body: ioutil.NopCloser(b), StatusCode: http.StatusOK}, nil
+	}
+	apps, err := cm.queryServices()
+	if err != nil {
+		a.FailNow(err.Error())
+	}
+	a.Equal(1, len(apps))
+	app, _ := NewApp("test-service", "1.1.1.1/32", []string{"consul"}, []string{}, "consul")
+	a.True(app.Equal(apps[0]))
+
+	// test no match
+	client.get = func(url string) (*http.Response, error) {
+		b := bytes.NewBuffer([]byte(mockConsulData["single-app-no-match"]))
+		return &http.Response{Body: ioutil.NopCloser(b), StatusCode: http.StatusOK}, nil
+	}
+	apps, err = cm.queryServices()
+	if err != nil {
+		a.FailNow(err.Error())
+	}
+	a.Equal(0, len(apps))
+
+	// test missing vip
+	client.get = func(url string) (*http.Response, error) {
+		b := bytes.NewBuffer([]byte(mockConsulData["single-app-no-vip"]))
+		return &http.Response{Body: ioutil.NopCloser(b), StatusCode: http.StatusOK}, nil
+	}
+	apps, _ = cm.queryServices()
+	a.Equal(0, len(apps))
+}
+
+func TestHealthCheck(t *testing.T) {
+	a := assert.New(t)
+	client := &MockClient{}
+	cm := &ConsulMon{node: "test-node1", client: client}
+
+	// test remote checks
+	cm.addr = "http://remote/check"
+	client.get = func(url string) (*http.Response, error) {
+		b := bytes.NewBuffer([]byte(mockConsulCheckData["remote-pass"]))
+		return &http.Response{Body: ioutil.NopCloser(b), StatusCode: http.StatusOK}, nil
+	}
+	check, err := cm.healthCheck("test-service")
+	if err != nil {
+		a.FailNow(err.Error())
+	}
+	a.True(check)
+	client.get = func(url string) (*http.Response, error) {
+		b := bytes.NewBuffer([]byte(mockConsulCheckData["remote-fail"]))
+		return &http.Response{Body: ioutil.NopCloser(b), StatusCode: http.StatusOK}, nil
+	}
+	check, _ = cm.healthCheck("test-service")
+	a.False(check)
+
+	// test local checks
+	cm.addr = "http://localhost/check"
+	client.get = func(url string) (*http.Response, error) {
+		b := bytes.NewBuffer([]byte(mockConsulCheckData["local-pass"]))
+		return &http.Response{Body: ioutil.NopCloser(b), StatusCode: http.StatusOK}, nil
+	}
+	check, _ = cm.healthCheck("test-service")
+	if err != nil {
+		a.FailNow(err.Error())
+	}
+	a.True(check)
+	cm.addr = "http://127.0.0.1/check"
+	client.get = func(url string) (*http.Response, error) {
+		b := bytes.NewBuffer([]byte(mockConsulCheckData["local-fail"]))
+		return &http.Response{Body: ioutil.NopCloser(b), StatusCode: http.StatusOK}, nil
+	}
+	check, _ = cm.healthCheck("test-service")
+	a.False(check)
+}