main_test.go

// Integration tests :)

package main

import (
	"fmt"
	"os"
	"path/filepath"
	"strings"
	"testing"

	"github.com/skx/cpmulator/consoleout"
	"github.com/skx/cpmulator/cpm"
)

// TestDriveChange ensures the drive-letter changes after
// changing drives.
func TestDriveChange(t *testing.T) {

	t.Setenv("BDOS_ADDRESS", "0xb000")
	t.Setenv("BIOS_ADDRESS", "0xbf00")

	obj, err := cpm.New(cpm.WithOutputDriver("logger"))
	if err != nil {
		t.Fatalf("Create CP/M failed")
	}

	// Load the CCP binary - resetting RAM in the process.
	err = obj.LoadCCP()
	if err != nil {
		t.Fatalf("load CCP failed")
	}

	obj.SetDrives(false)

	obj.StuffText("C:\r\nEXIT\r\n")
	// Run it
	err = obj.Execute([]string{})
	if err != nil && err != cpm.ErrHalt {
		t.Fatalf("failed to run: %s", err)
	}

	// Get our output handle
	helper := obj.GetOutputDriver()
	l, ok := helper.(*consoleout.OutputLoggingDriver)
	if !ok {
		t.Fatalf("failed to cast output driver")
	}

	// Get output written to the screen, and remove newlines
	out := l.GetOutput()
	out = strings.ReplaceAll(out, "\n", "")
	out = strings.ReplaceAll(out, "\r", "")
	if out != `A>C>C>` {
		t.Fatalf("unexpected output '%v'", out)
	}

	// Reset the text - confirm it is now empty
	l.Reset()
	if l.GetOutput() != "" {
		t.Fatalf("resetting our history didn't work")
	}
}

// TestReadWriteRand invokes our help-samples to read/write
// records - via the external API.
func TestReadWriteRand(t *testing.T) {

	t.Setenv("BDOS_ADDRESS", "0xb000")
	t.Setenv("BIOS_ADDRESS", "0xbf00")

	obj, err := cpm.New()
	if err != nil {
		t.Fatalf("Create CP/M failed")
	}

	// Load the CCP binary - resetting RAM in the process.
	err = obj.LoadCCP()
	if err != nil {
		t.Fatalf("load CCP failed")
	}

	obj.SetDrives(false)
	obj.SetDrivePath("A", "samples/")
	obj.StuffText("WRITE foo\nREAD foo\nEXIT\n")

	// Run it
	err = obj.Execute([]string{})
	if err != nil && err != cpm.ErrBoot {
		t.Fatalf("failed to run: %s", err)
	}

	// Remove the generated file
	os.Remove(filepath.Join("samples", "FOO"))
}

// TestCompleteLighthouse plays our Lighthouse game, to completion.
//
// It uses the fast/hacky solution rather than the slow/normal/real one
// just to cut down on the scripting.
//
// However it is a great test to see that things work as expected.
func TestCompleteLighthouse(t *testing.T) {

	t.Setenv("BDOS_ADDRESS", "0xb000")
	t.Setenv("BIOS_ADDRESS", "0xbf00")

	obj, err := cpm.New(cpm.WithOutputDriver("logger"))
	if err != nil {
		t.Fatalf("Create CP/M failed")
	}

	// Load the CCP binary - resetting RAM in the process.
	err = obj.LoadCCP()
	if err != nil {
		t.Fatalf("load CCP failed")
	}

	obj.SetDrives(false)
	obj.SetDrivePath("A", "dist/")
	obj.StuffText("\nLIHOUSE\nAAAA\ndown\nEXAMINE DESK\nTAKE METEOR\nUP\n\nn\nquit\n")

	// Run it
	err = obj.Execute([]string{})
	if err != nil && err != cpm.ErrBoot {
		t.Fatalf("failed to run: %s", err)
	}

	// Get our output handle
	helper := obj.GetOutputDriver()
	l, ok := helper.(*consoleout.OutputLoggingDriver)
	if !ok {
		t.Fatalf("failed to cast output driver")
	}

	// Get the text written to the screen
	out := l.GetOutput()

	fmt.Printf("\n\nOUTPUT: %s\n\n", out)
	// Ensure the game was completed - easy path.
	if !strings.Contains(out, "Congratulations") {
		t.Fatalf("failed to win")
	}
	if !strings.Contains(out, "You won") {
		t.Fatalf("failed to win")
	}

	// Reset the text - confirm it is now empty
	l.Reset()
	if l.GetOutput() != "" {
		t.Fatalf("resetting our history didn't work")
	}
}