From 47a1403591a5b42bf1234ea5500a97b82f9ff686 Mon Sep 17 00:00:00 2001 From: Ken Tsang Date: Thu, 10 Aug 2017 17:51:47 +0100 Subject: [PATCH] Refactor code to add updated_at --- app/dao/inbound_numbers_dao.py | 4 ++++ tests/app/dao/test_inbound_numbers_dao.py | 2 +- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/app/dao/inbound_numbers_dao.py b/app/dao/inbound_numbers_dao.py index 1c184b7f6..4a4143ffa 100644 --- a/app/dao/inbound_numbers_dao.py +++ b/app/dao/inbound_numbers_dao.py @@ -15,6 +15,10 @@ def dao_get_inbound_number_for_service(service_id): return InboundNumber.query.filter(InboundNumber.service_id == service_id).first() +def dao_get_inbound_number(inbound_number_id): + return InboundNumber.query.filter(InboundNumber.id == inbound_number_id).first() + + @transactional def dao_set_inbound_number_to_service(service_id, inbound_number): inbound_number.service_id = service_id diff --git a/tests/app/dao/test_inbound_numbers_dao.py b/tests/app/dao/test_inbound_numbers_dao.py index 9d59a8110..d38622d86 100644 --- a/tests/app/dao/test_inbound_numbers_dao.py +++ b/tests/app/dao/test_inbound_numbers_dao.py @@ -76,6 +76,6 @@ def test_set_inbound_number_active_flag(notify_db, notify_db_session, sample_ser dao_set_inbound_number_active_flag(inbound_number.id, active=active) - inbound_number = dao_get_inbound_number_for_service(sample_service.id) + inbound_number = dao_get_inbound_number(inbound_number.id) assert inbound_number.active is active