def test_set_members(self):
userfeatures = UserFeatures(self._agi, self._cursor, self._args)
userfeatures._set_feature_list = Mock()
userfeatures._set_caller = Mock()
userfeatures._set_lines = Mock()
userfeatures._set_user = Mock()
userfeatures._set_members()
objects.ExtenFeatures = Mock()
old_user, objects.User = objects.User, Mock()
self.assertEqual(userfeatures._userid, self._variables['XIVO_USERID'])
self.assertEqual(userfeatures._dstid, self._variables['XIVO_DSTID'])
self.assertEqual(userfeatures._lineid, self._variables['XIVO_LINEID'])
self.assertEqual(userfeatures._zone, self._variables['XIVO_CALLORIGIN'])
self.assertEqual(userfeatures._bypass_filter, self._variables['XIVO_CALLFILTER_BYPASS'])
self.assertEqual(userfeatures._srcnum, self._variables['XIVO_SRCNUM'])
self.assertEqual(userfeatures._dstnum, self._variables['XIVO_DSTNUM'])
self.assertTrue(userfeatures._set_feature_list.called)
self.assertTrue(userfeatures._set_caller.called)
self.assertTrue(userfeatures._set_lines.called)
self.assertTrue(userfeatures._set_user.called)
objects.User, old_user = old_user, None
def test_execute(self):
userfeatures = UserFeatures(self._agi, self._cursor, self._args)
userfeatures._set_members = Mock()
userfeatures._set_xivo_ifaces = Mock()
userfeatures._set_user_filter = Mock()
userfeatures._boss_secretary_filter = Mock()
userfeatures.execute()
self.assertEqual(userfeatures._set_members.call_count, 1)
self.assertEqual(userfeatures._set_xivo_ifaces.call_count, 1)
