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)