WeasyPrint/weasy/tests/test_css.py

# coding: utf8

#  WeasyPrint converts web documents (HTML, CSS, ...) to PDF.
#  Copyright (C) 2011  Simon Sapin
#
#  This program is free software: you can redistribute it and/or modify
#  it under the terms of the GNU Affero General Public License as
#  published by the Free Software Foundation, either version 3 of the
#  License, or (at your option) any later version.
#
#  This program is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#  GNU Affero General Public License for more details.
#
#  You should have received a copy of the GNU Affero General Public License
#  along with this program.  If not, see <http://www.gnu.org/licenses/>.

import os.path
from attest import Tests, assert_hook
import attest
import cssutils

from .. import css

from . import resource_filename, parse_html


suite = Tests()


@suite.test
def test_style_dict():
    style = css.StyleDict({
        'margin-left': cssutils.css.PropertyValue('12px'),
        'display': cssutils.css.PropertyValue('block')
    })
    assert style.display == 'block'
    assert style.margin_left == 12
    with attest.raises(AttributeError):
        style.position


@suite.test
def test_find_stylesheets():
    document = parse_html('doc1.html')

    sheets = list(css.find_stylesheets(document))
    assert len(sheets) == 2
    # Also test that stylesheets are in tree order
    assert [s.href.rsplit('/', 1)[-1] for s in sheets] \
        == ['sheet1.css', 'doc1.html']

    rules = list(rule for sheet in sheets
                      for rule in css.resolve_import_media(sheet, 'print'))
    assert len(rules) == 8
    # Also test appearance order
    assert [rule.selectorText for rule in rules] \
        == ['li', 'p', 'ul', 'a', 'a:after', ':first', 'ul',
            'body > h1:first-child']


@suite.test
def test_expand_shorthands():
    sheet = cssutils.parseFile(resource_filename('sheet2.css'))
    assert sheet.cssRules[0].selectorText == 'li'
    style = sheet.cssRules[0].style
    assert style['margin'] == '2em 0'
    assert style['margin-bottom'] == '3em'
    assert style['margin-left'] == '4em'
    assert not style['margin-top']
    css.expand_shorthands(sheet)
    # expand_shorthands() builds new style object
    style = sheet.cssRules[0].style
    assert not style['margin']
    assert style['margin-top'] == '2em'
    assert style['margin-right'] == '0'
    assert style['margin-bottom'] == '2em', \
        "3em was before the shorthand, should be masked"
    assert style['margin-left'] == '4em', \
        "4em was after the shorthand, should not be masked"


@suite.test
def test_annotate_document():
    user_stylesheet = cssutils.parseFile(resource_filename('user.css'))
    ua_stylesheet = cssutils.parseFile(resource_filename('mini_ua.css'))
    document = parse_html('doc1.html')

    css.annotate_document(document, [user_stylesheet], [ua_stylesheet])

    # Element objects behave a lists of their children
    head, body = document
    h1, p, ul = body
    li = list(ul)
    a, = li[0]
    after = a.pseudo_elements['after']

    assert h1.style['background-image'][0].absolute_uri == 'file://' \
        + os.path.abspath(resource_filename('logo_small.png'))

    assert h1.style.font_weight == '700'

    sides = ('-top', '-right', '-bottom', '-left')
    # 32px = 1em * font-size: 2em * initial 16px
    for side, expected_value in zip(sides, ('32px', '0', '32px', '0')):
        assert p.style['margin' + side].value == expected_value

    # 32px = 2em * initial 16px
    for side, expected_value in zip(sides, ('32px', '32px', '32px', '32px')):
        assert ul.style['margin' + side].value == expected_value

    # thick = 5px, 0.25 inches = 96*.25 = 24px
    for side, expected_value in zip(sides, ('0', '5px', '0', '24px')):
        assert ul.style['border' + side + '-width'].value == expected_value

    # 32px = 2em * initial 16px
    # 64px = 4em * initial 16px
    for side, expected_value in zip(sides, ('32px', '0', '32px', '64px')):
        assert li[0].style['margin' + side].value == expected_value

    assert a.style.text_decoration == 'underline'

    color = a.style['color'][0]
    assert (color.red, color.green, color.blue, color.alpha) == (255, 0, 0, 1)

    # The href attr should be as in the source, not made absolute.
    assert ''.join(v.value for v in after.style['content']) == ' [home.html]'

    # TODO much more tests here: test that origin and selector precedence
    # and inheritance are correct, ...


@suite.test
def test_default_stylesheet():
    document = parse_html('doc1.html')
    css.annotate_document(document)
    assert document.head.style.display == 'none', \
        'The HTML4 user-agent stylesheet was not applied'