Added websocket and added package to device_test.go