grid_modification_test.go

package sudoku

import (
	"testing"
)

func TestCopyWithModifications(t *testing.T) {
	sourceGrid := NewGrid()
	sourceGrid.MutableCell(0, 0).SetMark(1, true)
	sourceGrid.MutableCell(0, 0).SetExcluded(1, true)

	sourceGridRO := sourceGrid.Copy()

	if !isGridImpl(sourceGridRO) {
		t.Fatal("Expected grid.Copy to return an immutable grid")
	}

	gridFive := NewGrid()
	gridFive.MutableCell(0, 0).SetNumber(5)

	gridMarks := NewGrid()
	gridMarksCell := gridMarks.MutableCell(0, 0)
	gridMarksCell.SetMark(2, true)
	gridMarksCell.SetMark(1, false)

	gridExcludes := sourceGrid.MutableCopy()
	gridExcludesCell := gridExcludes.MutableCell(0, 0)
	gridExcludesCell.SetExcluded(2, true)
	gridExcludesCell.SetExcluded(1, false)

	tests := []struct {
		modifications GridModification
		expected      Grid
		description   string
	}{
		{
			GridModification{
				&CellModification{
					Cell:   CellRef{0, 0},
					Number: 5,
				},
			},
			gridFive,
			"Single valid number",
		},
		{
			GridModification{
				&CellModification{
					Cell:   CellRef{0, 0},
					Number: DIM + 1,
				},
			},
			sourceGrid,
			"Single invalid number",
		},
		{
			GridModification{
				&CellModification{
					Cell:   CellRef{0, 0},
					Number: -1,
					MarksChanges: map[int]bool{
						1:       false,
						2:       true,
						DIM + 1: true,
					},
				},
			},
			gridMarks,
			"Marks",
		},
		{
			GridModification{
				&CellModification{
					Cell:   CellRef{0, 0},
					Number: -1,
					ExcludesChanges: map[int]bool{
						1:       false,
						2:       true,
						DIM + 1: true,
					},
				},
			},
			gridExcludes,
			"Excludes",
		},
	}

	for i, test := range tests {
		result := sourceGrid.CopyWithModifications(test.modifications)
		if result.Diagram(true) != test.expected.Diagram(true) {
			t.Error("Test", i, "failed", test.description, "Got", result.Diagram(true), "expected", test.expected.Diagram(true))
		}

		//Also test the non-mutalbe grid implementation
		nonMutableResult := sourceGridRO.CopyWithModifications(test.modifications)
		if result.Diagram(true) != test.expected.Diagram(true) {
			t.Error("Test", i, "failed with non-mutable copy", test.description, "Got", nonMutableResult.Diagram(true), "expected", test.expected.Diagram(true))
		}
	}

}