1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283 |
- package streamformatter
- import (
- "encoding/json"
- "errors"
- "strings"
- "testing"
- "github.com/docker/docker/pkg/jsonmessage"
- "github.com/stretchr/testify/assert"
- "github.com/stretchr/testify/require"
- )
- func TestRawProgressFormatterFormatStatus(t *testing.T) {
- sf := rawProgressFormatter{}
- res := sf.formatStatus("ID", "%s%d", "a", 1)
- assert.Equal(t, "a1\r\n", string(res))
- }
- func TestRawProgressFormatterFormatProgress(t *testing.T) {
- sf := rawProgressFormatter{}
- progress := &jsonmessage.JSONProgress{
- Current: 15,
- Total: 30,
- Start: 1,
- }
- res := sf.formatProgress("id", "action", progress, nil)
- out := string(res)
- assert.True(t, strings.HasPrefix(out, "action [===="))
- assert.Contains(t, out, "15B/30B")
- assert.True(t, strings.HasSuffix(out, "\r"))
- }
- func TestFormatStatus(t *testing.T) {
- res := FormatStatus("ID", "%s%d", "a", 1)
- expected := `{"status":"a1","id":"ID"}` + streamNewline
- assert.Equal(t, expected, string(res))
- }
- func TestFormatError(t *testing.T) {
- res := FormatError(errors.New("Error for formatter"))
- expected := `{"errorDetail":{"message":"Error for formatter"},"error":"Error for formatter"}` + "\r\n"
- assert.Equal(t, expected, string(res))
- }
- func TestFormatJSONError(t *testing.T) {
- err := &jsonmessage.JSONError{Code: 50, Message: "Json error"}
- res := FormatError(err)
- expected := `{"errorDetail":{"code":50,"message":"Json error"},"error":"Json error"}` + streamNewline
- assert.Equal(t, expected, string(res))
- }
- func TestJsonProgressFormatterFormatProgress(t *testing.T) {
- sf := &jsonProgressFormatter{}
- progress := &jsonmessage.JSONProgress{
- Current: 15,
- Total: 30,
- Start: 1,
- }
- res := sf.formatProgress("id", "action", progress, nil)
- msg := &jsonmessage.JSONMessage{}
- require.NoError(t, json.Unmarshal(res, msg))
- assert.Equal(t, "id", msg.ID)
- assert.Equal(t, "action", msg.Status)
- // The progress will always be in the format of:
- // [=========================> ] 15B/30B 412910h51m30s
- // The last entry '404933h7m11s' is the timeLeftBox.
- // However, the timeLeftBox field may change as progress.String() depends on time.Now().
- // Therefore, we have to strip the timeLeftBox from the strings to do the comparison.
- // Compare the progress strings before the timeLeftBox
- expectedProgress := "[=========================> ] 15B/30B"
- // if terminal column is <= 110, expectedProgressShort is expected.
- expectedProgressShort := " 15B/30B"
- if !(strings.HasPrefix(msg.ProgressMessage, expectedProgress) ||
- strings.HasPrefix(msg.ProgressMessage, expectedProgressShort)) {
- t.Fatalf("ProgressMessage without the timeLeftBox must be %s or %s, got: %s",
- expectedProgress, expectedProgressShort, msg.ProgressMessage)
- }
- assert.Equal(t, progress, msg.Progress)
- }
|