From dc9cd47afcdff353121f6ac3954ded259a6c234a Mon Sep 17 00:00:00 2001
From: Daniel Schadt <kingdread@gmx.de>
Date: Thu, 12 Nov 2015 21:51:33 +0100
Subject: [PATCH] pdfjs: add unittest for script generation

---
 tests/unit/browser/test_webpage.py | 37 ++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)
 create mode 100644 tests/unit/browser/test_webpage.py

diff --git a/tests/unit/browser/test_webpage.py b/tests/unit/browser/test_webpage.py
new file mode 100644
index 000000000..660aa3398
--- /dev/null
+++ b/tests/unit/browser/test_webpage.py
@@ -0,0 +1,37 @@
+# vim: ft=python fileencoding=utf-8 sts=4 sw=4 et:
+
+# Copyright 2015 Daniel Schadt
+#
+# This file is part of qutebrowser.
+#
+# qutebrowser is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# qutebrowser 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 General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with qutebrowser.  If not, see <http://www.gnu.org/licenses/>.
+
+"""Tests for browser.webpage."""
+
+import textwrap
+
+from qutebrowser.browser import webpage
+
+
+def test_generate_pdfjs_script():
+    data = b'\x00foobar\xFF'
+    expected = textwrap.dedent("""
+        var data = new Uint8Array([
+        0,102,111,111,98,97,114,255,]);
+        PDFJS.getDocument(data).then(function(pdf) {
+            PDFView.load(pdf);
+        });
+    """)
+    actual = webpage._generate_pdfjs_script(data)
+    assert actual.strip() == expected.strip()