package day2

import (
	"testing"

	"gitea.paas.celticinfo.fr/oabrivard/aoc2023/utils"
)

func TestPuzzleCount(t *testing.T) {
	lines := []string{
		"Game 1: 3 blue, 4 red; 1 red, 2 green, 6 blue; 2 green",
		"Game 2: 1 blue, 2 green; 3 green, 4 blue, 1 red; 1 green, 1 blue",
		"Game 3: 8 green, 6 blue, 20 red; 5 blue, 4 red, 13 green; 5 green, 1 red",
		"Game 4: 1 green, 3 red, 6 blue; 3 green, 6 red; 3 green, 15 blue, 14 red",
		"Game 5: 6 red, 1 blue, 3 green; 2 blue, 1 red, 2 green",
	}

	result := puzzleCount(lines)

	if result != 8 {
		t.Fatalf("expected 8, got %d", result)
	}
}

func TestPuzzleCountWithInput(t *testing.T) {
	lines := utils.ReadLines("input.txt")
	result := puzzleCount(lines)

	if result != 2377 {
		t.Fatalf("expected 2377, got %d", result)
	}
}

func TestPuzzlePower(t *testing.T) {
	lines := []string{
		"Game 1: 3 blue, 4 red; 1 red, 2 green, 6 blue; 2 green",
		"Game 2: 1 blue, 2 green; 3 green, 4 blue, 1 red; 1 green, 1 blue",
		"Game 3: 8 green, 6 blue, 20 red; 5 blue, 4 red, 13 green; 5 green, 1 red",
		"Game 4: 1 green, 3 red, 6 blue; 3 green, 6 red; 3 green, 15 blue, 14 red",
		"Game 5: 6 red, 1 blue, 3 green; 2 blue, 1 red, 2 green",
	}

	result := puzzlePower(lines)

	if result != 2286 {
		t.Fatalf("expected 2286, got %d", result)
	}
}

func TestPuzzlePowerWithInput(t *testing.T) {
	lines := utils.ReadLines("input.txt")
	result := puzzlePower(lines)

	if result != 71220 {
		t.Fatalf("expected 71220, got %d", result)
	}
}