from nose.tools import istest, assert_equal

from mammoth import html_paths, document_matchers, styles
from mammoth.styles.parser.style_mapping_parser import parse_style_mapping
from mammoth.styles.parser.tokeniser import tokenise
from mammoth.styles.parser.token_iterator import TokenIterator


@istest
def document_matcher_is_mapped_to_html_path_using_fat_arrow():
    assert_equal(
        styles.style(document_matchers.paragraph(), html_paths.path([html_paths.element(["h1"])])),
        read_style_mapping("p => h1")
    )


def read_style_mapping(string):
    return parse_style_mapping(TokenIterator(tokenise(string)))