From c7f11133d3933905700737e17314f462973a456a Mon Sep 17 00:00:00 2001 From: Andrew Shumway Date: Mon, 16 Oct 2023 15:02:52 -0600 Subject: [PATCH] Added another flag in appropriate test --- app/service/rest.py | 9 ++++++++- tests/app/service/test_rest.py | 9 +++++++-- 2 files changed, 15 insertions(+), 3 deletions(-) diff --git a/app/service/rest.py b/app/service/rest.py index 0426bb086..e06ebd8c0 100644 --- a/app/service/rest.py +++ b/app/service/rest.py @@ -226,8 +226,15 @@ def create_service(): # unpack valid json into service object valid_service = Service.from_json(data) + # Grabbing flag from request object for default sms sender + create_default_sms_sender = bool( + request.args.get("create_default_sms_sender", False) + ) + with transaction(): - dao_create_service(valid_service, user, create_default_sms_sender=True) + dao_create_service( + valid_service, user, create_default_sms_sender=create_default_sms_sender + ) set_default_free_allowance_for_service(service=valid_service, year_start=None) return jsonify(data=service_schema.dump(valid_service)), 201 diff --git a/tests/app/service/test_rest.py b/tests/app/service/test_rest.py index 9cf0829ea..c1cd6b741 100644 --- a/tests/app/service/test_rest.py +++ b/tests/app/service/test_rest.py @@ -380,8 +380,13 @@ def test_create_service( "created_by": str(sample_user.id), } + create_default_sms_sender = True + json_resp = admin_request.post( - "service.create_service", _data=data, _expected_status=201 + "service.create_service", + _data=data, + _expected_status=201, + create_default_sms_sender=create_default_sms_sender, ) assert json_resp["data"]["id"] @@ -399,7 +404,7 @@ def test_create_service( ) assert json_resp["data"]["name"] == "created service" - + # insert_service_sms_sender(service, current_app.config["FROM_NUMBER"]) service_sms_senders = ServiceSmsSender.query.filter_by( service_id=service_db.id ).all()