alfresco-community-repo/source/test-java/org/alfresco/repo/virtual/bundle/VirtualRatingServiceExtensionTest.java

/*
 * #%L
 * Alfresco Repository
 * %%
 * Copyright (C) 2005 - 2016 Alfresco Software Limited
 * %%
 * This file is part of the Alfresco software. 
 * If the software was purchased under a paid Alfresco license, the terms of 
 * the paid license agreement will prevail.  Otherwise, the software is 
 * provided under the following open source license terms:
 * 
 * Alfresco is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 * 
 * Alfresco 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 Lesser General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public License
 * along with Alfresco. If not, see <http://www.gnu.org/licenses/>.
 * #L%
 */

package org.alfresco.repo.virtual.bundle;

import org.alfresco.model.ContentModel;
import org.alfresco.repo.security.authentication.AuthenticationUtil;
import org.alfresco.repo.security.authentication.AuthenticationUtil.RunAsWork;
import org.alfresco.repo.virtual.VirtualizationIntegrationTest;
import org.alfresco.repo.virtual.ref.GetActualNodeRefMethod;
import org.alfresco.repo.virtual.ref.Reference;
import org.alfresco.service.cmr.rating.RatingService;
import org.alfresco.service.cmr.rating.RatingServiceException;
import org.alfresco.service.cmr.repository.NodeRef;
import org.alfresco.service.cmr.security.AccessStatus;
import org.junit.Test;

public class VirtualRatingServiceExtensionTest extends VirtualizationIntegrationTest
{
    private final static String LIKES_RATING_SCHEME = "likesRatingScheme";

    private final static String FIVE_STAR_RATING_SCHEME = "fiveStarRatingScheme";

    private RatingService ratingService;

    private NodeRef vf1Node2;

    private NodeRef virtualContent;

    private String user1;

    private String user2;

    @Override
    protected void setUp() throws Exception
    {
        super.setUp();

        ratingService = VirtualPermissionServiceExtensionTest.ctx.getBean("ratingService",
                                                                          RatingService.class);

        user1 = "user1";

        user2 = "user2";

        vf1Node2 = nodeService.getChildByName(this.virtualFolder1NodeRef,
                                              ContentModel.ASSOC_CONTAINS,
                                              "Node2");

        virtualContent = createContent(vf1Node2,
                                       "virtualContent").getChildRef();
    }

    private void applyRatingAs(final NodeRef targetNode, final float rating, final String ratingSchemeName,
                String asUser) throws RatingServiceException
    {

        String fau = AuthenticationUtil.getFullyAuthenticatedUser();
        try
        {
            AuthenticationUtil.setFullyAuthenticatedUser(asUser);
            RunAsWork<Void> applyRatingsAsWork = new RunAsWork<Void>()
            {

                @Override
                public Void doWork() throws Exception
                {
                    ratingService.applyRating(targetNode,
                                              rating,
                                              ratingSchemeName);
                    return null;
                }

            };
            AuthenticationUtil.runAs(applyRatingsAsWork,
                                     asUser);
        }
        finally
        {
            AuthenticationUtil.setFullyAuthenticatedUser(fau);
        }
    }

    @Test
    public void testApplyRatings() throws Exception
    {
        assertTrue(Reference.isReference(virtualContent));
        NodeRef actualNodeRef = Reference.fromNodeRef(virtualContent).execute(new GetActualNodeRefMethod(environment));

        applyRatingAs(virtualContent,
                      1f,
                      LIKES_RATING_SCHEME,
                      user1);

        applyRatingAs(virtualContent,
                      1f,
                      LIKES_RATING_SCHEME,
                      user2);

        assertEquals(1f,
                     ratingService.getAverageRating(virtualContent,
                                                    LIKES_RATING_SCHEME));
        assertEquals(1f,
                     ratingService.getAverageRating(actualNodeRef,
                                                    LIKES_RATING_SCHEME));

        applyRatingAs(virtualContent,
                      1f,
                      FIVE_STAR_RATING_SCHEME,
                      user1);
        applyRatingAs(virtualContent,
                      3f,
                      FIVE_STAR_RATING_SCHEME,
                      user2);
        assertEquals(2f,
                     ratingService.getAverageRating(virtualContent,
                                                    FIVE_STAR_RATING_SCHEME));
        assertEquals(2f,
                     ratingService.getAverageRating(actualNodeRef,
                                                    FIVE_STAR_RATING_SCHEME));
    }
}