-
Notifications
You must be signed in to change notification settings - Fork 80
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Feat: Test cases for POST and GET /user/personal_background api endpo…
…ints Refactored Background enum logic Refactor test to include other fields for physical, mental ability and socio economic Remove POST api endpoint for create user personal background Remove Create personal background dao
- Loading branch information
1 parent
803beb4
commit 3ce0bf7
Showing
4 changed files
with
196 additions
and
60 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,193 @@ | ||
| import unittest | ||
| from http import HTTPStatus, cookies | ||
| from unittest.mock import patch, Mock | ||
| import requests | ||
| from requests.exceptions import HTTPError | ||
| from flask import json | ||
| from flask_restx import marshal | ||
| from app import messages | ||
| from tests.base_test_case import BaseTestCase | ||
| from app.api.request_api_utils import post_request, get_request, BASE_MS_API_URL, AUTH_COOKIE | ||
| from app.api.models.user import full_user_api_model, get_user_personal_background_response_model | ||
| from tests.test_data import user1 | ||
| from app.database.models.ms_schema.user import UserModel | ||
| from app.database.models.bit_schema.personal_background import PersonalBackgroundModel | ||
|
|
||
|
|
||
| class TestGetUserPersonalBackgroundApi(BaseTestCase): | ||
|
|
||
| @patch("requests.post") | ||
| def setUp(self, mock_login): | ||
| super(TestGetUserPersonalBackgroundApi, self).setUp() | ||
|
|
||
| success_message = {"access_token": "this is fake token", "access_expiry": 1601478236} | ||
| success_code = HTTPStatus.OK | ||
|
|
||
| mock_login_response = Mock() | ||
| mock_login_response.json.return_value = success_message | ||
| mock_login_response.status_code = success_code | ||
| mock_login.return_value = mock_login_response | ||
| mock_login.raise_for_status = json.dumps(success_code) | ||
|
|
||
| user_login_success = { | ||
| "username": user1.get("username"), | ||
| "password": user1.get("password") | ||
| } | ||
|
|
||
| with self.client: | ||
| login_response = self.client.post( | ||
| "/login", | ||
| data=json.dumps(user_login_success), | ||
| follow_redirects=True, | ||
| content_type="application/json", | ||
| ) | ||
|
|
||
| test_user = UserModel( | ||
| name=user1["name"], | ||
| username=user1["username"], | ||
| password=user1["password"], | ||
| email=user1["email"], | ||
| terms_and_conditions_checked=user1["terms_and_conditions_checked"] | ||
| ) | ||
| test_user.need_mentoring = user1["need_mentoring"] | ||
| test_user.available_to_mentor = user1["available_to_mentor"] | ||
|
|
||
| test_user.save_to_db() | ||
|
|
||
| self.test_user_data = UserModel.find_by_email(test_user.email) | ||
|
|
||
| AUTH_COOKIE["user_id"] = self.test_user_data.id | ||
|
|
||
| self.response_personal_background = { | ||
| "user_id": int(AUTH_COOKIE["user_id"].value), | ||
| "gender": "Female", | ||
| "age": "Between 55 to 64 yo", | ||
| "ethnicity": "Middle Eastern/North African (MENA)", | ||
| "sexual_orientation": "Prefer not to say", | ||
| "religion": "Islam", | ||
| "physical_ability": "With/had limited physical ability (or with/had some type of physical disability/ies)", | ||
| "mental_ability": "With/previously had some type of mental disorders", | ||
| "socio_economic": "Lower Middle class (e.g. blue collars in skilled trades/Paralegals/Bank tellers/Sales/Clerical-Admin/other support workers)", | ||
| "highest_education": "Prefer not to say", | ||
| "years_of_experience": "Prefer not to say", | ||
| "gender_other": "They", | ||
| "ethnicity_other": "", | ||
| "sexual_orientation_other": "", | ||
| "religion_other": "Daoism", | ||
| "physical_ability_other": "", | ||
| "mental_ability_other": "", | ||
| "socio_economic_other": "", | ||
| "highest_education_other": "", | ||
| "is_public": False | ||
| } | ||
|
|
||
|
|
||
| @patch("requests.get") | ||
| def test_api_dao_get_user_personal_background_successfully(self, mock_get_personal_background): | ||
| success_message = self.response_personal_background | ||
| success_code = HTTPStatus.OK | ||
|
|
||
| mock_get_response = Mock() | ||
| mock_get_response.json.return_value = success_message | ||
| mock_get_response.status_code = success_code | ||
|
|
||
| mock_get_personal_background.return_value = mock_get_response | ||
| mock_get_personal_background.raise_for_status = json.dumps(success_code) | ||
|
|
||
| # prepare existing personal background | ||
| others = { | ||
| "gender_other": self.response_personal_background["gender_other"], | ||
| "ethnicity_other": self.response_personal_background["ethnicity_other"], | ||
| "sexual_orientation_other": self.response_personal_background["sexual_orientation_other"], | ||
| "religion_other": self.response_personal_background["religion_other"], | ||
| "physical_ability_other": self.response_personal_background["physical_ability_other"], | ||
| "mental_ability_other": self.response_personal_background["mental_ability_other"], | ||
| "socio_economic_other": self.response_personal_background["socio_economic_other"], | ||
| "highest_education_other": self.response_personal_background["highest_education_other"], | ||
| } | ||
|
|
||
| personal_background = PersonalBackgroundModel( | ||
| int(AUTH_COOKIE["user_id"].value), | ||
| "FEMALE", | ||
| "AGE_55_TO_64", | ||
| "MIDDLE_EASTERN", | ||
| "DECLINED", | ||
| "ISLAM", | ||
| "WITH_DISABILITY", | ||
| "WITH_DISORDER", | ||
| "LOWER_MIDDLE", | ||
| "DECLINED", | ||
| "DECLINED" | ||
| ) | ||
| personal_background.others = others | ||
| personal_background.is_public = self.response_personal_background["is_public"] | ||
|
|
||
| personal_background.save_to_db() | ||
|
|
||
| with self.client: | ||
| response = self.client.get( | ||
| "/user/personal_background", | ||
| headers={"Authorization": AUTH_COOKIE["Authorization"].value}, | ||
| follow_redirects=True, | ||
| content_type="application/json", | ||
| ) | ||
|
|
||
| test_user_personal_background_data = PersonalBackgroundModel.query.filter_by(user_id=self.test_user_data.id).first() | ||
| self.assertEqual(test_user_personal_background_data.user_id, response.json["user_id"]) | ||
| self.assertEqual(test_user_personal_background_data.gender.value, response.json["gender"]) | ||
| self.assertEqual(test_user_personal_background_data.age.value, response.json["age"]) | ||
| self.assertEqual(test_user_personal_background_data.ethnicity.value, response.json["ethnicity"]) | ||
| self.assertEqual(test_user_personal_background_data.sexual_orientation.value, response.json["sexual_orientation"]) | ||
| self.assertEqual(test_user_personal_background_data.religion.value, response.json["religion"]) | ||
| self.assertEqual(test_user_personal_background_data.physical_ability.value, response.json["physical_ability"]) | ||
| self.assertEqual(test_user_personal_background_data.mental_ability.value, response.json["mental_ability"]) | ||
| self.assertEqual(test_user_personal_background_data.socio_economic.value, response.json["socio_economic"]) | ||
| self.assertEqual(test_user_personal_background_data.highest_education.value, response.json["highest_education"]) | ||
| self.assertEqual(test_user_personal_background_data.years_of_experience.value, response.json["years_of_experience"]) | ||
| self.assertEqual(test_user_personal_background_data.others["gender_other"], response.json["gender_other"]) | ||
| self.assertEqual(test_user_personal_background_data.others["ethnicity_other"], response.json["ethnicity_other"]) | ||
| self.assertEqual(test_user_personal_background_data.others["sexual_orientation_other"], response.json["sexual_orientation_other"]) | ||
| self.assertEqual(test_user_personal_background_data.others["religion_other"], response.json["religion_other"]) | ||
| self.assertEqual(test_user_personal_background_data.others["physical_ability_other"], response.json["physical_ability_other"]) | ||
| self.assertEqual(test_user_personal_background_data.others["mental_ability_other"], response.json["mental_ability_other"]) | ||
| self.assertEqual(test_user_personal_background_data.others["socio_economic_other"], response.json["socio_economic_other"]) | ||
| self.assertEqual(test_user_personal_background_data.others["highest_education_other"], response.json["highest_education_other"]) | ||
| self.assertEqual(test_user_personal_background_data.is_public, response.json["is_public"]) | ||
| self.assertEqual(response.json, success_message) | ||
| self.assertEqual(response.status_code, success_code) | ||
|
|
||
|
|
||
|
|
||
|
|
||
| @patch("requests.get") | ||
| def test_api_dao_get_non_existence_additional_info(self, mock_get_personal_background): | ||
| error_message = messages.PERSONAL_BACKGROUND_DOES_NOT_EXIST | ||
| error_code = HTTPStatus.NOT_FOUND | ||
|
|
||
| mock_response = Mock() | ||
| mock_error = Mock() | ||
| http_error = requests.exceptions.HTTPError() | ||
| mock_response.raise_for_status.side_effect = http_error | ||
| mock_get_personal_background.return_value = mock_response | ||
| mock_error.json.return_value = error_message | ||
| mock_error.status_code = error_code | ||
| mock_get_personal_background.side_effect = requests.exceptions.HTTPError(response=mock_error) | ||
|
|
||
| with self.client: | ||
| response = self.client.get( | ||
| "/user/personal_background", | ||
| headers={"Authorization": AUTH_COOKIE["Authorization"].value}, | ||
| follow_redirects=True, | ||
| content_type="application/json", | ||
| ) | ||
|
|
||
| test_user_personal_background_data = PersonalBackgroundModel.query.filter_by(user_id=self.test_user_data.id).first() | ||
| self.assertEqual(test_user_personal_background_data, None) | ||
| self.assertEqual(response.json, error_message) | ||
| self.assertEqual(response.status_code, error_code) | ||
|
|
||
|
|
||
| if __name__ == "__main__": | ||
| unittest.main() | ||
|
|
||
|
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters