test.erp5.ShaSecurityMixin.py 2.83 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42
# -*- coding: utf-8 -*-
##############################################################################
#
# Copyright (c) 2011 Nexedi SA and Contributors. All Rights Reserved.
#                    Lucas Carvalho <lucas@nexedi.com>
#
# WARNING: This program as such is intended to be used by professional
# programmers who take the whole responsability of assessing all potential
# consequences resulting from its eventual inadequacies and bugs
# End users who are looking for a ready-to-use solution with commercial
# garantees and support are strongly adviced to contract a Free Software
# Service Company
#
# This program is Free Software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
#
##############################################################################


from AccessControl.SecurityManagement import newSecurityManager


class ShaSecurityMixin(object):
  """
    ShaSecurity - Mixin Class
  """

  def afterSetUp(self):
    """
      Initialize the ERP5 site.
    """
43
    self.lucas_user = self.createUser('lucas', 'lucas').Person_getUserId()
44

45
    self.toto_user = self.createUser('toto', 'toto').Person_getUserId()
46 47 48 49 50

  def createUser(self, reference, password):
    """
      Create a user with basic information
    """
51
    person = self.portal.portal_catalog.getResultValue(portal_type='Person', reference=reference)
52 53 54
    if person is None:
      person = self.portal.person_module.newContent(portal_type='Person')
      person.edit(first_name=reference,
55 56 57 58 59 60 61
                  reference=reference)
      login = person.newContent(
        portal_type='ERP5 Login',
        reference=reference,
        password=password,
      )
      login.validate()
62
      self.tic()
63 64 65 66

    create = True
    group=self.group
    for assignment in person.contentValues(portal_type="Assignment"):
67
      if assignment.getGroup() == self.group:
68 69 70
        create = False

    if create:
71
      assignment = person.newContent(portal_type='Assignment')
72
      assignment.edit(group=self.group)
73
      assignment.open()
74
      self.tic()
75

76 77
    return person

78 79 80 81 82 83 84 85 86
  def changeUser(self, user_id):
    """
      Change the current user to user_id
    """
    user_folder = self.getPortal().acl_users
    user = user_folder.getUserById(user_id).__of__(user_folder)
    newSecurityManager(None, user)