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
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
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
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
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
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]
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]
from django.conf.urls import url
from usaspending_api.awards.v2.views.subawards import SubawardsViewSet
urlpatterns = [
url(r'^$', SubawardsViewSet.as_view())
]
from django.conf.urls import url from usaspending_api.awards.v2.views.subawards import SubawardsViewSet urlpatterns = [url(r"^$", SubawardsViewSet.as_view())]