Commit 412ed17c authored by Ash McKenzie's avatar Ash McKenzie Committed by Igor Drozdov

New console package for writing to the console

parent 629e3bf9
package console
import (
"fmt"
"io"
"strings"
)
func DisplayWarningMessage(message string, out io.Writer) {
DisplayWarningMessages([]string{message}, out)
}
func DisplayInfoMessage(message string, out io.Writer) {
DisplayInfoMessages([]string{message}, out)
}
func DisplayWarningMessages(messages []string, out io.Writer) {
DisplayMessages(messages, out, true)
}
func DisplayInfoMessages(messages []string, out io.Writer) {
DisplayMessages(messages, out, false)
}
func DisplayMessages(messages []string, out io.Writer, displayDivider bool) {
if noMessages(messages) {
return
}
displayBlankLineOrDivider(out, displayDivider)
for _, msg := range messages {
fmt.Fprintf(out, formatLine(msg))
}
displayBlankLineOrDivider(out, displayDivider)
}
func noMessages(messages []string) bool {
if len(messages) == 0 {
return true
}
for _, msg := range messages {
if len(strings.TrimSpace(msg)) > 0 {
return false
}
}
return true
}
func formatLine(message string) string {
return fmt.Sprintf("remote: %v\n", message)
}
func displayBlankLineOrDivider(out io.Writer, displayDivider bool) {
if displayDivider {
fmt.Fprintf(out, divider())
} else {
fmt.Fprintf(out, blankLine())
}
}
func blankLine() string {
return formatLine("")
}
func divider() string {
ruler := strings.Repeat("=", 72)
return fmt.Sprintf("%v%v%v", blankLine(), formatLine(ruler), blankLine())
}
package console
import (
"bytes"
"testing"
"github.com/stretchr/testify/require"
)
func TestDisplayWarningMessage(t *testing.T) {
tests := []struct {
name string
message string
wantOut string
}{
{
name: "empty",
message: "",
wantOut: "",
},
{
name: "basically empty",
message: " ",
wantOut: "",
},
{
name: "something",
message: "something",
wantOut: `remote:
remote: ========================================================================
remote:
remote: something
remote:
remote: ========================================================================
remote:
`,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
out := &bytes.Buffer{}
DisplayWarningMessage(tt.message, out)
require.Equal(t, tt.wantOut, out.String())
})
}
}
func TestDisplayWarningMessages(t *testing.T) {
tests := []struct {
name string
messages []string
wantOut string
}{
{
name: "empty",
messages: []string{""},
wantOut: "",
},
{
name: "basically empty",
messages: []string{" "},
wantOut: "",
},
{
name: "something",
messages: []string{"something", "here"},
wantOut: `remote:
remote: ========================================================================
remote:
remote: something
remote: here
remote:
remote: ========================================================================
remote:
`,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
out := &bytes.Buffer{}
DisplayWarningMessages(tt.messages, out)
require.Equal(t, tt.wantOut, out.String())
})
}
}
func TestDisplayInfoMessage(t *testing.T) {
tests := []struct {
name string
message string
wantOut string
}{
{
name: "empty",
message: "",
wantOut: "",
},
{
name: "basically empty",
message: " ",
wantOut: "",
},
{
name: "something",
message: "something",
wantOut: `remote:
remote: something
remote:
`,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
out := &bytes.Buffer{}
DisplayInfoMessage(tt.message, out)
require.Equal(t, tt.wantOut, out.String())
})
}
}
func TestDisplayInfoMessages(t *testing.T) {
tests := []struct {
name string
messages []string
wantOut string
}{
{
name: "empty",
messages: []string{""},
wantOut: "",
},
{
name: "basically empty",
messages: []string{" "},
wantOut: "",
},
{
name: "something",
messages: []string{"something", "here"},
wantOut: "remote: \nremote: something\nremote: here\nremote: \n",
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
out := &bytes.Buffer{}
DisplayInfoMessages(tt.messages, out)
require.Equal(t, tt.wantOut, out.String())
})
}
}
func Test_noMessages(t *testing.T) {
tests := []struct {
name string
messages []string
want bool
}{
{
name: "empty",
messages: []string{""},
want: true,
},
{
name: "basically empty",
messages: []string{" "},
want: true,
},
{
name: "something",
messages: []string{"something", "here"},
want: false,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
require.Equal(t, tt.want, noMessages(tt.messages))
})
}
}
func Test_formatLine(t *testing.T) {
require.Equal(t, "remote: something\n", formatLine("something"))
}
func Test_divider(t *testing.T) {
want := `remote:
remote: ========================================================================
remote:
`
require.Equal(t, want, divider())
}
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment