Skip to content
4 changes: 2 additions & 2 deletions examples/check_example/main_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -20,14 +20,14 @@ func TestMyMain(t *testing.T) {
expected = "[OK] - value is 10"

if !strings.Contains(actual, expected) {
t.Fatalf("expected %v, got %v", expected, actual)
t.Fatalf("expected %q, got %q", expected, actual)
}

actual = testhelper.RunMainTest(main, "--warning", "10", "--value", "11")
expected = "[WARNING] - value is 11"

if !strings.Contains(actual, expected) {
t.Fatalf("expected %v, got %v", expected, actual)
t.Fatalf("expected %q\n, got %q", expected, actual)
}
}

Expand Down
19 changes: 6 additions & 13 deletions examples/check_example2/main.go
Original file line number Diff line number Diff line change
Expand Up @@ -11,28 +11,22 @@ func main() {

var overall result.Overall

check1 := result.PartialResult{}
check1 := result.NewPartialResult()

check1.Output = "Check1"
err := check1.SetState(check.OK)

if err != nil {
check.ExitError(err)
}
check1.SetState(check.OK)

check1.Perfdata.Add(&perfdata.Perfdata{
Label: "foo",
Value: 23,
})

check2 := result.PartialResult{}
overall.AddSubcheck(check1)

check2.Output = "Check2"
err = check2.SetState(check.Warning)
check2 := result.NewPartialResult()

if err != nil {
check.ExitError(err)
}
check2.Output = "Check2"
check2.SetState(check.Warning)

check2.Perfdata.Add(&perfdata.Perfdata{
Label: "bar",
Expand All @@ -43,7 +37,6 @@ func main() {
Value: 46,
})

overall.AddSubcheck(check1)
overall.AddSubcheck(check2)

check.Exit(overall.GetStatus(), overall.GetOutput())
Expand Down
4 changes: 2 additions & 2 deletions examples/check_example2/main_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ import (
func TestMyMain(t *testing.T) {
actual := testhelper.RunMainTest(main)

expected := `[WARNING] - states: warning=1 ok=1
expected := `[WARNING] - Check2
\_ [OK] Check1
\_ [WARNING] Check2
|foo=23 bar=42 'foo2 bar'=46
Expand All @@ -19,7 +19,7 @@ would exit with code 1
`

if actual != expected {
t.Fatalf("expected %v, got %v", expected, actual)
t.Fatalf("expected %q\n, got %q", expected, actual)
}
}

Expand Down
Loading
Loading