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_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]
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_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