collector: add bounds check and test for tcpstat collector (#1134)

Signed-off-by: Matt Layher <mdlayher@gmail.com>
This commit is contained in:
Matt Layher 2018-10-27 03:21:36 -04:00 committed by Ben Kochie
commit 778124a56c
2 changed files with 25 additions and 0 deletions

View file

@ -15,9 +15,30 @@ package collector
import (
"os"
"strings"
"testing"
)
func Test_parseTCPStatsError(t *testing.T) {
tests := []struct {
name string
in string
}{
{
name: "too few fields",
in: "hello world",
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
if _, err := parseTCPStats(strings.NewReader(tt.in)); err == nil {
t.Fatal("expected an error, but none occurred")
}
})
}
}
func TestTCPStat(t *testing.T) {
file, err := os.Open("fixtures/proc/net/tcpstat")
if err != nil {