test_validators.py

import pytest
from osf.exceptions import ValidationValueError

from osf.models import validators
from osf_tests.factories import SubjectFactory

# Ported from tests/framework/test_mongo.py

def test_string_required_passes_with_string():
    assert validators.string_required('Hi!') is True

def test_string_required_fails_when_empty():
    with pytest.raises(ValidationValueError):
        validators.string_required(None)
    with pytest.raises(ValidationValueError):
        validators.string_required('')

@pytest.mark.django_db
def test_validate_expand_subject_hierarchy():
    fruit = SubjectFactory()
    apple = SubjectFactory(parent=fruit)
    grapes = SubjectFactory(parent=fruit)
    raisins = SubjectFactory(parent=grapes)

    # test send in two flattened hierarchies that share a base
    subject_list = [fruit._id, apple._id, grapes._id]
    expanded = validators.expand_subject_hierarchy(subject_list)
    assert len(expanded) == 3
    assert fruit in expanded
    assert apple in expanded
    assert grapes in expanded

    # test send in third level of a 3-tier hierarchy
    subject_list = [raisins._id]
    expanded = validators.expand_subject_hierarchy(subject_list)
    assert len(expanded) == 3
    assert raisins in expanded
    assert grapes in expanded
    assert fruit in expanded

    # test send in first and third levels
    subject_list = [raisins._id, fruit._id]
    expanded = validators.expand_subject_hierarchy(subject_list)
    assert len(expanded) == 3
    assert raisins in expanded
    assert grapes in expanded
    assert fruit in expanded

    # test invalid hierarchy
    subject_list = [fruit._id, '12345_bad_id']
    with pytest.raises(ValidationValueError):
        validators.expand_subject_hierarchy(subject_list)