cb7e06bdb9
All sync status reads now also check for failure status, using the same logic as the normal Scanner (the only difference being that for normal scanners, length is encoded as a hex string, but for sync scanners it's in binary little-endian).
101 lines
2.8 KiB
Go
101 lines
2.8 KiB
Go
package goadb
|
|
|
|
import (
|
|
"io"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/zach-klippenstein/goadb/util"
|
|
"github.com/zach-klippenstein/goadb/wire"
|
|
)
|
|
|
|
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, "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 TestReadErrorNotFound(t *testing.T) {
|
|
s := wire.NewSyncScanner(strings.NewReader(
|
|
"FAIL\031\000\000\000No such file or directory"))
|
|
_, err := newSyncFileReader(s)
|
|
assert.True(t, util.HasErrCode(err, util.FileNoExistError))
|
|
assert.EqualError(t, err, "FileNoExistError: no such file or directory")
|
|
}
|