from Testing.makerequest import makerequest
from AccessControl.SecurityManagement import newSecurityManager, noSecurityManager
from AccessControl.User import User, UnrestrictedUser, SpecialUser
from AccessControl.unauthorized import Unauthorized

def loginAnonymous():
    """ """
    noSecurityManager()
    anonymous_user = SpecialUser('Anonymous User','',('Anonymous',), [])
    newSecurityManager(None, anonymous_user)

def loginUnrestricted():
    """ """
    noSecurityManager()
    god = UnrestrictedUser('god', 'god', [], '')
    newSecurityManager(None, god)
    return god

def loginAs(username, roles):
    """ """
    noSecurityManager()
    user = User(username, username, roles, '')
    newSecurityManager(None, user)
    return user

def loginUser(user, zope_app=None):
    """ """
    newSecurityManager(None, user)
    if zope_app:
        zope_app.REQUEST.set('AUTHENTICATED_USER', user)

def logout():
    ''' simulates logout '''
    noSecurityManager()

def setupUser(folder, user_name, roles):
    '''Creates the user in the given folder (folder) acl_users.'''
    uf = getattr(folder, 'acl_users')
    uf._addUser(user_name, 'secret', 'secret', roles, ())
    user = uf.getUserById(user_name).__of__(uf)
    return user

#--------------------

def addCMFsite(zope):
    ''' creates a CMF site in the given zope (folder) '''
    zope.manage_addProduct['CMFDefault'].manage_addCMFSite('RegistrationTest')
    site = getattr(zope, 'RegistrationTest')
    return site

def registerCMFuser(site, user_name):
    ''' creates a CMF portal member in the given CMF site'''
    site.portal_registration.addMember(user_name, 'secret')