#! /usr/bin/env python3

import pathlib
import pytest
import day2 as aoc

PUZZLE_DIR = pathlib.Path(__file__).parent


# these test fixtures setup the test, mainly by reading the filename into a string in this simple case.
@pytest.fixture
def example():
    puzzle_input = (PUZZLE_DIR / "example").read_text().strip()
    return aoc.parse(puzzle_input)


def test_parse(example):
    """Test that input is parsed properly"""
    assert example == [
        aoc.PasswordSpec(first=1, second=3, letter="a", password="abcde"),
        aoc.PasswordSpec(first=1, second=3, letter="b", password="cdefg"),
        aoc.PasswordSpec(first=2, second=9, letter="c", password="ccccccccc"),
    ]


def test_part1(example):
    """Test part 1 on example input"""
    assert aoc.part1(example) == 2


def test_part2(example):
    """Test part 2 on example input"""
    assert aoc.part2(example) == 1