forked from elastic/beats
-
Notifications
You must be signed in to change notification settings - Fork 0
/
tcp_test.go
125 lines (105 loc) · 3.17 KB
/
tcp_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
// Licensed to Elasticsearch B.V. under one or more contributor
// license agreements. See the NOTICE file distributed with
// this work for additional information regarding copyright
// ownership. Elasticsearch B.V. licenses this file to you under
// the Apache License, Version 2.0 (the "License"); you may
// not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied. See the License for the
// specific language governing permissions and limitations
// under the License.
package tcp
import (
"fmt"
"net/http/httptest"
"testing"
"github.com/stretchr/testify/require"
"net/http"
"github.com/elastic/beats/heartbeat/hbtest"
"github.com/elastic/beats/heartbeat/monitors"
"github.com/elastic/beats/libbeat/beat"
"github.com/elastic/beats/libbeat/common"
"github.com/elastic/beats/libbeat/common/mapval"
btesting "github.com/elastic/beats/libbeat/testing"
"github.com/elastic/beats/libbeat/testing/mapvaltest"
)
func testTCPCheck(t *testing.T, host string, port uint16) *beat.Event {
config := common.NewConfig()
config.SetString("hosts", 0, host)
config.SetInt("ports", 0, int64(port))
jobs, err := create(monitors.Info{}, config)
require.NoError(t, err)
job := jobs[0]
event, _, err := job.Run()
require.NoError(t, err)
return &event
}
func tcpMonitorChecks(host string, ip string, port uint16, status string) mapval.Validator {
id := fmt.Sprintf("tcp-tcp@%s:%d", host, port)
return hbtest.MonitorChecks(id, host, ip, "tcp", status)
}
func TestUpEndpointJob(t *testing.T) {
server := httptest.NewServer(hbtest.HelloWorldHandler(http.StatusOK))
defer server.Close()
port, err := hbtest.ServerPort(server)
require.NoError(t, err)
event := testTCPCheck(t, "localhost", port)
mapvaltest.Test(
t,
mapval.Strict(mapval.Compose(
hbtest.MonitorChecks(
fmt.Sprintf("tcp-tcp@localhost:%d", port),
"localhost",
"127.0.0.1",
"tcp",
"up",
),
hbtest.RespondingTCPChecks(port),
mapval.MustCompile(mapval.Map{
"resolve": mapval.Map{
"host": "localhost",
"ip": "127.0.0.1",
"rtt.us": mapval.IsDuration,
},
}),
)),
event.Fields,
)
}
func TestConnectionRefusedEndpointJob(t *testing.T) {
ip := "127.0.0.1"
port, err := btesting.AvailableTCP4Port()
require.NoError(t, err)
event := testTCPCheck(t, ip, port)
dialErr := fmt.Sprintf("dial tcp %s:%d", ip, port)
mapvaltest.Test(
t,
mapval.Strict(mapval.Compose(
tcpMonitorChecks(ip, ip, port, "down"),
hbtest.ErrorChecks(dialErr, "io"),
hbtest.TCPBaseChecks(port),
)),
event.Fields,
)
}
func TestUnreachableEndpointJob(t *testing.T) {
ip := "203.0.113.1"
port := uint16(1234)
event := testTCPCheck(t, ip, port)
dialErr := fmt.Sprintf("dial tcp %s:%d", ip, port)
mapvaltest.Test(
t,
mapval.Strict(mapval.Compose(
tcpMonitorChecks(ip, ip, port, "down"),
hbtest.ErrorChecks(dialErr, "io"),
hbtest.TCPBaseChecks(port),
)),
event.Fields,
)
}