pytiled_parser/tests/unit2/test_parser.py

import pytest

import xml.etree.ElementTree as etree

from contextlib import contextmanager
from typing import Callable, List, Optional, Tuple

from pytiled_parser import objects, xml_parser, utilities


@contextmanager
def does_not_raise():
    yield


def _get_root_element(xml: str) -> etree.Element:
    return etree.fromstring(xml)


layer_data = [
    (
        '<layer id="1" name="Tile Layer 1" width="10" height="10">'
        "</layer>",
        (int(1), "Tile Layer 1", None, None, None),
    ),
    (
        '<layer id="2" name="Tile Layer 2" width="10" height="10" opacity="0.5">'
        "</layer>",
        (int(2), "Tile Layer 2", None, float(0.5), None),
    ),
    (
        '<layer id="5" name="Tile Layer 4" width="10" height="10" offsetx="49" offsety="-50">'
        "<properties>"
        "</properties>"
        "</layer>",
        (
            int(5),
            "Tile Layer 4",
            objects.OrderedPair(49, -50),
            None,
            "properties",
        ),
    ),
]


@pytest.mark.parametrize("xml,expected", layer_data)
def test_parse_layer(xml, expected, monkeypatch):
    def mockreturn(properties):
        return "properties"

    monkeypatch.setattr(xml_parser, "_parse_properties_element", mockreturn)

    assert xml_parser._parse_layer(_get_root_element(xml)) == expected


@pytest.mark.parametrize(
    "test_input,expected",
    [
        ("#001122", (0x00, 0x11, 0x22, 0xFF)),
        ("001122", (0x00, 0x11, 0x22, 0xFF)),
        ("#FF001122", (0x00, 0x11, 0x22, 0xFF)),
        ("FF001122", (0x00, 0x11, 0x22, 0xFF)),
        ("FF001122", (0x00, 0x11, 0x22, 0xFF)),
    ],
)
def test_color_parsing(test_input, expected):
    """
    Tiled has a few different types of color representations.
    """
    assert utilities.parse_color(test_input) == expected


data_csv = [
    (
        "\n1,2,3,4,5,6,7,8,\n"
        "9,10,11,12,13,14,15,16,\n"
        "17,18,19,20,21,22,23,24,\n"
        "25,26,27,28,29,30,31,32,\n"
        "33,34,35,36,37,38,39,40,\n"
        "41,42,43,44,45,46,47,48\n",
        [
            [1, 2, 3, 4, 5, 6, 7, 8],
            [9, 10, 11, 12, 13, 14, 15, 16],
            [17, 18, 19, 20, 21, 22, 23, 24],
            [25, 26, 27, 28, 29, 30, 31, 32],
            [33, 34, 35, 36, 37, 38, 39, 40],
            [41, 42, 43, 44, 45, 46, 47, 48],
        ],
    ),
    ("\n0,0,0,0,0\n", [[0, 0, 0, 0, 0]]),
]


@pytest.mark.parametrize("data_csv,expected", data_csv)
def test_decode_csv_data(data_csv, expected):
    assert xml_parser._decode_csv_data(data_csv) == expected


data_base64 = [
    (
        "AQAAAAIAAAADAAAABAAAAAUAAAAGAAAABwAAAAgAAAAJAAAACgAAAAsAAAAMAAAADQAAAA4AAAAPAAAAEAAAABEAAAASAAAAEwAAABQAAAAVAAAAFgAAABcAAAAYAAAAGQAAABoAAAAbAAAAHAAAAB0AAAAeAAAAHwAAACAAAAAhAAAAIgAAACMAAAAkAAAAJQAAACYAAAAnAAAAKAAAACkAAAAqAAAAKwAAACwAAAAtAAAALgAAAC8AAAAwAAAA",
        8,
        None,
        [
            [1, 2, 3, 4, 5, 6, 7, 8],
            [9, 10, 11, 12, 13, 14, 15, 16],
            [17, 18, 19, 20, 21, 22, 23, 24],
            [25, 26, 27, 28, 29, 30, 31, 32],
            [33, 34, 35, 36, 37, 38, 39, 40],
            [41, 42, 43, 44, 45, 46, 47, 48],
        ],
        does_not_raise(),
    ),
    (
        "eJwNwwUSgkAAAMAzEQOwUCzExPb/r2N3ZlshhLYdu/bsGzkwdujIsRMTUzOnzpy7cGnuyrWFG7fu3Huw9GjlybMXr968W/vw6cu3H7/+/NsAMw8EmQ==",
        8,
        "zlib",
        [
            [1, 2, 3, 4, 5, 6, 7, 8],
            [9, 10, 11, 12, 13, 14, 15, 16],
            [17, 18, 19, 20, 21, 22, 23, 24],
            [25, 26, 27, 28, 29, 30, 31, 32],
            [33, 34, 35, 36, 37, 38, 39, 40],
            [41, 42, 43, 44, 45, 46, 47, 48],
        ],
        does_not_raise(),
    ),
    (
        "H4sIAAAAAAAAAw3DBRKCQAAAwDMRA7BQLMTE9v+vY3dmWyGEth279uwbOTB26MixExNTM6fOnLtwae7KtYUbt+7ce7D0aOXJsxev3rxb+/Dpy7cfv/782wAcvDirwAAAAA==",
        8,
        "gzip",
        [
            [1, 2, 3, 4, 5, 6, 7, 8],
            [9, 10, 11, 12, 13, 14, 15, 16],
            [17, 18, 19, 20, 21, 22, 23, 24],
            [25, 26, 27, 28, 29, 30, 31, 32],
            [33, 34, 35, 36, 37, 38, 39, 40],
            [41, 42, 43, 44, 45, 46, 47, 48],
        ],
        does_not_raise(),
    ),
    (
        "SGVsbG8gV29ybGQh",
        8,
        "lzma",
        [
            [1, 2, 3, 4, 5, 6, 7, 8],
            [9, 10, 11, 12, 13, 14, 15, 16],
            [17, 18, 19, 20, 21, 22, 23, 24],
            [25, 26, 27, 28, 29, 30, 31, 32],
            [33, 34, 35, 36, 37, 38, 39, 40],
            [41, 42, 43, 44, 45, 46, 47, 48],
        ],
        pytest.raises(ValueError),
    ),
]


@pytest.mark.parametrize(
    "data_base64,width,compression,expected,raises", data_base64
)
def test_decode_base64_data(
    data_base64, width, compression, expected, raises
):
    with raises:
        assert (
            xml_parser._decode_base64_data(data_base64, width, compression)
            == expected
        )