001    /**
002     * Copyright (c) 2000-present Liferay, Inc. All rights reserved.
003     *
004     * This library is free software; you can redistribute it and/or modify it under
005     * the terms of the GNU Lesser General Public License as published by the Free
006     * Software Foundation; either version 2.1 of the License, or (at your option)
007     * any later version.
008     *
009     * This library is distributed in the hope that it will be useful, but WITHOUT
010     * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
011     * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
012     * details.
013     */
014    
015    package com.liferay.portal.kernel.test.context;
016    
017    import com.liferay.portal.model.User;
018    import com.liferay.portal.security.auth.PrincipalThreadLocal;
019    import com.liferay.portal.security.permission.PermissionChecker;
020    import com.liferay.portal.security.permission.PermissionCheckerFactoryUtil;
021    import com.liferay.portal.security.permission.PermissionThreadLocal;
022    
023    /**
024     * @author Adolfo P??rez
025     */
026    public class ContextUserReplace implements AutoCloseable {
027    
028            public ContextUserReplace(User user) throws Exception {
029                    this(user, PermissionCheckerFactoryUtil.create(user));
030            }
031    
032            public ContextUserReplace(User user, PermissionChecker permissionChecker) {
033                    _name = PrincipalThreadLocal.getName();
034                    _permissionCheker = PermissionThreadLocal.getPermissionChecker();
035    
036                    PrincipalThreadLocal.setName(user.getUserId());
037                    PermissionThreadLocal.setPermissionChecker(permissionChecker);
038            }
039    
040            @Override
041            public void close() {
042                    PrincipalThreadLocal.setName(_name);
043                    PermissionThreadLocal.setPermissionChecker(_permissionCheker);
044            }
045    
046            private final String _name;
047            private final PermissionChecker _permissionCheker;
048    
049    }