001
014
015 package com.liferay.portal.kernel.test.rule.callback;
016
017 import com.liferay.portal.kernel.exception.PortalException;
018 import com.liferay.portal.kernel.test.util.TestPropsValues;
019 import com.liferay.portal.security.auth.CompanyThreadLocal;
020
021 import org.junit.runner.Description;
022
023
026 public class CompanyProviderTestCallback extends BaseTestCallback<Long, Void> {
027
028 public static final CompanyProviderTestCallback INSTANCE =
029 new CompanyProviderTestCallback();
030
031 @Override
032 public void afterClass(Description description, Long previousCompanyId) {
033 CompanyThreadLocal.setCompanyId(previousCompanyId);
034 }
035
036 @Override
037 public Long beforeClass(Description description) throws PortalException {
038 Long companyId = CompanyThreadLocal.getCompanyId();
039
040 CompanyThreadLocal.setCompanyId(TestPropsValues.getCompanyId());
041
042 return companyId;
043 }
044
045 private CompanyProviderTestCallback() {
046 }
047
048 }