55 lines
974 B
Go
55 lines
974 B
Go
|
package goadb
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"github.com/zach-klippenstein/goadb/wire"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestGetServerVersion(t *testing.T) {
|
||
|
|
client := &HostClient{mockDialer(&MockServer{
|
||
|
|
Status: wire.StatusSuccess,
|
||
|
|
Messages: []string{"000a"},
|
||
|
|
})}
|
||
|
|
|
||
|
|
v, err := client.GetServerVersion()
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, 10, v)
|
||
|
|
}
|
||
|
|
|
||
|
|
func mockDialer(s *MockServer) Dialer {
|
||
|
|
return func() (*wire.Conn, error) {
|
||
|
|
return &wire.Conn{s, s, s}, nil
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
type MockServer struct {
|
||
|
|
Status wire.StatusCode
|
||
|
|
Messages []string
|
||
|
|
|
||
|
|
nextMsgIndex int
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *MockServer) ReadStatus() (wire.StatusCode, error) {
|
||
|
|
return s.Status, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *MockServer) ReadMessage() ([]byte, error) {
|
||
|
|
if s.nextMsgIndex >= len(s.Messages) {
|
||
|
|
return nil, io.EOF
|
||
|
|
}
|
||
|
|
|
||
|
|
s.nextMsgIndex++
|
||
|
|
return []byte(s.Messages[s.nextMsgIndex-1]), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *MockServer) SendMessage(msg []byte) error {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *MockServer) Close() error {
|
||
|
|
return nil
|
||
|
|
}
|