Blob Blame History Raw
import tempfile
import filecmp
import contextlib
import pathlib
import sys
import subprocess
import time

from org_fedora_oscap import data_fetch


PORT = 8001


@contextlib.contextmanager
def serve_directory_in_separate_process(port):
    args = [sys.executable, "-m", "http.server", str(port)]
    proc = subprocess.Popen(
        args,
        stdout=subprocess.DEVNULL,
        stderr=subprocess.DEVNULL)
    # give the server some time to start
    time.sleep(0.4)
    yield
    proc.terminate()
    proc.wait()


def test_file_retreival():
    filename_to_test = pathlib.Path(__file__)
    relative_filename_to_test = filename_to_test.relative_to(pathlib.Path.cwd())

    temp_file = tempfile.NamedTemporaryFile()
    temp_filename = temp_file.name

    with serve_directory_in_separate_process(PORT):
        data_fetch._curl_fetch(
            "http://localhost:{}/{}".format(PORT, relative_filename_to_test), temp_filename)

    assert filecmp.cmp(relative_filename_to_test, temp_filename)


def test_supported_url():
    assert data_fetch.can_fetch_from("http://example.com")
    assert data_fetch.can_fetch_from("https://example.com")


def test_unsupported_url():
    assert not data_fetch.can_fetch_from("aaaaa")