package handlers

import (
	"os"
	"testing"

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

func TestLoadFileCommandReturnZeroLengthOnFileNotFound(t *testing.T) {
	mockIoData := a2io.MockIoData{
		BytesToRead:  []byte("FILE_DOES_NOT_EXIST\x00"),
		BytesWritten: make([]byte, 1000),
	}

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

	SetCommunication(&mockIo)

	LoadFileCommand()

	got := mockIoData.BytesWritten

	if got[0] != 0 && got[1] != 0 {
		t.Errorf("MenuCommand() sent = %s; want 00", got)
	}
}

func TestLoadFileCommandReturnsFile(t *testing.T) {
	mockIoData := a2io.MockIoData{
		BytesToRead:  []byte("loadFile.go\x00"),
		BytesWritten: make([]byte, 10000),
	}

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

	SetCommunication(&mockIo)

	LoadFileCommand()

	fileInfo, _ := os.Stat("loadFile.go")

	got := mockIoData.BytesWritten

	lowSize := fileInfo.Size() & 0xFF
	hiSize := (fileInfo.Size() & 0xFF00) >> 8

	if got[0] != byte(lowSize) && got[1] != byte(hiSize) {
		t.Errorf("MenuCommand() fileSize = %02X%02X; want %02X%02X", got[1], got[0], hiSize, lowSize)
	}

	if mockIoData.NumberBytesWritten-2 != int(fileInfo.Size()) {
		t.Errorf("MenuCommand() byte count = %04X; want %04X", mockIoData.NumberBytesWritten-2, fileInfo.Size())
	}
}