package jenkins import ( "encoding/json" "net/http" "net/http/httptest" "sort" "strings" "testing" "time" "github.com/influxdata/telegraf/internal" "github.com/influxdata/telegraf/testutil" ) func TestJobRequest(t *testing.T) { tests := []struct { input jobRequest output string }{ { jobRequest{}, "", }, { jobRequest{ name: "1", parents: []string{"3", "2"}, }, "3/2/1", }, } for _, test := range tests { output := test.input.hierarchyName() if output != test.output { t.Errorf("Expected %s, got %s\n", test.output, output) } } } func TestResultCode(t *testing.T) { tests := []struct { input string output int }{ {"SUCCESS", 0}, {"Failure", 1}, {"NOT_BUILT", 2}, {"UNSTABLE", 3}, {"ABORTED", 4}, } for _, test := range tests { output := mapResultCode(test.input) if output != test.output { t.Errorf("Expected %d, got %d\n", test.output, output) } } } type mockHandler struct { // responseMap is the path to repsonse interface // we will ouput the serialized response in json when serving http // example '/computer/api/json': *gojenkins. responseMap map[string]interface{} } func (h mockHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) { o, ok := h.responseMap[r.URL.Path] if !ok { w.WriteHeader(http.StatusNotFound) return } b, err := json.Marshal(o) if err != nil { w.WriteHeader(http.StatusInternalServerError) return } if len(b) == 0 { w.WriteHeader(http.StatusNoContent) return } w.Write(b) } func TestGatherNodeData(t *testing.T) { tests := []struct { name string input mockHandler output *testutil.Accumulator wantErr bool }{ { name: "bad node data", input: mockHandler{ responseMap: map[string]interface{}{ "/api/json": struct{}{}, "/computer/api/json": nodeResponse{ Computers: []node{ {}, {}, {}, }, }, }, }, wantErr: true, }, { name: "bad empty monitor data", input: mockHandler{ responseMap: map[string]interface{}{ "/api/json": struct{}{}, "/computer/api/json": nodeResponse{ Computers: []node{ {DisplayName: "master"}, {DisplayName: "node1"}, }, }, }, }, wantErr: true, }, { name: "filtered nodes", input: mockHandler{ responseMap: map[string]interface{}{ "/api/json": struct{}{}, "/computer/api/json": nodeResponse{ Computers: []node{ {DisplayName: "ignore-1"}, {DisplayName: "ignore-2"}, }, }, }, }, }, { name: "normal data collection", input: mockHandler{ responseMap: map[string]interface{}{ "/api/json": struct{}{}, "/computer/api/json": nodeResponse{ Computers: []node{ { DisplayName: "master", MonitorData: monitorData{ HudsonNodeMonitorsArchitectureMonitor: "linux", HudsonNodeMonitorsResponseTimeMonitor: struct { Average int64 `json:"average"` }{ Average: 10032, }, HudsonNodeMonitorsDiskSpaceMonitor: nodeSpaceMonitor{ Path: "/path/1", Size: 123, }, HudsonNodeMonitorsTemporarySpaceMonitor: nodeSpaceMonitor{ Path: "/path/2", Size: 245, }, HudsonNodeMonitorsSwapSpaceMonitor: struct { SwapAvailable float64 `json:"availableSwapSpace"` SwapTotal float64 `json:"totalSwapSpace"` MemoryAvailable float64 `json:"availablePhysicalMemory"` MemoryTotal float64 `json:"totalPhysicalMemory"` }{ SwapAvailable: 212, SwapTotal: 500, MemoryAvailable: 101, MemoryTotal: 500, }, }, Offline: false, }, }, }, }, }, output: &testutil.Accumulator{ Metrics: []*testutil.Metric{ { Tags: map[string]string{ "node_name": "master", "arch": "linux", "status": "online", "disk_path": "/path/1", "temp_path": "/path/2", }, Fields: map[string]interface{}{ "response_time": int64(10032), "disk_available": float64(123), "temp_available": float64(245), "swap_available": float64(212), "swap_total": float64(500), "memory_available": float64(101), "memory_total": float64(500), }, }, }, }, }, } for _, test := range tests { ts := httptest.NewServer(test.input) defer ts.Close() j := &Jenkins{ URL: ts.URL, ResponseTimeout: internal.Duration{Duration: time.Microsecond}, NodeExclude: []string{"ignore-1", "ignore-2"}, } te := j.initialize(&http.Client{Transport: &http.Transport{}}) acc := new(testutil.Accumulator) j.gatherNodesData(acc) if err := acc.FirstError(); err != nil { te = err } if !test.wantErr && te != nil { t.Fatalf("%s: failed %s, expected to be nil", test.name, te.Error()) } else if test.wantErr && te == nil { t.Fatalf("%s: expected err, got nil", test.name) } if test.output == nil && len(acc.Metrics) > 0 { t.Fatalf("%s: collected extra data", test.name) } else if test.output != nil && len(test.output.Metrics) > 0 { for k, m := range test.output.Metrics[0].Tags { if acc.Metrics[0].Tags[k] != m { t.Fatalf("%s: tag %s metrics unmatch Expected %s, got %s\n", test.name, k, m, acc.Metrics[0].Tags[k]) } } for k, m := range test.output.Metrics[0].Fields { if acc.Metrics[0].Fields[k] != m { t.Fatalf("%s: field %s metrics unmatch Expected %v(%T), got %v(%T)\n", test.name, k, m, m, acc.Metrics[0].Fields[k], acc.Metrics[0].Fields[k]) } } } } } func TestInitialize(t *testing.T) { mh := mockHandler{ responseMap: map[string]interface{}{ "/api/json": struct{}{}, }, } ts := httptest.NewServer(mh) defer ts.Close() mockClient := &http.Client{Transport: &http.Transport{}} tests := []struct { // name of the test name string input *Jenkins output *Jenkins wantErr bool }{ { name: "bad jenkins config", input: &Jenkins{ URL: "http://a bad url", ResponseTimeout: internal.Duration{Duration: time.Microsecond}, }, wantErr: true, }, { name: "has filter", input: &Jenkins{ URL: ts.URL, ResponseTimeout: internal.Duration{Duration: time.Microsecond}, JobExclude: []string{"job1", "job2"}, NodeExclude: []string{"node1", "node2"}, }, }, { name: "default config", input: &Jenkins{ URL: ts.URL, ResponseTimeout: internal.Duration{Duration: time.Microsecond}, }, output: &Jenkins{ MaxConnections: 5, MaxSubJobPerLayer: 10, }, }, } for _, test := range tests { te := test.input.initialize(mockClient) if !test.wantErr && te != nil { t.Fatalf("%s: failed %s, expected to be nil", test.name, te.Error()) } else if test.wantErr && te == nil { t.Fatalf("%s: expected err, got nil", test.name) } if test.output != nil { if test.input.client == nil { t.Fatalf("%s: failed %s, jenkins instance shouldn't be nil", test.name, te.Error()) } if test.input.MaxConnections != test.output.MaxConnections { t.Fatalf("%s: different MaxConnections Expected %d, got %d\n", test.name, test.output.MaxConnections, test.input.MaxConnections) } } } } func TestGatherJobs(t *testing.T) { tests := []struct { name string input mockHandler output *testutil.Accumulator wantErr bool }{ { name: "empty job", input: mockHandler{ responseMap: map[string]interface{}{ "/api/json": &jobResponse{}, }, }, }, { name: "bad inner jobs", input: mockHandler{ responseMap: map[string]interface{}{ "/api/json": &jobResponse{ Jobs: []innerJob{ {Name: "job1"}, }, }, }, }, wantErr: true, }, { name: "jobs has no build", input: mockHandler{ responseMap: map[string]interface{}{ "/api/json": &jobResponse{ Jobs: []innerJob{ {Name: "job1"}, }, }, "/job/job1/api/json": &jobResponse{}, }, }, }, { name: "bad build info", input: mockHandler{ responseMap: map[string]interface{}{ "/api/json": &jobResponse{ Jobs: []innerJob{ {Name: "job1"}, }, }, "/job/job1/api/json": &jobResponse{ LastBuild: jobBuild{ Number: 1, }, }, }, }, wantErr: true, }, { name: "ignore building job", input: mockHandler{ responseMap: map[string]interface{}{ "/api/json": &jobResponse{ Jobs: []innerJob{ {Name: "job1"}, }, }, "/job/job1/api/json": &jobResponse{ LastBuild: jobBuild{ Number: 1, }, }, "/job/job1/1/api/json": &buildResponse{ Building: true, }, }, }, }, { name: "ignore old build", input: mockHandler{ responseMap: map[string]interface{}{ "/api/json": &jobResponse{ Jobs: []innerJob{ {Name: "job1"}, }, }, "/job/job1/api/json": &jobResponse{ LastBuild: jobBuild{ Number: 2, }, }, "/job/job1/2/api/json": &buildResponse{ Building: false, Timestamp: 100, }, }, }, }, { name: "gather metrics", input: mockHandler{ responseMap: map[string]interface{}{ "/api/json": &jobResponse{ Jobs: []innerJob{ {Name: "job1"}, {Name: "job2"}, }, }, "/job/job1/api/json": &jobResponse{ LastBuild: jobBuild{ Number: 3, }, }, "/job/job2/api/json": &jobResponse{ LastBuild: jobBuild{ Number: 1, }, }, "/job/job1/3/api/json": &buildResponse{ Building: false, Result: "SUCCESS", Duration: 25558, Timestamp: (time.Now().Unix() - int64(time.Minute.Seconds())) * 1000, }, "/job/job2/1/api/json": &buildResponse{ Building: false, Result: "FAILURE", Duration: 1558, Timestamp: (time.Now().Unix() - int64(time.Minute.Seconds())) * 1000, }, }, }, output: &testutil.Accumulator{ Metrics: []*testutil.Metric{ { Tags: map[string]string{ "name": "job1", "result": "SUCCESS", }, Fields: map[string]interface{}{ "duration": int64(25558), "result_code": 0, }, }, { Tags: map[string]string{ "name": "job2", "result": "FAILURE", }, Fields: map[string]interface{}{ "duration": int64(1558), "result_code": 1, }, }, }, }, }, { name: "gather sub jobs, jobs filter", input: mockHandler{ responseMap: map[string]interface{}{ "/api/json": &jobResponse{ Jobs: []innerJob{ {Name: "apps"}, {Name: "ignore-1"}, }, }, "/job/apps/api/json": &jobResponse{ Jobs: []innerJob{ {Name: "k8s-cloud"}, {Name: "chronograf"}, {Name: "ignore-all"}, }, }, "/job/apps/job/ignore-all/api/json": &jobResponse{ Jobs: []innerJob{ {Name: "1"}, {Name: "2"}, }, }, "/job/apps/job/chronograf/api/json": &jobResponse{ LastBuild: jobBuild{ Number: 1, }, }, "/job/apps/job/k8s-cloud/api/json": &jobResponse{ Jobs: []innerJob{ {Name: "PR-100"}, {Name: "PR-101"}, {Name: "PR-ignore2"}, }, }, "/job/apps/job/k8s-cloud/job/PR-100/api/json": &jobResponse{ LastBuild: jobBuild{ Number: 1, }, }, "/job/apps/job/k8s-cloud/job/PR-101/api/json": &jobResponse{ LastBuild: jobBuild{ Number: 4, }, }, "/job/apps/job/chronograf/1/api/json": &buildResponse{ Building: false, Result: "FAILURE", Duration: 1558, Timestamp: (time.Now().Unix() - int64(time.Minute.Seconds())) * 1000, }, "/job/apps/job/k8s-cloud/job/PR-101/4/api/json": &buildResponse{ Building: false, Result: "SUCCESS", Duration: 76558, Timestamp: (time.Now().Unix() - int64(time.Minute.Seconds())) * 1000, }, "/job/apps/job/k8s-cloud/job/PR-100/1/api/json": &buildResponse{ Building: false, Result: "SUCCESS", Duration: 91558, Timestamp: (time.Now().Unix() - int64(time.Minute.Seconds())) * 1000, }, }, }, output: &testutil.Accumulator{ Metrics: []*testutil.Metric{ { Tags: map[string]string{ "name": "PR-100", "parents": "apps/k8s-cloud", "result": "SUCCESS", }, Fields: map[string]interface{}{ "duration": int64(91558), "result_code": 0, }, }, { Tags: map[string]string{ "name": "PR-101", "parents": "apps/k8s-cloud", "result": "SUCCESS", }, Fields: map[string]interface{}{ "duration": int64(76558), "result_code": 0, }, }, { Tags: map[string]string{ "name": "chronograf", "parents": "apps", "result": "FAILURE", }, Fields: map[string]interface{}{ "duration": int64(1558), "result_code": 1, }, }, }, }, }, } for _, test := range tests { ts := httptest.NewServer(test.input) defer ts.Close() j := &Jenkins{ URL: ts.URL, MaxBuildAge: internal.Duration{Duration: time.Hour}, ResponseTimeout: internal.Duration{Duration: time.Microsecond}, JobExclude: []string{ "ignore-1", "apps/ignore-all/*", "apps/k8s-cloud/PR-ignore2", }, } te := j.initialize(&http.Client{Transport: &http.Transport{}}) acc := new(testutil.Accumulator) acc.SetDebug(true) j.gatherJobs(acc) if err := acc.FirstError(); err != nil { te = err } if !test.wantErr && te != nil { t.Fatalf("%s: failed %s, expected to be nil", test.name, te.Error()) } else if test.wantErr && te == nil { t.Fatalf("%s: expected err, got nil", test.name) } if test.output != nil && len(test.output.Metrics) > 0 { // sort metrics sort.Slice(acc.Metrics, func(i, j int) bool { return strings.Compare(acc.Metrics[i].Tags["name"], acc.Metrics[j].Tags["name"]) < 0 }) for i := range test.output.Metrics { for k, m := range test.output.Metrics[i].Tags { if acc.Metrics[i].Tags[k] != m { t.Fatalf("%s: tag %s metrics unmatch Expected %s, got %s\n", test.name, k, m, acc.Metrics[i].Tags[k]) } } for k, m := range test.output.Metrics[i].Fields { if acc.Metrics[i].Fields[k] != m { t.Fatalf("%s: field %s metrics unmatch Expected %v(%T), got %v(%T)\n", test.name, k, m, m, acc.Metrics[i].Fields[k], acc.Metrics[0].Fields[k]) } } } } } }