import re
from pycalver import parse


def test_readme_pycalver1():
    version_str = "v201712.0001-alpha"
    version_info = parse.PYCALVER_RE.match(version_str).groupdict()

    assert version_info == {
        "version" : "v201712.0001-alpha",
        "calver"  : "v201712",
        "year"    : "2017",
        "month"   : "12",
        "build"   : ".0001",
        "release" : "-alpha",
    }


def test_readme_pycalver2():
    version_str = "v201712.0033"
    version_info = parse.PYCALVER_RE.match(version_str).groupdict()

    assert version_info == {
        "version" : "v201712.0033",
        "calver"  : "v201712",
        "year"    : "2017",
        "month"   : "12",
        "build"   : ".0033",
        "release" : None,
    }


def test_re_pattern_parts():
    part_re_by_name = {
        part_name: re.compile(part_re_str)
        for part_name, part_re_str in parse.RE_PATTERN_PARTS.items()
    }

    cases = [
        ("pep440_version", "201712.31", "201712.31"),
        ("pep440_version", "v201712.0032", None),
        ("pep440_version", "201712.0033-alpha", None),
        ("version", "v201712.0034", "v201712.0034"),
        ("version", "v201712.0035-alpha", "v201712.0035-alpha"),
        ("version", "v201712.0036-alpha0", "v201712.0036-alpha"),
        ("version", "v201712.0037-pre", "v201712.0037"),
        ("version", "201712.38a0", None),
        ("version", "201712.0039", None),
        ("calver", "v201712", "v201712"),
        ("calver", "v201799", "v201799"),   # maybe date validation should be a thing
        ("calver", "201712", None),
        ("calver", "v20171", None),
        ("build", ".0012", ".0012"),
        ("build", ".11012", ".11012"),
        ("build", ".012", None),
        ("build", "11012", None),
        ("release", "-alpha", "-alpha"),
        ("release", "-beta", "-beta"),
        ("release", "-dev", "-dev"),
        ("release", "-rc", "-rc"),
        ("release", "-post", "-post"),
        ("release", "-pre", ""),
        ("release", "alpha", ""),
    ]

    for part_name, line, expected in cases:
        part_re = part_re_by_name[part_name]
        result = part_re.search(line)
        if result is None:
            assert expected is None, (part_name, line)
        else:
            result_val = result.group(0)
            assert result_val == expected, (part_name, line)


def test_parse_version_info():
    version_str = "v201712.0001-alpha"
    version_nfo = parse.parse_version_info(version_str)

    assert version_nfo.pep440_version == "201712.1a0"
    assert version_nfo.version == "v201712.0001-alpha"
    assert version_nfo.calver == "v201712"
    assert version_nfo.year == "2017"
    assert version_nfo.month == "12"
    assert version_nfo.build == ".0001"
    assert version_nfo.release == "-alpha"

    version_str = "v201712.0001"
    version_nfo = parse.parse_version_info(version_str)

    assert version_nfo.pep440_version == "201712.1"
    assert version_nfo.version == "v201712.0001"
    assert version_nfo.calver == "v201712"
    assert version_nfo.year == "2017"
    assert version_nfo.month == "12"
    assert version_nfo.build == ".0001"
    assert version_nfo.release is None


def test_default_parse_patterns():
    lines = [
        "# setup.py",
        "import setuptools",
        "__version__ = 'v201712.0002-alpha'",
        "setuptools.setup(",
        "...",
        "   version='201712.2a0',",
    ]

    patterns = [
        "{version}",
        "{pep440_version}",
    ]

    matches = parse.parse_patterns(lines, patterns)
    assert len(matches) == 2

    assert matches[0].lineno == 2
    assert matches[1].lineno == 5

    assert matches[0].pattern == patterns[0]
    assert matches[1].pattern == patterns[1]

    assert matches[0].match == "v201712.0002-alpha"
    assert matches[1].match == "201712.2a0"


def test_explicit_parse_patterns():
    lines = [
        "# setup.py",
        "import setuptools",
        "__version__ = 'v201712.0002-alpha'",
        "setuptools.setup(",
        "...",
        "   version='201712.2a0',",
    ]

    patterns = [
        "__version__ = '{version}'",
        "version='{pep440_version}'",
    ]

    matches = parse.parse_patterns(lines, patterns)
    assert len(matches) == 2

    assert matches[0].lineno == 2
    assert matches[1].lineno == 5

    assert matches[0].pattern == patterns[0]
    assert matches[1].pattern == patterns[1]

    assert matches[0].match == "__version__ = 'v201712.0002-alpha'"
    assert matches[1].match == "version='201712.2a0'"


def test_badge_parse_patterns():
    lines = [
        ":alt: PyPI version",
        "",
        ".. |version| image:: https://img.shields.io/badge/CalVer-v201809.0002--beta-blue.svg",
        ":target: https://calver.org/",
        ":alt: CalVer v201809.0002-beta",
        "",
    ]

    patterns = [
        "badge/CalVer-{calver}{build}-{release}-blue.svg",
        ":alt: CalVer {version}",
    ]

    matches = parse.parse_patterns(lines, patterns)
    assert len(matches) == 2

    assert matches[0].lineno == 2
    assert matches[1].lineno == 4

    assert matches[0].pattern == patterns[0]
    assert matches[1].pattern == patterns[1]

    assert matches[0].match == "badge/CalVer-v201809.0002--beta-blue.svg"
    assert matches[1].match == ":alt: CalVer v201809.0002-beta"