goadb/device_watcher_test.go

package goadb

import (
	"log"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/zach-klippenstein/goadb/util"
	"github.com/zach-klippenstein/goadb/wire"
)

func TestParseDeviceStatesSingle(t *testing.T) {
	states, err := parseDeviceStates(`192.168.56.101:5555	emulator-state
`)

	assert.NoError(t, err)
	assert.Len(t, states, 1)
	assert.Equal(t, "emulator-state", states["192.168.56.101:5555"])
}

func TestParseDeviceStatesMultiple(t *testing.T) {
	states, err := parseDeviceStates(`192.168.56.101:5555	emulator-state
0x0x0x0x	usb-state
`)

	assert.NoError(t, err)
	assert.Len(t, states, 2)
	assert.Equal(t, "emulator-state", states["192.168.56.101:5555"])
	assert.Equal(t, "usb-state", states["0x0x0x0x"])
}

func TestParseDeviceStatesMalformed(t *testing.T) {
	_, err := parseDeviceStates(`192.168.56.101:5555	emulator-state
0x0x0x0x
`)

	assert.True(t, util.HasErrCode(err, util.ParseError))
	assert.Equal(t, "invalid device state line 1: 0x0x0x0x", err.(*util.Err).Message)
}

func TestCalculateStateDiffsUnchangedEmpty(t *testing.T) {
	oldStates := map[string]string{}
	newStates := map[string]string{}

	diffs := calculateStateDiffs(oldStates, newStates)

	assert.Empty(t, diffs)
}

func TestCalculateStateDiffsUnchangedNonEmpty(t *testing.T) {
	oldStates := map[string]string{
		"1": "device",
		"2": "device",
	}
	newStates := map[string]string{
		"1": "device",
		"2": "device",
	}

	diffs := calculateStateDiffs(oldStates, newStates)

	assert.Empty(t, diffs)
}

func TestCalculateStateDiffsOneAdded(t *testing.T) {
	oldStates := map[string]string{}
	newStates := map[string]string{
		"serial": "added",
	}

	diffs := calculateStateDiffs(oldStates, newStates)

	assert.Equal(t, []DeviceStateChangedEvent{
		DeviceStateChangedEvent{"serial", "", "added"},
	}, diffs)
}

func TestCalculateStateDiffsOneRemoved(t *testing.T) {
	oldStates := map[string]string{
		"serial": "removed",
	}
	newStates := map[string]string{}

	diffs := calculateStateDiffs(oldStates, newStates)

	assert.Equal(t, []DeviceStateChangedEvent{
		DeviceStateChangedEvent{"serial", "removed", ""},
	}, diffs)
}

func TestCalculateStateDiffsOneAddedOneUnchanged(t *testing.T) {
	oldStates := map[string]string{
		"1": "device",
	}
	newStates := map[string]string{
		"1": "device",
		"2": "added",
	}

	diffs := calculateStateDiffs(oldStates, newStates)

	assert.Equal(t, []DeviceStateChangedEvent{
		DeviceStateChangedEvent{"2", "", "added"},
	}, diffs)
}

func TestCalculateStateDiffsOneRemovedOneUnchanged(t *testing.T) {
	oldStates := map[string]string{
		"1": "removed",
		"2": "device",
	}
	newStates := map[string]string{
		"2": "device",
	}

	diffs := calculateStateDiffs(oldStates, newStates)

	assert.Equal(t, []DeviceStateChangedEvent{
		DeviceStateChangedEvent{"1", "removed", ""},
	}, diffs)
}

func TestCalculateStateDiffsOneAddedOneRemoved(t *testing.T) {
	oldStates := map[string]string{
		"1": "removed",
	}
	newStates := map[string]string{
		"2": "added",
	}

	diffs := calculateStateDiffs(oldStates, newStates)

	assert.Equal(t, []DeviceStateChangedEvent{
		DeviceStateChangedEvent{"1", "removed", ""},
		DeviceStateChangedEvent{"2", "", "added"},
	}, diffs)
}

func TestCalculateStateDiffsOneChangedOneUnchanged(t *testing.T) {
	oldStates := map[string]string{
		"1": "oldState",
		"2": "device",
	}
	newStates := map[string]string{
		"1": "newState",
		"2": "device",
	}

	diffs := calculateStateDiffs(oldStates, newStates)

	assert.Equal(t, []DeviceStateChangedEvent{
		DeviceStateChangedEvent{"1", "oldState", "newState"},
	}, diffs)
}

func TestCalculateStateDiffsMultipleChangedMultipleUnchanged(t *testing.T) {
	oldStates := map[string]string{
		"1": "oldState",
		"2": "oldState",
	}
	newStates := map[string]string{
		"1": "newState",
		"2": "newState",
	}

	diffs := calculateStateDiffs(oldStates, newStates)

	assert.Equal(t, []DeviceStateChangedEvent{
		DeviceStateChangedEvent{"1", "oldState", "newState"},
		DeviceStateChangedEvent{"2", "oldState", "newState"},
	}, diffs)
}

func TestCalculateStateDiffsOneAddedOneRemovedOneChanged(t *testing.T) {
	oldStates := map[string]string{
		"1": "oldState",
		"2": "removed",
	}
	newStates := map[string]string{
		"1": "newState",
		"3": "added",
	}

	diffs := calculateStateDiffs(oldStates, newStates)

	assert.Equal(t, []DeviceStateChangedEvent{
		DeviceStateChangedEvent{"1", "oldState", "newState"},
		DeviceStateChangedEvent{"2", "removed", ""},
		DeviceStateChangedEvent{"3", "", "added"},
	}, diffs)
}

func TestPublishDevicesRestartsServer(t *testing.T) {
	starter := &MockServerStarter{}
	dialer := &MockServer{
		Status: wire.StatusSuccess,
		Errs: []error{
			nil, nil, nil, // Successful dial.
			util.Errorf(util.ConnectionResetError, "failed first read"),
			util.Errorf(util.ServerNotAvailable, "failed redial"),
		},
	}
	watcher := deviceWatcherImpl{
		config:      ClientConfig{dialer},
		eventChan:   make(chan DeviceStateChangedEvent),
		startServer: starter.StartServer,
	}

	publishDevices(&watcher)

	assert.Empty(t, dialer.Errs)
	assert.Equal(t, []string{"host:track-devices"}, dialer.Requests)
	assert.Equal(t, []string{"Dial", "SendMessage", "ReadStatus", "ReadMessage", "Dial"}, dialer.Trace)
	err := watcher.err.Load().(*util.Err)
	assert.Equal(t, util.ServerNotAvailable, err.Code)
	assert.Equal(t, 1, starter.startCount)
}

type MockServerStarter struct {
	startCount int
	err        error
}

func (s *MockServerStarter) StartServer() error {
	log.Printf("Starting mock server")
	if s.err == nil {
		s.startCount += 1
		return nil
	} else {
		return s.err
	}
}