3 y1W@sdZddlmZmZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddlZ ddlZddlZddlmZmZmZmZmZmZddlmZddlmZmZmZmZdd l m!Z!dd l"m#Z#m$Z$m%Z%dd l"m&Z&dd l"m'Z'dd l(m)Z)e j*Z+ej,ddZ-ddZ.ddZ/ddZ0eddZ1eddZ2d6ddZ3eddZ4edd Z5ed!d"Z6ed#d$Z7d%d&Z8ed'd(Z9ed)d*Z:d7d+d,Z;ed-d.Zed2d3Z?ed4d5Z@dS)8z weasyprint.tests.test_api ------------------------- Test the public API. :copyright: Copyright 2011-2014 Simon Sapin and contributors, see AUTHORS. :license: BSD, see LICENSE for details. )divisionunicode_literalsN)resource_filenameassert_no_logs capture_logsTestHTML http_servertemp_directory)image_to_pixels)urljoin urlencodeurlparse_uses_relative iteritems)path2url)HTMLCSSdefault_url_fetcher)__main__) navigator) _TaggedTupleccs>t0tj}ztj|dVWdtj|XWdQRXdS)z2Change the current directory in a context manager.N) CHDIR_LOCKosgetcwdchdir)pathZold_dirrY/var/www/html/enquirykeeper_venv/lib/python3.6/site-packages/weasyprint/tests/test_api.pyr,s   rc Cst|d }|jSQRXdS)zShortcut for reading a file.rbN)openread)filenamefdrrr read_file8s r$c Cs$t|d}|j|WdQRXdS)zShortcut for reading a file.wbN)r write)r"contentr#rrr write_file>s r(c .Kst|}t|}|||f|||fd|i|||fd|i||||f|||fd|i|||fd|i|t|d}|||f|WdQRXt|d}||fd|i|WdQRXt|d}||fd|i|WdQRXt|d}|j}WdQRXttjjtjtjj d|}|||f|||f||d||j d pzd } ||f|j | |d|WdQRXt j t|d d d WdQRXdS)z1Common code for testing the HTML and CSS classes.Zguessr"urlrNfile_obj resources)stringbase_urlencodingutf8ZfooZbar)r"r))rrr r!rrrdirname__file__joingetdecodepytestZraises TypeError) class_basenamecheckkwargsZabsolute_filenamer)r#r'Zrelative_filenamer.rrr_test_resourceDs2       r;c Csddd}ttd|ttd|ddttjjt~tjjdd}tj j |}|t||d |t|d d d |j \}}|j d kst tjj|ddd|t|dd WdQRXdS)z(Test the constructor for the HTML class.TcSs|jjdkstdd|jDddgks,t|j\}}dd|Dddd d gksTt|d }|jd ksjt|rt|jd }|jdst|jdstn|jdkstdS)z@Check that a parsed HTML document looks like resources/doc1.htmlhtmlcSsg|] }|jqSr)tag).0childrrr gsz9test_html_parsing..check_doc1..headbodycSsg|] }|jqSr)r=)r>r?rrrr@ish1pZulZdivru)WeasyPrint test document (with Ünicōde)z pattern.pngzfile:z&weasyprint/tests/resources/pattern.pngN) root_elementr=AssertionErrortextr r- startswithendswith)r< has_base_urlZ_headrBrCr)rrr check_doc1ds  z%test_html_parsing..check_doc1z doc1.htmlzdoc1_UTF-16BE.htmlzUTF-16BE)r.r+)treer-)rLF)rJrAbasez resources/)Zhref.N)T)r;rrrrr0r1r2lxmlr<parseZgetrootr=rFetreeZ SubElement)rKr"rLrAZ_bodyrrrtest_html_parsingas    rRcCs(dd}ttd|ttd|dddS)z'Test the constructor for the CSS class.cSs|jjd }|jjdkst|j\}}|jdks6t|j\}|jdksLt|jdksZt|j\}|jdksptt |j |j}|j dst|j dstd S) z@Check that a parsed stylsheet looks like resources/utf8-test.cssrz h1::beforer'uI løvë Unicodezbackground-imageURIzfile:z&weasyprint/tests/resources/pattern.pngN) Z stylesheetZrulesselectorZas_cssrFZ declarationsnamevaluetyper r-rHrI)cssZruler'Z backgroundr,Z url_valuer)rrr check_csss  z#test_css_parsing..check_cssz utf8-test.csszlatin1-test.csslatin1)r.N)r;r)rZrrrtest_css_parsings r\Fc Cs8ddlm}m}m}m}|r||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||g}d} n|r ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||g}d} n|r ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||g}d} n||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||g}d} tjjtj |} |d| | t | | | dj |dS)Nr)_rBassert_pixels_equalZapi_png) test_drawr]r^r_r`cairo ImageSurfacecreate_from_pngioBytesIOr r2) png_bytesx2blankrotatedr]r^r_r`Zexpected_pixelssizesurfacerrrcheck_png_patternsh">>>>>>>>>>>>>>>B"" rpcCsXtd}d}d}t||d}t|d}|j|gd}|j|gd}|jdsPt|jds^tt|Gd d d t}|}|j||gd|j |kst|} |j| |gd| j |kstt } t j j | d } t j j | d } |j| |gd|j| |gdt| |kstt| |ks*tt j j | d } t j j | d} t| d}|j||gdWdQRXt| d} |j| |gdWdQRXt| |kstt| |kstWdQRX|j|gdd} t| ddt|dd}|j|gd}t|ddt||ddjtj|jdgd|ks0ttd||f|ddj|ksTtdS)z#Test rendering with the Python API.z dummy.htmlzz @page { margin: 2px; size: 8px; background: #fff } body { margin: 0; font-size: 0 } img { image-rendering: pixelated } @media screen { img { transform: rotate(-90deg) } } )r,r-)r,) stylesheetssPNG  s%PDFc@s$eZdZddZddZddZdS)z%test_python_render..fake_filecSs g|_dS)N)chunks)selfrrr__init__sz.test_python_render..fake_file.__init__cSs|jjt|dddS)N)rrappendbytes)rsdatarrrr&sz+test_python_render..fake_file.writecSs dj|jS)Nrc)r2rr)rsrrrgetvaluesz.test_python_render..fake_file.getvalueN)__name__ __module__ __qualname__rtr&rxrrrr fake_filesr|z1.pngz1.pdfz2.pngz2.pdfr%N)rq resolutionT)rkscreen)r, media_type)rm)r,r-rr/z%s)rrr write_png write_pdfrHrFrpobjectrxr rrr2r$r rhriencode)r-Z html_stringZ css_stringr<rYrj pdf_bytesr|Zpng_fileZpdf_filetempZ png_filenameZ pdf_filename x2_png_bytesZ screen_cssrotated_png_bytesrrrtest_python_rendersZ            rc Csd}d}d|d|}d|}ttdVt|dd}|j}|j}|jd d }t|dd d j}td|dd j} WdQRXt|t|ddt| dddIdd} tV} t| @ttd} td| td|td|td|j dj dtd|td|| d| dtd|ks>t td |ksPt | d!td"|ksjt t j j| d} | | d#td$|kst tt j j| d}| |d%td&|kst | d'td(|kst | d)| d*td+|kst td,|kst | d-| d.td/|ks6t td0|ksHt | d1}||ks^t | d2|d3ttd4td4|kst | d5|d3}||kst | d6| d7| d8td9|kst td:|kst td;|kst | d<}|jd=d>ks t | d?}|jd=d@ks(t | dA}|jd=dBksDt | dC}||ksZt | dD}| dE|kstt ||kst t jdFt jdFt}| d5|d3}WdQRXt|d@kst |d>jdGst || kst | dH|d3}||kst WdQRXWdQRXdS)Jz)Test rendering with the command-line API.s @page { margin: 2px; size: 8px; background: #fff } @media screen { img { transform: rotate(-90deg) } } body { margin: 0; font-size: 0 } sss$z dummy.html)r,r-r})r~r)r,r-r)r,NT)rm)rlrcc SsBtj|}tj}ztt_tj|j||dWdtt_X|jS)N)stdinstdout)rhrirrrmainsplitrx)argsrrrrrrun?s z%test_command_line_render..runz pattern.pngz no_css.htmlz combined.htmlzcombined-UTF-16BE.htmlasciizUTF-16BEz linked.htmlz style.csszcombined.html out1.pngzcombined.html out2.pdfzout1.pngzout2.pdfz3combined-UTF-16BE.html out3.png --encoding UTF-16BEzout3.pngz out4.pngzout4.pngz out5.pngzout5.pngzlinked.html out6.pngzout6.pngzcombined.html out7 -f pngzcombined.html out8 --format pdfZout7Zout8zno_css.html out9.pngz"no_css.html out10.png -s style.csszout9.pngz out10.pngz--format png combined.html -z - out11.png)rz out11.pngz--format png - -z+combined.html out13.png --media-type screenz!combined.html out12.png -m screenzlinked.html out14.png -m screenz out12.pngz out13.pngz out14.pngz-f pdf combined.html -s attachmentrz%-f pdf -a pattern.png combined.html -rz2-f pdf -a style.css -a pattern.png combined.html -r z-f png -r 192 linked.html -z%-f png --resolution 192 linked.html -z%linked.html - -f png --resolution 192Z subdirectoryzWARNING: Failed to load imagez--format png --base-url .. - -)rc)rrrrrrpr r$r(r4rrFrrr2rcountmkdirrlenrH)rYr<ZcombinedZlinkedZhtml_objrrjrrZempty_png_bytesrrZ pattern_bytesZcombined_absoluteZ combined_urlrlogsrrrtest_command_line_render%s                     rcCs$d}t|dj}t|d}t}t|t||tjd|gksNttjd\}t|j|ksntt|j|ksttj |tjdgkstt|dj|t ||kstt j ddkr tj |tjdgkstt|dj|t ||ks tWdQRXWdQRXdS) z7Test non-ASCII filenames both in Unicode or bytes form.s )r,u UnicödérN.rN) rrrpr rr(rlistdirrFremover$sys version_info)r<rjZunicode_filenamerZbytes_filenamerrrtest_unicode_filenamess(      rcCstdd}tdd}|j|gd}|jds0t|j|gj|ksHt|j|gd}|j|gdd}|j\}|jdks|t|j dkst|j|ddfkst|j |gj|ddfkstt j t j dd}|jt j|tj}|j|t|jt j t j dd}t j|}|jd d |jtj d |jdd|j|tj}|j|t|jdd |j|gdd}|j\}|j|j fdkst|jd d \}} } | | fdkstt|dddd} |j|gdd}|j\}|j|j fdkst| |jdd dksttddj}|j\} } | j| j fdksHt| j| j fd ks^t|j}| |d!ksxt|j | | gj|kst| |j | gjd"kst| |j | gjd#kstdS)$Nz)r,z @page { margin: 2px; size: 8px; background: #fff } html { background: #00f; } body { background: #f00; width: 1px; height: 1px } )rqs%PDFT)Zenable_hintingrar )rmr})r~rb)rkcSs@|\}}}tjjtj|}|j|jf||fks8t||fS)N)rerfrgrhriZ get_widthZ get_heightrF)resultrjwidthheightrorrrpng_sizes z$test_low_level_api..png_sizegfffff&b@ z

 r)rara)rbrb)rara)rr)rr)rr)rr)rr)rr)rrrrHrFrenderrpagesrrcopyrerfZ FORMAT_ARGB32ZpaintZContextrhriZ write_to_pngrprx translaterotatemathpi)r<rYrrjdocumentpageror*contextrrrZpage_1Zpage_2rrrrtest_low_level_apis`           rc Csx|D]}|j}x0t|D]$\}\}}t|dt|df||<qW|j}xht|D]\\}}|j} |\} } \}}} } t| | t|dt|dt| dt| dff}| |_|||<qRW|j}x|d?dggd:dgfgd=d>dS)NFcSsHt|dj}|rt|jdd|jD|ks4t|j|ksDtdS)N)r,cSsg|] }|jqSr)r)r>rDrrrr@.sz.assert_bookmarks..)rrrrrFZmake_bookmark_tree)r<Zexpected_by_pageZ expected_treerrrrrassert_bookmarks*s  z(test_bookmarks..assert_bookmarksz

a

b

c

d

e

rarrbrrcr dea

Title 1

Title 2

Title 3

Title 4

Title 5

Title 6

Title 7

Title 8

Title 9

Title 10

Title 11

Title 1Title 2dTitle 3Title 4,Title 5Title 6Title 7Title 8Title 9Title 10Title 11Xz

A

depth 1

B

depth 2

C

depth 1

D

depth 2

E

depth 3

Ar_C(D<EPa

A

h2 depth 1

B

h4 depth 2

C

h3 depth 2

D

h5 depth 3

E

h1 depth 1

F

h2 depth 2

G

h2 depth 2

H

h4 depth 3

I

h1 depth 1

rFFZGnHIu

ééz7

! !2zo

! T)rz~

! )Frr)rrrrr)rrrrr )rrrrr)r rrrr)rrr)rrr)rrr)rrr )rrr)rrrrr)rrrrr)rrrrr)r rrrr)r rrrr)rrrrr)r rrrr)rrrrr)r rrrr)rrrrr)rrrrr)r rr)rrr)rrr)rrr)rrr)rrr)rrr)rrr)rrr)rrr)rrr)rrrrr)r rrrr)rr_rrr)r rrrr)rrrrr)rrr)rrr)rrr)rrr)rrr)rrrrr)r rrrr)rr_rrr)rrrrr)rrrrr)rrrrr)r rrrr)r rrrr)rrrrr)rrr)rrr)rrr)rrr)rrr)rrr)rrr)rrr)rrr)rrrrr)rrr )rrrrr)rrr )rrrrr)rrr )rrrrr)rrr )rrrr)rrrrtest_bookmarks(s       r c CsHtdfdfdd}|dd,d.d0d2gd4ggdd5id d6igd8d:d=d@gdCgg|ddEggigdGggdd|ddIggigdKggdd|dggigggddgd|dggigggddgd|ddMdOggd dPigdSdUggdd|d!dWggd dXigd[ggdd|d"d]d_ggd d`igdcggdd%gd|d°gd dfigdiggd)d*dS)jNz Fc Sst0}t||dj}|r&t|jt|j} WdQRXt|t|ksPtx"t ||D]\} } | | ks\tq\Wdd|jD|kstdd|jD|kst| |kstdS)N)r,r-cSsg|] }|jqSr)r)r>rDrrrr@sz4test_links..assert_links..cSsg|] }|jqSr)r)r>rDrrrr@s) rrrrrlistZ resolve_linksrrFzip) r<Zexpected_links_by_pageZexpected_anchors_by_pageZexpected_resolved_linksr-warningsrrrZresolved_linksmessageZexpectedrrr assert_linkss z test_links..assert_linksan

Hello, World

externalhttp://weasyprint.orgrrinternallipsumrr hellorru +http://weasyprint.org/foo/lipsum/%C3%A9_%E9rzhttp://weasyprint.org/foo/bar/)r-u
zz2WARNING: Relative URI reference without a base URI)r-rz)
znWARNING: Ignored `-weasy-link: url(../lipsum)` at 1:1, Relative URI reference without a base URI: '../lipsum'.z http://weasyprint.org/z
z missingz6WARNING: No anchor #missing for internal URI referencez rrT)rrrrr)rrrrrrr)rrr rrrr)rrr!rrrr)rrr"rrrr)rrr#)rr)rrrrrr)rrr$rrrr)rrr%rrrrrrr)rr&r'rrrrrrr)rr(r)rrrrrrr)rr*r+rrrr)rrr,rrrr)rrr-rrrr)rrr.rrrr)rrr/rrrr)rrr0rrrr)rrr1)rrrrrrrrr)rr2r3rrrr)rrr4rrrr)rrr5)rrrrrrrrr)rr6r7rrrr)rrr8rrrr)rrr9)rrrrrrrrr)rr:r;rrrr)rrr<)rrrrrrrrr)rr=r>)r)rrrr test_linkss~     r?cs^gfdd}|t|pid}djtj||}tdksDtd\}}|t||fS)Ncsj||fdS)N)ru)statusheaders)start_response_callsrrstart_response;sz#wsgi_client..start_response)Z PATH_INFOZ QUERY_STRINGrcrr)rr2rZapprrFdict)Z path_infoZqs_argsrCenvironresponser@rAr)rBr wsgi_client8s  rGc Cst}td\}}}|dks$t|ddks4t|ttjksFttd\}}}|dks`ttd\}}}|jd}|dkst|djd std |kstd |kstd |ksttj j |d }t |dt |}xtd|tdd|igD]z\}}}|jd}|dkst|djd s,td |ks:td|ksHtd|ksVtd|ksdtd|kstqWtd|\}}}|dkst|ddkst|jdstd|kstdj d}d|d|kstWdQRXdS)Nz /favicon.icoz200 OKz Content-Typez image/x-iconz/lipsumz 404 Not Found/r/z text/html;z#WeasyPrint Navigatorzz test.htmls

Lorem ipsum

bar

z/view/r)z z href="#foo">z( href="/view/http://weasyprint.org">z/pdf/zapplication/pdfs%PDFs;/A << /Type /Action /S /URI /URI (http://weasyprint.org) >>uLorem ipsumz utf-16-bes << /Title (s) /A << /Type /Action /S /GoTo)r rGrFr$rZFAVICONr4rHrrr2r(rr)rr@rArBr"r)rrrrtest_navigatorEsL            rIzweasyprint-customc sttdfddtdtdddfdd }|d |d |d |d |d|dt}|dddWdQRXt|dkst|djdstdd}td|djdS)Nz pattern.pngcs4|dkrtddS|dkr(tdddSt|SdS)Nz(weasyprint-custom:foo/%C3%A9_%e9_patternz image/png)r, mime_typezweasyprint-custom:foo/bar.cssu&body { background: url(é_%e9_pattern)ztext/css)rDr)r)) pattern_pngrrfetchers  z!test_url_fetcher..fetcherz dummy.htmlzi @page { size: 8px; margin: 2px; background: #fff } body { margin: 0; font-size: 0 } )r,r-Fcs(t|d}t|jgd|ddS)N)r, url_fetcherr-)rq)rl)rrpr)r<rl)r-rYrLrrtestsztest_url_fetcher..testzu6uDuO
  • z@z=z T)rlrrz1WARNING: Failed to load image at "custom:foo/bar"cSs|dks ttdddS)Nz weasyprint-custom:%C3%A9_%e9.cssrztext/css)r,rJ)rFrD)r)rrr fetcher_2s z#test_url_fetcher..fetcher_2u?)r,rM)F) r$rrrrrFrHrr)rNrrOr)r-rYrLrKrtest_url_fetcher{s(    rPc CsJdd}|d|dddgddd d d d gd ddd|ddddgddS)Nc[s|jdd|jdg|jdg|jdd|jdd|jdd|jdd|jdgtt|d jj|ks|tdS) Ntitleauthorskeywords generator descriptioncreatedmodified attachments)r,) setdefaultvarsrrmetadatarF)r<metarrr assert_metas        z#test_html_meta..assert_metazu Test document

    Another title

    z I Me & Myselfz Smith, Johnz Test documentuHuman after allr<rYZpdfz Python; cairouBlah… z2011-04Z2013)rRrQrTrSrUrVrWz One Two Three ZOnerZMe)rQrRr)r]rrrtest_html_metas  r^c sddtfdddddddZ}t|djjd d ksFtt|d jjd d ksbtt|d jjd d ks~tWdQRXdS) NcSs0tj}tj|dd}|j||j|jS)Nr%)fileobjmode)rhrigzipZGzipFiler&closerx)rwr*Z gzip_filerrr gzip_compresss  z test_http..gzip_compresscs&d|jddkrddgfSdgfS)NraHTTP_ACCEPT_ENCODINGrsContent-Encodings')rera)r3)env)rcrrsztest_http..cSs(d|jddkr tjddgfSdgfS)NdeflaterdrsContent-Encodings')rirh)r3zlibcompress)rfrrrrgs cSs0d|jddkr(tjddd d gfSdgfS) Nrhrdrsr rContent-Encodings'r)rlrh)r3rjrk)rfrrrrgs()z/gzipz/deflatez /raw-deflatez/gziprNokz/deflatez /raw-deflate)r rrEr3rF)Zroot_urlr)rcr test_https rn)FFF)N)A__doc__ __future__rrrrhrr contextlib threadingrarjZ lxml.htmlrOZ lxml.etreeZ cairocffirer5Z testing_utilsrrrrr r rdr compatr rrrZurlsrrrrrrrrrLockrcontextmanagerrr$r(r;rRr\rprrrrrr r?rGrIrurPr^rnrrrr sX        !  @ J v * L   3  , 4