knock/tests/tests.py

#!/usr/bin/env python3

from bs4 import BeautifulSoup
from urllib.parse import urlparse
from pathlib import Path
import requests, sys, subprocess, shutil, os

knock = Path("./result/bin/knock")

if not knock.exists():
    print("error: " + str(knock) + " does not exist", file=sys.stderr)
    sys.exit()

if len(sys.argv) != 2:
    print(
        "error: missing required argument: directory in which to perform the tests",
        file=sys.stderr
    )
    sys.exit()

print("Testing " + str(knock))

result = subprocess.run(knock)
if result.returncode != 0:
    print("Test failed: knock failed to describe itself")
    sys.exit()
print("---")

workspace = Path(sys.argv[1])
if workspace.exists():
    shutil.rmtree(workspace)
workspace.mkdir()

html = requests \
    .get("https://www.adobe.com/solutions/ebook/digital-editions/sample-ebook-library.html") \
    .text
soup = BeautifulSoup(html, 'html.parser')

links = []
for a_tag in soup.find_all('a'):
    if a_tag.string != "Download eBook":
        continue
    if not urlparse(a_tag.get("href")).path.endswith(".acsm"):
        continue
    
    links.append(a_tag.get("href"))

for i, link in enumerate(links):
    i = str(i)
    print("Testing URL #" + i + ":\n" + link)
    file = workspace.joinpath(i + ".acsm")

    r = requests.get(link)
    open(file, "wb").write(r.content)

    result = subprocess.run([knock, file])

    if result.returncode != 0:
        print("Test failed: knock failed to convert a file")
        sys.exit()

    print("Success\n---")

print("All tests passed")