def test__format_storage(self):
# (npa, nxx, xxxx, ext, expect, label)
tests = [
('519', '291', '5554', '123', '5192915554123', 'all parts'),
('519', '291', '5554', '', '5192915554', 'no ext'),
('', '291', '5554', '123', '', 'no npa'),
('519', '', '5554', '123', '', 'no nxx'),
('519', '291', '', '123', '', 'no xxxx'),
('', '', '', '', '', 'no parts'),
]
for t in tests:
phone = PhoneNumber('')
phone.npa = t[0]
phone.nxx = t[1]
phone.xxxx = t[2]
phone.ext = t[3]
self.assertEqual(phone.format_storage(), t[4])
# Handle non-typical phone number
value = 'Not Available'
phone = PhoneNumber(value)
self.assertEqual(phone.format_storage(), value)
phone = PhoneNumber(' ' + value + ' ')
self.assertEqual(phone.format_storage(), value)
def test__station(self):
phone = PhoneNumber('')
self.assertEqual(phone.station(), '')
phone.xxxx = '5554'
self.assertEqual(phone.station(), '5554')
