def test_is_main_line(self):
userfeatures = UserFeatures(self._agi, self._cursor, self._args)
self.assertFalse(userfeatures._is_main_line())
lineid = '97'
userfeatures._lineid = lineid
userfeatures._master_line = {'id': lineid}
self.assertTrue(userfeatures._is_main_line())
def test_set_pickup_info(self):
context = 'foo'
number = '101'
userfeatures = UserFeatures(self._agi, self._cursor, self._args)
self.assertEqual(None, userfeatures._pickup_context)
userfeatures._master_line = {'id': '2',
'number': number,
'context': context}
userfeatures._dstnum = number
userfeatures._set_pickup_info()
self.assertEqual(context, userfeatures._pickup_context)
self.assertEqual(number, userfeatures._pickup_exten)
