package handlers

import (
	"strings"
	"testing"

	"github.com/tjboldt/Apple2-IO-RPi/RaspberryPi/apple2driver/a2io"
)

func TestMenuCommand(t *testing.T) {
	mockIoData := a2io.MockIoData{
		ErrorToThrow: nil,
		BytesWritten: make([]byte, 1000),
	}

	mockIo := a2io.MockIo{Data: &mockIoData}

	SetCommunication(&mockIo)

	MenuCommand()

	want := "Apple2-IO-RPi"
	got := string(mockIoData.BytesWritten)

	if strings.Index(got, want) != 0 {
		t.Errorf("MenuCommand() sent = %s; want startsWith %s", got, want)
	}
}