Python SubawardsViewSetの例

コード例 #1

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

コード例 #2

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

コード例 #3

ファイル: test_subawards.py プロジェクト: 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

コード例 #4

ファイル: test_subawards.py プロジェクト: 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

コード例 #5

ファイル: test_subawards.py プロジェクト: 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

コード例 #6

ファイル: test_subawards.py プロジェクト: 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]

コード例 #7

ファイル: test_subawards.py プロジェクト: 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]

コード例 #8

from django.conf.urls import url
from usaspending_api.awards.v2.views.subawards import SubawardsViewSet

urlpatterns = [
    url(r'^$', SubawardsViewSet.as_view())
]

コード例 #9

from django.conf.urls import url
from usaspending_api.awards.v2.views.subawards import SubawardsViewSet

urlpatterns = [url(r"^$", SubawardsViewSet.as_view())]