Example #1
0
def test_args_handler():
    args = argparse.Namespace(version=True)
    reply = cli.args_handler(args)
    assert ("waybackpy version %s" % (__version__)) == reply

    args = argparse.Namespace(url=None, version=False)
    reply = cli.args_handler(args)
    assert ("waybackpy %s" % (__version__)) in str(reply)
Example #2
0
def test_near():
    args = argparse.Namespace(
        user_agent=
        "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_10_5) AppleWebKit/600.8.9 \
    (KHTML, like Gecko) Version/8.0.8 Safari/600.8.9",
        url="https://pypi.org/user/akamhy/",
        total=False,
        version=False,
        file=False,
        oldest=False,
        save=False,
        json=False,
        archive_url=False,
        newest=False,
        near=True,
        subdomain=False,
        known_urls=False,
        get=None,
        year=2020,
        month=7,
        day=15,
        hour=1,
        minute=1,
    )
    reply = cli.args_handler(args)
    assert "202007" in str(reply)

    uid = "".join(
        random.choice(string.ascii_lowercase + string.digits)
        for _ in range(6))
    url = "https://pypi.org/yfvjvycyc667r67ed67r" + uid
    args = argparse.Namespace(
        user_agent=None,
        url=url,
        total=False,
        version=False,
        file=False,
        oldest=False,
        save=False,
        json=False,
        archive_url=False,
        newest=False,
        near=True,
        subdomain=False,
        known_urls=False,
        get=None,
        year=2020,
        month=7,
        day=15,
        hour=1,
        minute=1,
    )
    reply = cli.args_handler(args)
    assert "Can not find archive for" in str(reply)
Example #3
0
def test_near():
    args = argparse.Namespace(
        user_agent=
        "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_10_5) AppleWebKit/600.8.9 \
    (KHTML, like Gecko) Version/8.0.8 Safari/600.8.9",
        url="https://pypi.org/user/akamhy/",
        total=False,
        version=False,
        oldest=False,
        save=False,
        json=False,
        archive_url=False,
        newest=False,
        near=True,
        alive=False,
        subdomain=False,
        known_urls=False,
        get=None,
        year=2020,
        month=7,
        day=15,
        hour=1,
        minute=1)
    reply = cli.args_handler(args)
    assert "202007" in str(reply)
Example #4
0
def test_oldest():
    args = argparse.Namespace(
        user_agent=None,
        url="https://pypi.org/user/akamhy/",
        total=False,
        version=False,
        file=False,
        oldest=True,
        save=False,
        json=False,
        archive_url=False,
        newest=False,
        near=False,
        subdomain=False,
        known_urls=False,
        get=None,
    )
    reply = cli.args_handler(args)
    assert "pypi.org/user/akamhy" in str(reply)

    uid = "".join(
        random.choice(string.ascii_lowercase + string.digits)
        for _ in range(6))
    url = "https://pypi.org/yfvjvycyc667r67ed67r" + uid
    args = argparse.Namespace(
        user_agent=None,
        url=url,
        total=False,
        version=False,
        file=False,
        oldest=True,
        save=False,
        json=False,
        archive_url=False,
        newest=False,
        near=False,
        subdomain=False,
        known_urls=False,
        get=None,
    )
    reply = cli.args_handler(args)
    assert "Can not find archive for" in str(reply)
Example #5
0
def test_oldest():
    args = argparse.Namespace(user_agent=None,
                              url="https://pypi.org/user/akamhy/",
                              total=False,
                              version=False,
                              oldest=True,
                              save=False,
                              newest=False,
                              near=False,
                              alive=False,
                              subdomain=False,
                              known_urls=False,
                              get=None)
    reply = cli.args_handler(args)
    assert "pypi.org/user/akamhy" in reply
Example #6
0
def test_json():
    args = argparse.Namespace(user_agent=None,
                              url="https://pypi.org/user/akamhy/",
                              total=False,
                              version=False,
                              oldest=False,
                              save=False,
                              json=True,
                              archive_url=False,
                              newest=False,
                              near=False,
                              alive=False,
                              subdomain=False,
                              known_urls=False,
                              get=None)
    reply = cli.args_handler(args)
    assert "archived_snapshots" in str(reply)
Example #7
0
def test_known_urls():
    args = argparse.Namespace(
        user_agent=
        "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_10_5) AppleWebKit/600.8.9 \
    (KHTML, like Gecko) Version/8.0.8 Safari/600.8.9",
        url="https://akamhy.github.io",
        total=False,
        version=False,
        oldest=False,
        save=False,
        newest=False,
        near=False,
        alive=True,
        subdomain=True,
        known_urls=True,
        get=None)
    reply = cli.args_handler(args)
    assert "github" in reply
Example #8
0
def test_total_archives():
    args = argparse.Namespace(
        user_agent=
        "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_10_5) AppleWebKit/600.8.9 \
    (KHTML, like Gecko) Version/8.0.8 Safari/600.8.9",
        url="https://pypi.org/user/akamhy/",
        total=True,
        version=False,
        oldest=False,
        save=False,
        newest=False,
        near=False,
        alive=False,
        subdomain=False,
        known_urls=False,
        get=None)
    reply = cli.args_handler(args)
    assert isinstance(reply, int)
Example #9
0
def test_known_urls():
    args = argparse.Namespace(
        user_agent="Mozilla/5.0 (Macintosh; Intel Mac OS X 10_10_5) AppleWebKit/600.8.9 \
    (KHTML, like Gecko) Version/8.0.8 Safari/600.8.9",
        url="https://www.keybr.com",
        total=False,
        version=False,
        file=True,
        oldest=False,
        save=False,
        json=False,
        archive_url=False,
        newest=False,
        near=False,
        subdomain=False,
        known_urls=True,
        get=None,
    )
    reply = cli.args_handler(args)
    assert "keybr" in str(reply)
Example #10
0
def test_save():

    args = argparse.Namespace(
        user_agent=None,
        url="https://hfjfjfjfyu6r6rfjvj.fjhgjhfjgvjm",
        total=False,
        version=False,
        file=False,
        oldest=False,
        save=True,
        json=False,
        archive_url=False,
        newest=False,
        near=False,
        subdomain=False,
        known_urls=False,
        get=None,
    )
    reply = cli.args_handler(args)
    assert "could happen because either your waybackpy" in str(reply)
Example #11
0
def test_get():
    args = argparse.Namespace(
        user_agent=
        "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_10_5) AppleWebKit/600.8.9 \
    (KHTML, like Gecko) Version/8.0.8 Safari/600.8.9",
        url="https://github.com/akamhy",
        total=False,
        version=False,
        file=False,
        oldest=False,
        save=False,
        json=False,
        archive_url=False,
        newest=False,
        near=False,
        subdomain=False,
        known_urls=False,
        get="url",
    )
    reply = cli.args_handler(args)
    assert "waybackpy" in str(reply)

    args = argparse.Namespace(
        user_agent=
        "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_10_5) AppleWebKit/600.8.9 \
    (KHTML, like Gecko) Version/8.0.8 Safari/600.8.9",
        url="https://github.com/akamhy/waybackpy",
        total=False,
        version=False,
        file=False,
        oldest=False,
        save=False,
        json=False,
        archive_url=False,
        newest=False,
        near=False,
        subdomain=False,
        known_urls=False,
        get="oldest",
    )
    reply = cli.args_handler(args)
    assert "waybackpy" in str(reply)

    args = argparse.Namespace(
        user_agent=
        "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_10_5) AppleWebKit/600.8.9 \
    (KHTML, like Gecko) Version/8.0.8 Safari/600.8.9",
        url="https://akamhy.github.io/waybackpy/",
        total=False,
        version=False,
        file=False,
        oldest=False,
        save=False,
        json=False,
        archive_url=False,
        newest=False,
        near=False,
        subdomain=False,
        known_urls=False,
        get="newest",
    )
    reply = cli.args_handler(args)
    assert "waybackpy" in str(reply)

    args = argparse.Namespace(
        user_agent=
        "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_10_5) AppleWebKit/600.8.9 \
    (KHTML, like Gecko) Version/8.0.8 Safari/600.8.9",
        url="https://pypi.org/user/akamhy/",
        total=False,
        version=False,
        file=False,
        oldest=False,
        save=False,
        json=False,
        archive_url=False,
        newest=False,
        near=False,
        subdomain=False,
        known_urls=False,
        get="foobar",
    )
    reply = cli.args_handler(args)
    assert "get the source code of the" in str(reply)
Example #12
0
def test_get():
    args = argparse.Namespace(
        user_agent=
        "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_10_5) AppleWebKit/600.8.9 \
    (KHTML, like Gecko) Version/8.0.8 Safari/600.8.9",
        url="https://pypi.org/user/akamhy/",
        total=False,
        version=False,
        oldest=False,
        save=False,
        newest=False,
        near=False,
        alive=False,
        subdomain=False,
        known_urls=False,
        get="url")
    reply = cli.args_handler(args)
    assert "waybackpy" in reply

    args = argparse.Namespace(
        user_agent=
        "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_10_5) AppleWebKit/600.8.9 \
    (KHTML, like Gecko) Version/8.0.8 Safari/600.8.9",
        url="https://pypi.org/user/akamhy/",
        total=False,
        version=False,
        oldest=False,
        save=False,
        newest=False,
        near=False,
        alive=False,
        subdomain=False,
        known_urls=False,
        get="oldest")
    reply = cli.args_handler(args)
    assert "waybackpy" in reply

    args = argparse.Namespace(
        user_agent=
        "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_10_5) AppleWebKit/600.8.9 \
    (KHTML, like Gecko) Version/8.0.8 Safari/600.8.9",
        url="https://pypi.org/user/akamhy/",
        total=False,
        version=False,
        oldest=False,
        save=False,
        newest=False,
        near=False,
        alive=False,
        subdomain=False,
        known_urls=False,
        get="newest")
    reply = cli.args_handler(args)
    assert "waybackpy" in reply

    if codecov_python:
        args = argparse.Namespace(
            user_agent=
            "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_10_5) AppleWebKit/600.8.9 \
        (KHTML, like Gecko) Version/8.0.8 Safari/600.8.9",
            url="https://pypi.org/user/akamhy/",
            total=False,
            version=False,
            oldest=False,
            save=False,
            newest=False,
            near=False,
            alive=False,
            subdomain=False,
            known_urls=False,
            get="save")
        reply = cli.args_handler(args)
        assert "waybackpy" in reply

    args = argparse.Namespace(
        user_agent=
        "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_10_5) AppleWebKit/600.8.9 \
    (KHTML, like Gecko) Version/8.0.8 Safari/600.8.9",
        url="https://pypi.org/user/akamhy/",
        total=False,
        version=False,
        oldest=False,
        save=False,
        newest=False,
        near=False,
        alive=False,
        subdomain=False,
        known_urls=False,
        get="BullShit")
    reply = cli.args_handler(args)
    assert "get the source code of the" in reply