116 lines
3.1 KiB
Go
116 lines
3.1 KiB
Go
package adb
|
|
|
|
import (
|
|
"io"
|
|
"io/ioutil"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/timoxa0/goadb/wire"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestReadNextChunk(t *testing.T) {
|
|
s := wire.NewSyncScanner(strings.NewReader(
|
|
"DATA\006\000\000\000hello DATA\005\000\000\000worldDONE"))
|
|
|
|
// Read 1st chunk
|
|
reader, err := readNextChunk(s)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, int64(6), reader.(*io.LimitedReader).N)
|
|
buf := make([]byte, 10)
|
|
n, err := reader.Read(buf)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, 6, n)
|
|
assert.Equal(t, "hello ", string(buf[:6]))
|
|
|
|
// Read 2nd chunk
|
|
reader, err = readNextChunk(s)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, int64(5), reader.(*io.LimitedReader).N)
|
|
buf = make([]byte, 10)
|
|
n, err = reader.Read(buf)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, 5, n)
|
|
assert.Equal(t, "world", string(buf[:5]))
|
|
|
|
// Read DONE
|
|
_, err = readNextChunk(s)
|
|
assert.Equal(t, io.EOF, err)
|
|
}
|
|
func TestReadNextChunkInvalidChunkId(t *testing.T) {
|
|
s := wire.NewSyncScanner(strings.NewReader(
|
|
"ATAD\006\000\000\000hello "))
|
|
|
|
// Read 1st chunk
|
|
_, err := readNextChunk(s)
|
|
assert.EqualError(t, err, "AssertionError: expected chunk id 'DATA' or 'DONE', but got 'ATAD'")
|
|
}
|
|
|
|
func TestReadMultipleCalls(t *testing.T) {
|
|
s := wire.NewSyncScanner(strings.NewReader(
|
|
"DATA\006\000\000\000hello DATA\005\000\000\000worldDONE"))
|
|
reader, err := newSyncFileReader(s)
|
|
assert.NoError(t, err)
|
|
|
|
firstByte := make([]byte, 1)
|
|
_, err = io.ReadFull(reader, firstByte)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "h", string(firstByte))
|
|
|
|
restFirstChunkBytes := make([]byte, 5)
|
|
_, err = io.ReadFull(reader, restFirstChunkBytes)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "ello ", string(restFirstChunkBytes))
|
|
|
|
secondChunkBytes := make([]byte, 5)
|
|
_, err = io.ReadFull(reader, secondChunkBytes)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "world", string(secondChunkBytes))
|
|
|
|
_, err = io.ReadFull(reader, make([]byte, 5))
|
|
assert.Equal(t, io.EOF, err)
|
|
}
|
|
|
|
func TestReadAll(t *testing.T) {
|
|
s := wire.NewSyncScanner(strings.NewReader(
|
|
"DATA\006\000\000\000hello DATA\005\000\000\000worldDONE"))
|
|
reader, err := newSyncFileReader(s)
|
|
assert.NoError(t, err)
|
|
|
|
buf := make([]byte, 20)
|
|
_, err = io.ReadFull(reader, buf)
|
|
assert.Equal(t, io.ErrUnexpectedEOF, err)
|
|
assert.Equal(t, "hello world\000", string(buf[:12]))
|
|
}
|
|
|
|
func TestReadError(t *testing.T) {
|
|
s := wire.NewSyncScanner(strings.NewReader(
|
|
"FAIL\004\000\000\000fail"))
|
|
_, err := newSyncFileReader(s)
|
|
assert.EqualError(t, err, "AdbError: server error for read-chunk request: fail ({Request:read-chunk ServerMsg:fail})")
|
|
}
|
|
|
|
func TestReadEmpty(t *testing.T) {
|
|
s := wire.NewSyncScanner(strings.NewReader(
|
|
"DONE"))
|
|
r, err := newSyncFileReader(s)
|
|
assert.NoError(t, err)
|
|
|
|
// Multiple read calls that return EOF is a valid case.
|
|
for i := 0; i < 5; i++ {
|
|
data, err := ioutil.ReadAll(r)
|
|
assert.NoError(t, err)
|
|
assert.Empty(t, data)
|
|
}
|
|
}
|
|
|
|
func TestReadErrorNotFound(t *testing.T) {
|
|
s := wire.NewSyncScanner(strings.NewReader(
|
|
"FAIL\031\000\000\000No such file or directory"))
|
|
_, err := newSyncFileReader(s)
|
|
assert.True(t, HasErrCode(err, FileNoExistError))
|
|
assert.EqualError(t, err, "FileNoExistError: no such file or directory")
|
|
}
|