Brendan Le Glaunec
76 lines
2.4 KiB
Go
76 lines
2.4 KiB
Go
package ui_test
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/Ullaakut/cameradar/v6"
|
|
"github.com/Ullaakut/cameradar/v6/internal/ui"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestPlainReporter_Outputs(t *testing.T) {
|
|
t.Run("prints events", func(t *testing.T) {
|
|
out := &bytes.Buffer{}
|
|
reporter := ui.NewPlainReporter(out, true)
|
|
|
|
reporter.Start(cameradar.StepScan, "starting")
|
|
reporter.Progress(cameradar.StepScan, "working")
|
|
reporter.Debug(cameradar.StepScan, "details")
|
|
reporter.Done(cameradar.StepScan, "finished")
|
|
reporter.Error(cameradar.StepScan, errors.New("boom"))
|
|
reporter.Summary([]cameradar.Stream{}, nil)
|
|
|
|
content := out.String()
|
|
assert.Contains(t, content, " [STEP] Scan targets: starting")
|
|
assert.Contains(t, content, " [INFO] Scan targets: working")
|
|
assert.Contains(t, content, " [DBUG] Scan targets: details")
|
|
assert.Contains(t, content, " [DONE] Scan targets: finished")
|
|
assert.Contains(t, content, " [EROR] Scan targets: boom")
|
|
assert.Contains(t, content, "Summary\n-------\nAccessible streams: 0")
|
|
})
|
|
|
|
t.Run("respects debug flag and empty input", func(t *testing.T) {
|
|
out := &bytes.Buffer{}
|
|
reporter := ui.NewPlainReporter(out, false)
|
|
|
|
reporter.Debug(cameradar.StepScan, "hidden")
|
|
reporter.Progress(cameradar.StepScan, "")
|
|
reporter.Error(cameradar.StepScan, nil)
|
|
|
|
content := out.String()
|
|
assert.NotContains(t, content, "DBUG")
|
|
assert.Equal(t, "", strings.TrimSpace(content))
|
|
})
|
|
}
|
|
|
|
func TestPlainReporter_PrintStartup(t *testing.T) {
|
|
t.Run("prints build info and options", func(t *testing.T) {
|
|
out := &bytes.Buffer{}
|
|
reporter := ui.NewPlainReporter(out, false)
|
|
|
|
reporter.PrintStartup(ui.BuildInfo{Version: "v1.2.3", Commit: "abcdefghi"}, []string{
|
|
"targets: 127.0.0.1",
|
|
"ports: 554",
|
|
})
|
|
|
|
content := out.String()
|
|
assert.Contains(t, content, " [INFO] Startup: Running cameradar version 1.2.3, commit abcdefg")
|
|
assert.Contains(t, content, " [INFO] Startup: targets: 127.0.0.1")
|
|
assert.Contains(t, content, " [INFO] Startup: ports: 554")
|
|
})
|
|
|
|
t.Run("prints only build info when options empty", func(t *testing.T) {
|
|
out := &bytes.Buffer{}
|
|
reporter := ui.NewPlainReporter(out, false)
|
|
|
|
reporter.PrintStartup(ui.BuildInfo{Version: "", Commit: "none"}, nil)
|
|
|
|
content := out.String()
|
|
assert.Contains(t, content, " [INFO] Startup: Running cameradar version dev, commit unknown")
|
|
assert.Equal(t, 1, strings.Count(content, " Startup: "))
|
|
})
|
|
}
|