diff options
| author | Lars-Dominik Braun <lars@6xq.net> | 2019-03-21 15:04:32 +0100 |
|---|---|---|
| committer | Lars-Dominik Braun <lars@6xq.net> | 2019-03-21 15:04:32 +0100 |
| commit | 4643112b8f5f208f360e2e7c9431931b6526131e (patch) | |
| tree | 2f4b00bc53eaef7c343f65796f49e02af0839c93 /crocoite/test_behavior.py | |
| parent | 38078be63466f5cd78ad97963502e47b8435c07f (diff) | |
| download | crocoite-4643112b8f5f208f360e2e7c9431931b6526131e.tar.gz crocoite-4643112b8f5f208f360e2e7c9431931b6526131e.tar.bz2 crocoite-4643112b8f5f208f360e2e7c9431931b6526131e.zip | |
behavior: Test Screenshot
Diffstat (limited to 'crocoite/test_behavior.py')
| -rw-r--r-- | crocoite/test_behavior.py | 60 |
1 files changed, 50 insertions, 10 deletions
diff --git a/crocoite/test_behavior.py b/crocoite/test_behavior.py index bc2dab6..bbbd8ba 100644 --- a/crocoite/test_behavior.py +++ b/crocoite/test_behavior.py @@ -18,8 +18,9 @@ # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN # THE SOFTWARE. -import asyncio, os, yaml, re +import asyncio, os, yaml, re, math, struct from functools import partial +from operator import attrgetter import pytest from yarl import URL @@ -28,7 +29,8 @@ from aiohttp import web import pkg_resources from .logger import Logger from .devtools import Process -from .behavior import Scroll, Behavior, ExtractLinks, ExtractLinksEvent, Crash +from .behavior import Scroll, Behavior, ExtractLinks, ExtractLinksEvent, Crash, \ + Screenshot, ScreenshotEvent from .controller import SinglePageController, EventHandler from .devtools import Crashed @@ -99,17 +101,17 @@ async def test_click_match (match, url): # keep this aligned with click.js assert re.match (match, url.host, re.I) -class ExtractLinksCheck(EventHandler): - """ Test adapter that accumulates all incoming links from ExtractLinks """ - __slots__ = ('links') + +class AccumHandler (EventHandler): + """ Test adapter that accumulates all incoming items """ + __slots__ = ('data') def __init__ (self): super().__init__ () - self.links = [] + self.data = [] def push (self, item): - if isinstance (item, ExtractLinksEvent): - self.links.extend (item.links) + self.data.append (item) async def simpleServer (url, response): async def f (req): @@ -151,13 +153,17 @@ async def test_extract_links (): </body></html>""") try: - handler = ExtractLinksCheck () + handler = AccumHandler () logger = Logger () controller = SinglePageController (url=url, logger=logger, service=Process (), behavior=[ExtractLinks], handler=[handler]) await controller.run () - assert sorted (handler.links) == sorted ([ + links = [] + for d in handler.data: + if isinstance (d, ExtractLinksEvent): + links.extend (d.links) + assert sorted (links) == sorted ([ url.with_path ('/relative'), url.with_fragment ('anchor'), URL ('http://example.com/absolute/'), @@ -186,3 +192,37 @@ async def test_crash (): finally: await runner.cleanup () +@pytest.mark.asyncio +async def test_screenshot (): + """ + Make sure screenshots are taken and have the correct dimensions. We can’t + and don’t want to check their content. + """ + # ceil(0) == 0, so starting with 1 + for expectHeight in (1, Screenshot.maxDim, Screenshot.maxDim+1, Screenshot.maxDim*2+Screenshot.maxDim//2): + url = URL.build (scheme='http', host='localhost', port=8080) + runner = await simpleServer (url, f'<html><body style="margin: 0; padding: 0;"><div style="height: {expectHeight}"></div></body></html>') + + try: + handler = AccumHandler () + logger = Logger () + controller = SinglePageController (url=url, logger=logger, + service=Process (), behavior=[Screenshot], handler=[handler]) + await controller.run () + + screenshots = list (filter (lambda x: isinstance (x, ScreenshotEvent), handler.data)) + assert len (screenshots) == math.ceil (expectHeight/Screenshot.maxDim) + totalHeight = 0 + for s in screenshots: + assert s.url == url + # PNG ident is fixed, IHDR is always the first chunk + assert s.data.startswith (b'\x89PNG\r\n\x1a\n\x00\x00\x00\x0dIHDR') + width, height = struct.unpack ('>II', s.data[16:24]) + assert height <= Screenshot.maxDim + totalHeight += height + # screenshot height is at least canvas height (XXX: get hardcoded + # value from devtools.Process) + assert totalHeight == max (expectHeight, 1080) + finally: + await runner.cleanup () + |