"""Utility used for unit test."""
from typing import Dict, Sequence


class MockNode:
    """Mock node class."""

    def __init__(self,
                 attrs: Dict[str, str],
                 child_attrs: Sequence[Dict[str, str]] = ({},),
                 is_child: bool = False):
        self.attributes = attrs
        self.childs = [MockNode(attrs, is_child=True) for attrs in child_attrs if attrs]

        self.is_child = is_child
        if 'bounds' not in self.attributes:
            self.attributes['bounds'] = '[0,0][384,384]'

    def __str__(self):
        xml_str_elements = []
        if not self.is_child:
            xml_str_elements.append("<?xml version='1.0' encoding='UTF-8' standalone='yes' ?>")
            xml_str_elements.append('<hierarchy rotation="0">')

        xml_str_elements.append('<node index="0"')
        for attr_key, attr_val in self.attributes.items():
            xml_str_elements.append(f' {attr_key}="{attr_val}"')
        xml_str_elements.append('>')

        for child in self.childs:
            xml_str_elements.append(str(child))

        xml_str_elements.append('</node>')

        if not self.is_child:
            xml_str_elements.append('</hierarchy>')

        return ''.join(xml_str_elements)