Skip to content

Commit

Permalink
fix lint and unit test
Browse files Browse the repository at this point in the history
  • Loading branch information
jadmsaadaot committed Jul 7, 2023
1 parent ebd2bbf commit b61dc77
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 26 deletions.
2 changes: 1 addition & 1 deletion met-api/src/met_api/services/report_setting_service.py
Original file line number Diff line number Diff line change
Expand Up @@ -141,7 +141,7 @@ def update_report_setting(cls, survey_id, new_report_settings):
"""Update report setting."""
survey = SurveyModel.find_by_id(survey_id)
if not survey:
raise KeyError(f'No survey found for {survey_id}')
raise KeyError(f'No survey found for {survey_id}')
for setting in new_report_settings:
report_setting_id = setting.get('id', None)
report_setting = ReportSettingModel.find_by_id(report_setting_id)
Expand Down
34 changes: 9 additions & 25 deletions met-api/tests/unit/api/test_report_setting.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,41 +16,25 @@
Test-Suite to ensure that the Report setting endpoint is working as expected.
"""
import json

from met_api.utils.enums import ContentType
from tests.utilities.factory_scenarios import TestJwtClaims, TestSurveyInfo
from tests.utilities.factory_utils import factory_auth_header, factory_survey_and_eng_model


def test_create_report_setting(client, jwt, session): # pylint:disable=unused-argument
"""Assert that an report setting can be POSTed."""
headers = factory_auth_header(jwt=jwt, claims=TestJwtClaims.staff_admin_role)
survey, eng = factory_survey_and_eng_model(TestSurveyInfo.survey3)
data = {
'id': survey.id,
'form_json': survey.form_json,
}
rv = client.post('/api/reportsetting/', data=json.dumps(data),
headers=headers, content_type=ContentType.JSON.value)
assert rv.status_code == 200
from tests.utilities.factory_scenarios import TestJwtClaims, TestReportSettingInfo, TestSurveyInfo
from tests.utilities.factory_utils import (
factory_auth_header, factory_survey_and_eng_model, factory_survey_report_setting_model)


def test_get_report_setting(client, jwt, session): # pylint:disable=unused-argument
"""Assert that report setting can be fetched."""
headers = factory_auth_header(jwt=jwt, claims=TestJwtClaims.staff_admin_role)
survey, eng = factory_survey_and_eng_model(TestSurveyInfo.survey3)
survey, _ = factory_survey_and_eng_model(TestSurveyInfo.survey3)

data = {
'id': survey.id,
'form_json': survey.form_json,
report_setting_data = {
**TestReportSettingInfo.report_setting_1,
'survey_id': survey.id,
}
rv = client.post('/api/reportsetting/', data=json.dumps(data),
headers=headers, content_type=ContentType.JSON.value)
assert rv.status_code == 200
factory_survey_report_setting_model(report_setting_data)

rv = client.get(
f'/api/reportsetting/{survey.id}',
f'/api/surveys/{survey.id}/reportsettings',
headers=headers,
content_type=ContentType.JSON.value
)
Expand Down

0 comments on commit b61dc77

Please sign in to comment.