ssbm-oro / racetime-obs

tests/test_qualifier.py

from datetime import timedelta
from gadgets.qualifier import Qualifier
from users_for_testing import get_test_entrant, get_test_entrants
from races_for_testing import get_test_race, time_ago


def get_test_qualifier():
    qualifier = Qualifier()
    qualifier.enabled = True
    qualifier.qualifier_cutoff = 3
    qualifier.par_source = "par source"
    qualifier.score_source = "score source"
    return qualifier


def test_no_one_finished(random_users):
    entrant = get_test_entrant(next(random_users), status_value="in_progress")
    entrants = [entrant]
    for i in range(0, 20):
        entrants.append(get_test_entrant(next(random_users)))
    race = get_test_race(entrants=entrants)
    qualifier = get_test_qualifier()
    qualifier.update_qualifier_text(race, entrant.user.full_name)
    assert qualifier.entrant_score == " "
    assert qualifier.par_text == " "


def test_some_finished(random_users):
    entrant = get_test_entrant(next(random_users), status_value="in_progress")
    first_place = get_test_entrant(
        next(random_users), status_value="finished",
        finished_at=time_ago(minutes=20),
        finish_time=timedelta(hours=1, minutes=30, microseconds=1), place=1,
    )
    second_place = get_test_entrant(
        next(random_users), status_value="finished",
        finished_at=time_ago(minutes=15),
        finish_time=timedelta(hours=1, minutes=35, microseconds=1), place=2,
    )
    entrants = (
        get_test_entrants(random_users, first_place, second_place)
    )
    race = get_test_race(entrants=entrants)
    qualifier = get_test_qualifier()
    qualifier.update_qualifier_text(race, entrant.user.full_name)
    assert qualifier.entrant_score == " "
    assert qualifier.par_text == " "


def test_cutoff_finished(random_users):
    entrant = get_test_entrant(next(random_users), status_value="in_progress")
    first_place = get_test_entrant(
        next(random_users), status_value="finished",
        finished_at=time_ago(minutes=20),
        finish_time=timedelta(hours=1, minutes=30, microseconds=1), place=1,
    )
    second_place = get_test_entrant(
        next(random_users), status_value="finished",
        finished_at=time_ago(minutes=15),
        finish_time=timedelta(hours=1, minutes=35, microseconds=1), place=2,
    )
    third_place = get_test_entrant(
        next(random_users), status_value="finished",
        finished_at=time_ago(minutes=10),
        finish_time=timedelta(hours=1, minutes=40, microseconds=1), place=3
    )
    entrants = get_test_entrants(
        random_users, first_place, second_place, third_place, entrant)
    race = get_test_race(entrants=entrants)
    qualifier = get_test_qualifier()
    qualifier.update_qualifier_text(race, entrant.user.full_name)
    assert qualifier.entrant_score == " "
    assert qualifier.par_text == "1:35:00.0"


def test_cutoff_and_entrant_finished(random_users):
    entrant = get_test_entrant(
        next(random_users), status_value="finished",
        finished_at=time_ago(minutes=5),
        finish_time=timedelta(hours=1, minutes=45, microseconds=1)
    )
    first_place = get_test_entrant(
        next(random_users), status_value="finished",
        finished_at=time_ago(minutes=20),
        finish_time=timedelta(hours=1, minutes=30, microseconds=1), place=1
    )
    second_place = get_test_entrant(
        next(random_users), status_value="finished",
        finished_at=time_ago(minutes=15),
        finish_time=timedelta(hours=1, minutes=35, microseconds=1), place=2
    )
    third_place = get_test_entrant(
        next(random_users), status_value="finished",
        finished_at=time_ago(minutes=10),
        finish_time=timedelta(hours=1, minutes=40, microseconds=1), place=3,
    )
    entrants = get_test_entrants(
        random_users, first_place, second_place, third_place, entrant)
    race = get_test_race(entrants=entrants)
    qualifier = get_test_qualifier()
    qualifier.update_qualifier_text(race, entrant.user.full_name)
    # par is 95 minutes, entrant's time is 105 minutes. 2-(105/95) ~= 0.89
    assert qualifier.entrant_score == "0.89"
    assert qualifier.par_text == "1:35:00.0"


def test_cutoff_and_entrant_finished_in_top(random_users):

This code seems to be duplicated in your project.

    entrant = get_test_entrant(
        next(random_users), status_value="finished",
        finished_at=time_ago(minutes=15),
        finish_time=timedelta(hours=1, minutes=35, microseconds=1), place=2
    )
    first_place = get_test_entrant(
        next(random_users), status_value="finished",
        finished_at=time_ago(minutes=20),
        finish_time=timedelta(hours=1, minutes=30, microseconds=1), place=1,
    )
    third_place = get_test_entrant(
        next(random_users), status_value="finished",
        finished_at=time_ago(minutes=10),
        finish_time=timedelta(hours=1, minutes=40, microseconds=1), place=3,
    )
    entrants = get_test_entrants(
        random_users, first_place, third_place, entrant)
    race = get_test_race(entrants=entrants)
    qualifier = get_test_qualifier()
    qualifier.update_qualifier_text(race, entrant.user.full_name)
    # par is 95 minutes, entrant's time is 95 minutes. 2-(95/95) = 1.00
    assert qualifier.entrant_score == "1.00"
    assert qualifier.par_text == "1:35:00.0"


def test_cutoff_and_entrant_finished_in_first(random_users):

This code seems to be duplicated in your project.

    entrant = get_test_entrant(
        next(random_users), status_value="finished",
        finished_at=time_ago(minutes=20),
        finish_time=timedelta(hours=1, minutes=30, microseconds=1), place=2
    )
    second_place = get_test_entrant(
        next(random_users), status_value="finished",
        finished_at=time_ago(minutes=15),
        finish_time=timedelta(hours=1, minutes=35, microseconds=1), place=1
    )
    third_place = get_test_entrant(
        next(random_users), status_value="finished",
        finished_at=time_ago(minutes=10),
        finish_time=timedelta(hours=1, minutes=40, microseconds=1), place=3
    )
    entrants = get_test_entrants(
        random_users, entrant, second_place, third_place)
    race = get_test_race(entrants=entrants)
    qualifier = get_test_qualifier()
    qualifier.update_qualifier_text(race, entrant.user.full_name)
    # par is 95 minutes, entrant's time is 90 minutes. 2-(90/95) ~= 1.05
    assert qualifier.entrant_score == "1.05"
    assert qualifier.par_text == "1:35:00.0"


def test_cutoff_and_entrant_finished_waay_ahead(random_users):

This code seems to be duplicated in your project.

    entrant = get_test_entrant(
        next(random_users), status_value="finished",
        finished_at=time_ago(minutes=20),
        finish_time=timedelta(hours=1, microseconds=1), place=2
    )
    second_place = get_test_entrant(
        next(random_users), status_value="finished",
        finished_at=time_ago(minutes=15),
        finish_time=timedelta(hours=1, minutes=35, microseconds=1), place=1
    )
    third_place = get_test_entrant(
        next(random_users), status_value="finished",
        finished_at=time_ago(minutes=10),
        finish_time=timedelta(hours=1, minutes=40, microseconds=1), place=3
    )
    entrants = get_test_entrants(
        random_users, entrant, second_place, third_place)
    race = get_test_race(entrants=entrants)
    qualifier = get_test_qualifier()
    qualifier.update_qualifier_text(race, entrant.user.full_name)
    # par is 95 minutes, entrant's time is 60 minutes. 2-(60/85) ~= 1.29
    # but max score is 1.05
    assert qualifier.entrant_score == "1.05"
    assert qualifier.par_text == "1:25:00.0"