grooveondemand/test/test_webserver.py

import pytest

import os
import sys

import atheris
import bottle
from boddle import boddle

from groove import webserver


def test_server():
    with boddle():
        webserver.index()
        assert bottle.response.status_code == 200


def test_auth_with_valid_credentials():
    with boddle(auth=(os.environ.get('USERNAME'), os.environ.get('PASSWORD'))):
        webserver.build()
        assert bottle.response.status_code == 200


def test_auth_random_input():

    def auth(fuzzed_input):
        with boddle(auth=(fuzzed_input, fuzzed_input)):
            response = webserver.build()
            assert response.status_code == 401

    atheris.Setup([sys.argv[0], "-atheris_runs=100000"], auth)
    try:
        atheris.Fuzz()
    except SystemExit:
        pass


@pytest.mark.parametrize('slug, expected', [
    ('playlist-one', 200),
    ('playlist-two', 200),
    ('playlist-three', 200),
    ('no such playlist', 404),
])
def test_playlist(db, slug, expected):
    with boddle():
        response = webserver.get_playlist(slug, db)
        assert response.status_code == expected


def test_playlist_on_empty_db(in_memory_db):
    with boddle():
        response = webserver.get_playlist('some-slug', in_memory_db)
        assert response.status_code == 404