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)