1 files changed, 33 insertions, 0 deletions
diff --git a/python/tests/test_html.py b/python/tests/test_html.py
new file mode 100644
index 0000000..9a81852
--- /dev/null
+++ b/python/tests/test_html.py
@@ -0,0 +1,33 @@
+
+import json
+import pytest
+import responses
+
+from sandcrawler.html import extract_fulltext_url
+
+def test_extract_fulltext_url():
+
+    resp = extract_fulltext_url("asdf", b"asdf")
+    assert resp == {}
+
+    resp = extract_fulltext_url(
+        "http://dummy-site/",
+        b"""<html>
+        <head>
+          <meta name="citation_pdf_url" content="http://www.example.com/content/271/20/11761.full.pdf">
+        </head>
+        <body>
+        <h1>my big article here</h1>
+        blah
+        </body>
+        </html>"""
+    )
+    assert resp['pdf_url'] == "http://www.example.com/content/271/20/11761.full.pdf"
+    assert resp['technique'] == "citation_pdf_url"
+
+    with open('tests/files/plos_one_article.html', 'rb') as f:
+        resp = extract_fulltext_url(
+            "https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0213978",
+            f.read(),
+        )
+    assert resp['pdf_url'] == "https://journals.plos.org/plosone/article/file?id=10.1371/journal.pone.0213978&type=printable"