Fix usPhone with empty phone number

This commit is contained in:
Patrick Smith 2019-02-13 11:36:17 -05:00
parent a8e9430196
commit 3e50ce3d48
2 changed files with 9 additions and 1 deletions

View File

@ -30,6 +30,8 @@ def justCents(value):
def usPhone(number):
if not number:
return ""
phone = re.sub(r"\D", "", number)
return "+1 ({}) {} - {}".format(phone[0:3], phone[3:6], phone[6:])

View File

@ -1,6 +1,6 @@
import pytest
from atst.filters import dollars, renderAuditEvent
from atst.filters import dollars, renderAuditEvent, usPhone
from atst.models import AuditEvent
@ -28,3 +28,9 @@ def test_render_audit_event_with_unknown_resource_type():
event = AuditEvent(resource_type="boat")
result = renderAuditEvent(event)
assert "<article" in result
def test_usPhone():
assert usPhone("1234567890") == "+1 (123) 456 - 7890"
assert usPhone(number=None) == ""
assert usPhone(number="") == ""