package main

import (
	"testing"
)

const testInput = `..@@.@@@@.
@@@.@.@.@@
@@@@@.@.@@
@.@@@@..@.
@@.@@@@.@@
.@@@@@@@.@
.@.@.@.@@@
@.@@@.@@@@
.@@@@@@@@.
@.@.@@@.@.
`

func TestPart1(t *testing.T) {
	d := &DayFour{}
	if err := d.ParseInput(testInput); err != nil {
		t.Fatalf("ParseInput failed: %v", err)
	}

	got, err := d.Part1()
	if err != nil {
		t.Fatalf("Part1 failed: %v", err)
	}

	want := 13
	if got != want {
		t.Errorf("Part1() = %d, want %d", got, want)
	}
}

func TestPart2(t *testing.T) {
	d := &DayFour{}
	if err := d.ParseInput(testInput); err != nil {
		t.Fatalf("ParseInput failed: %v", err)
	}

	got, err := d.Part2()
	if err != nil {
		t.Fatalf("Part2 failed: %v", err)
	}

	want := 43
	if got != want {
		t.Errorf("Part2() = %d, want %d", got, want)
	}
}