package term // import "github.com/docker/docker/pkg/term" import ( "bytes" "testing" "gotest.tools/assert" is "gotest.tools/assert/cmp" ) func TestEscapeProxyRead(t *testing.T) { t.Run("no escape keys, keys a", func(t *testing.T) { escapeKeys, _ := ToBytes("") keys, _ := ToBytes("a") reader := NewEscapeProxy(bytes.NewReader(keys), escapeKeys) buf := make([]byte, len(keys)) nr, err := reader.Read(buf) assert.NilError(t, err) assert.Equal(t, nr, len(keys)) assert.DeepEqual(t, keys, buf) }) t.Run("no escape keys, keys a,b,c", func(t *testing.T) { escapeKeys, _ := ToBytes("") keys, _ := ToBytes("a,b,c") reader := NewEscapeProxy(bytes.NewReader(keys), escapeKeys) buf := make([]byte, len(keys)) nr, err := reader.Read(buf) assert.NilError(t, err) assert.Equal(t, nr, len(keys)) assert.DeepEqual(t, keys, buf) }) t.Run("no escape keys, no keys", func(t *testing.T) { escapeKeys, _ := ToBytes("") keys, _ := ToBytes("") reader := NewEscapeProxy(bytes.NewReader(keys), escapeKeys) buf := make([]byte, len(keys)) nr, err := reader.Read(buf) assert.Assert(t, is.ErrorContains(err, ""), "Should throw error when no keys are to read") assert.Equal(t, nr, 0) assert.Check(t, is.Len(keys, 0)) assert.Check(t, is.Len(buf, 0)) }) t.Run("DEL escape key, keys a,b,c,+", func(t *testing.T) { escapeKeys, _ := ToBytes("DEL") keys, _ := ToBytes("a,b,c,+") reader := NewEscapeProxy(bytes.NewReader(keys), escapeKeys) buf := make([]byte, len(keys)) nr, err := reader.Read(buf) assert.NilError(t, err) assert.Equal(t, nr, len(keys)) assert.DeepEqual(t, keys, buf) }) t.Run("DEL escape key, no keys", func(t *testing.T) { escapeKeys, _ := ToBytes("DEL") keys, _ := ToBytes("") reader := NewEscapeProxy(bytes.NewReader(keys), escapeKeys) buf := make([]byte, len(keys)) nr, err := reader.Read(buf) assert.Assert(t, is.ErrorContains(err, ""), "Should throw error when no keys are to read") assert.Equal(t, nr, 0) assert.Check(t, is.Len(keys, 0)) assert.Check(t, is.Len(buf, 0)) }) t.Run("ctrl-x,ctrl-@ escape key, keys DEL", func(t *testing.T) { escapeKeys, _ := ToBytes("ctrl-x,ctrl-@") keys, _ := ToBytes("DEL") reader := NewEscapeProxy(bytes.NewReader(keys), escapeKeys) buf := make([]byte, len(keys)) nr, err := reader.Read(buf) assert.NilError(t, err) assert.Equal(t, nr, 1) assert.DeepEqual(t, keys, buf) }) t.Run("ctrl-c escape key, keys ctrl-c", func(t *testing.T) { escapeKeys, _ := ToBytes("ctrl-c") keys, _ := ToBytes("ctrl-c") reader := NewEscapeProxy(bytes.NewReader(keys), escapeKeys) buf := make([]byte, len(keys)) nr, err := reader.Read(buf) assert.Error(t, err, "read escape sequence") assert.Equal(t, nr, 0) assert.DeepEqual(t, keys, buf) }) t.Run("ctrl-c,ctrl-z escape key, keys ctrl-c,ctrl-z", func(t *testing.T) { escapeKeys, _ := ToBytes("ctrl-c,ctrl-z") keys, _ := ToBytes("ctrl-c,ctrl-z") reader := NewEscapeProxy(bytes.NewReader(keys), escapeKeys) buf := make([]byte, 1) nr, err := reader.Read(buf) assert.NilError(t, err) assert.Equal(t, nr, 0) assert.DeepEqual(t, keys[0:1], buf) nr, err = reader.Read(buf) assert.Error(t, err, "read escape sequence") assert.Equal(t, nr, 0) assert.DeepEqual(t, keys[1:], buf) }) t.Run("ctrl-c,ctrl-z escape key, keys ctrl-c,DEL,+", func(t *testing.T) { escapeKeys, _ := ToBytes("ctrl-c,ctrl-z") keys, _ := ToBytes("ctrl-c,DEL,+") reader := NewEscapeProxy(bytes.NewReader(keys), escapeKeys) buf := make([]byte, 1) nr, err := reader.Read(buf) assert.NilError(t, err) assert.Equal(t, nr, 0) assert.DeepEqual(t, keys[0:1], buf) buf = make([]byte, len(keys)) nr, err = reader.Read(buf) assert.NilError(t, err) assert.Equal(t, nr, len(keys)) assert.DeepEqual(t, keys, buf) }) t.Run("ctrl-c,ctrl-z escape key, keys ctrl-c,DEL", func(t *testing.T) { escapeKeys, _ := ToBytes("ctrl-c,ctrl-z") keys, _ := ToBytes("ctrl-c,DEL") reader := NewEscapeProxy(bytes.NewReader(keys), escapeKeys) buf := make([]byte, 1) nr, err := reader.Read(buf) assert.NilError(t, err) assert.Equal(t, nr, 0) assert.DeepEqual(t, keys[0:1], buf) buf = make([]byte, len(keys)) nr, err = reader.Read(buf) assert.NilError(t, err) assert.Equal(t, nr, len(keys)) assert.DeepEqual(t, keys, buf) }) }