Exemplos de SubawardsViewSet._business_logic em Python

Exemplo n.º 1

def test_all_subawards(mock_matviews_qs):
    mock_model_1 = MockModel(**subaward_1)
    mock_model_2 = MockModel(**subaward_2)
    mock_model_3 = MockModel(**subaward_3)

    add_to_mock_objects(mock_matviews_qs, [mock_model_1, mock_model_2, mock_model_3])

    test_payload = {
        "page": 1,
        "limit": 10,
        "order": "asc",
    }
    svs = SubawardsViewSet()
    test_params = svs._parse_and_validate_request(test_payload)
    subawards_logic = svs._business_logic(test_params)

    expected_response = [strip_award_id(subaward_1), strip_award_id(subaward_2), strip_award_id(subaward_3)]

    assert expected_response == subawards_logic

    test_payload['page'] = 2
    test_params = svs._parse_and_validate_request(test_payload)
    subawards_logic = svs._business_logic(test_params)
    assert [] == subawards_logic

    test_payload = {
        "order": "desc",
    }
    test_params = svs._parse_and_validate_request(test_payload)
    subawards_logic = svs._business_logic(test_params)
    assert [strip_award_id(subaward_3), strip_award_id(subaward_2), strip_award_id(subaward_1)] == subawards_logic

Exemplo n.º 2

Arquivo: test_subawards.py Projeto: Android-Recommended-Enterprise/usaspending-api

def test_all_subawards(mock_matviews_qs):
    mock_model_1 = MockModel(**subaward_1)
    mock_model_2 = MockModel(**subaward_2)
    mock_model_3 = MockModel(**subaward_3)

    add_to_mock_objects(mock_matviews_qs, [mock_model_1, mock_model_2, mock_model_3])

    test_payload = {"page": 1, "limit": 10, "order": "asc"}
    svs = SubawardsViewSet()
    test_params = svs._parse_and_validate_request(test_payload)
    subawards_logic = svs._business_logic(test_params)

    expected_response = [strip_award_id(subaward_1), strip_award_id(subaward_2), strip_award_id(subaward_3)]

    assert expected_response == subawards_logic

    test_payload["page"] = 2
    test_params = svs._parse_and_validate_request(test_payload)
    subawards_logic = svs._business_logic(test_params)
    assert [] == subawards_logic

    sub_1 = strip_award_id(subaward_1)
    sub_2 = strip_award_id(subaward_2)
    sub_3 = strip_award_id(subaward_3)

    assert request_with_sort("id") == [sub_3, sub_1, sub_2]
    assert request_with_sort("amount") == [sub_3, sub_2, sub_1]
    assert request_with_sort("action_date") == [sub_2, sub_1, sub_3]
    assert request_with_sort("recipient_name") == [sub_2, sub_3, sub_1]

Exemplo n.º 3

Arquivo: test_subawards.py Projeto: umeshh/usaspending-api

def test_all_subawards():
    create_subaward_test_data(subaward_1, subaward_2, subaward_3)

    test_payload = {"page": 1, "limit": 10, "order": "asc"}
    svs = SubawardsViewSet()
    test_params = svs._parse_and_validate_request(test_payload)
    subawards_logic = svs._business_logic(test_params)

    expected_response = [
        strip_award_id(subaward_1),
        strip_award_id(subaward_2),
        strip_award_id(subaward_3)
    ]

    assert expected_response == subawards_logic

    test_payload["page"] = 2
    test_params = svs._parse_and_validate_request(test_payload)
    subawards_logic = svs._business_logic(test_params)
    assert [] == subawards_logic

    sub_1 = strip_award_id(subaward_1)
    sub_2 = strip_award_id(subaward_2)
    sub_3 = strip_award_id(subaward_3)

    assert request_with_sort("id") == [sub_3, sub_1, sub_2]
    assert request_with_sort("amount") == [sub_3, sub_2, sub_1]
    assert request_with_sort("action_date") == [sub_2, sub_1, sub_3]
    assert request_with_sort("recipient_name") == [sub_2, sub_3, sub_1]

Exemplo n.º 4

def request_with_sort(sort):
    svs = SubawardsViewSet()
    test_payload = {
        "page": 1,
        "limit": 4,
        "sort": sort,
        "order": "desc",
    }
    test_params = svs._parse_and_validate_request(test_payload)
    subawards_logic = svs._business_logic(test_params)
    return subawards_logic

Exemplo n.º 5

Arquivo: test_subawards.py Projeto: umeshh/usaspending-api

def test_specific_award():
    create_subaward_test_data(subaward_10, subaward_11, subaward_12)

    test_payload = {"award_id": 99}

    svs = SubawardsViewSet()
    test_params = svs._parse_and_validate_request(test_payload)
    subawards_logic = svs._business_logic(test_params)

    expected_response = [
        strip_award_id(subaward_11),
        strip_award_id(subaward_10)
    ]

    assert expected_response == subawards_logic

Exemplo n.º 6

Arquivo: test_subawards.py Projeto: Android-Recommended-Enterprise/usaspending-api

def test_specific_award(mock_matviews_qs):
    mock_model_1 = MockModel(**subaward_10)
    mock_model_2 = MockModel(**subaward_11)
    mock_model_3 = MockModel(**subaward_12)

    add_to_mock_objects(mock_matviews_qs, [mock_model_1, mock_model_2, mock_model_3])

    test_payload = {"award_id": 99}

    svs = SubawardsViewSet()
    test_params = svs._parse_and_validate_request(test_payload)
    subawards_logic = svs._business_logic(test_params)

    expected_response = [strip_award_id(subaward_11), strip_award_id(subaward_10)]

    assert expected_response == subawards_logic

Exemplo n.º 7

Arquivo: test_subawards.py Projeto: fedspendingtransparency/usaspending-api

def test_specific_award(mock_matviews_qs):
    mock_model_1 = MockModel(**subaward_10)
    mock_model_2 = MockModel(**subaward_11)
    mock_model_3 = MockModel(**subaward_12)

    add_to_mock_objects(mock_matviews_qs, [mock_model_1, mock_model_2, mock_model_3])

    test_payload = {
        "award_id": 99
    }

    svs = SubawardsViewSet()
    test_params = svs._parse_and_validate_request(test_payload)
    subawards_logic = svs._business_logic(test_params)

    expected_response = [strip_award_id(subaward_11), strip_award_id(subaward_10)]

    assert expected_response == subawards_logic