de.hybris.platform.masterserver.impl.DefaultStatisticsGatewayTest.java Source code

Introduction

Here is the source code for de.hybris.platform.masterserver.impl.DefaultStatisticsGatewayTest.java

Source

/*
 * [y] hybris Platform
 *
 * Copyright (c) 2000-2013 hybris AG
 * All rights reserved.
 *
 * This software is the confidential and proprietary information of hybris
 * ("Confidential Information"). You shall not disclose such Confidential
 * Information and shall use it only in accordance with the terms of the
 * license agreement you entered into with hybris.
 * 
 *  
 */
package de.hybris.platform.masterserver.impl;

import static org.fest.assertions.Assertions.assertThat;
import static org.mockito.BDDMockito.given;
import static org.mockito.Matchers.anyString;
import static org.mockito.Matchers.eq;
import de.hybris.bootstrap.annotations.UnitTest;
import de.hybris.platform.core.PK;
import de.hybris.platform.masterserver.StatisticsPayloadEncryptor;

import com.hybris.statistics.collector.BusinessStatisticsCollector;
import com.hybris.statistics.collector.SystemStatisticsCollector;

import java.util.HashMap;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.CountDownLatch;

import org.apache.commons.lang.math.RandomUtils;
import org.junit.Before;
import org.junit.Test;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;

import com.google.common.collect.Sets;

@UnitTest
public class DefaultStatisticsGatewayTest {
    private DefaultStatisticsGateway statGateway, statGateway2;
    @Mock
    private BusinessStatisticsCollector businessCollector;
    @Mock
    private SystemStatisticsCollector systemCollector;
    @Mock
    private StatisticsPayloadEncryptor encryptor;
    @Mock
    private StatisticsPayload statisticsPayload;

    private boolean generateStats = false;

    @Before
    public void setUp() throws Exception {
        MockitoAnnotations.initMocks(this);
        final Set<BusinessStatisticsCollector> businessCollectors = Sets.newHashSet(businessCollector);
        final Set<SystemStatisticsCollector> systemCollectors = Sets.newHashSet(systemCollector);
        final Map<String, String> existingWebModules = new HashMap<>();
        existingWebModules.put("/hac", "hac");
        existingWebModules.put("/productcockpit", "productcockpit");

        statGateway = new DefaultStatisticsGateway(businessCollectors, systemCollectors, existingWebModules,
                encryptor) {
            @Override
            protected boolean isSendingStatsAgreedInLicense() {
                return true;
            }

            @Override
            protected boolean isGenerateStatsRequired() {
                return generateStats;
            }

            @Override
            protected PK getCurrentUserPk() {
                return PK.createFixedUUIDPK(0, RandomUtils.nextLong());
            }
        };

        statGateway2 = new DefaultStatisticsGateway(businessCollectors, systemCollectors, existingWebModules,
                encryptor) {
            @Override
            protected boolean isSendingStatsAgreedInLicense() {
                return true;
            }

            @Override
            protected boolean isGenerateStatsRequired() {
                return generateStats;
            }

            @Override
            protected PK getCurrentUserPk() {
                return null;
            }
        };

    }

    @Test
    public void shouldDontReturnPayloadIfItIsNotATime() {
        // given
        generateStats = false;

        // when
        final StatisticsPayload encodedStatistics = statGateway.getStatisticsPayload();

        // then
        assertThat(encodedStatistics).isNull();
    }

    @Test
    public void shouldDontReturnPayloadIfNonOfStatisticCollectorsReturnsDataAndItIsTimeToGenerateData() {
        // given
        generateStats = true;
        given(businessCollector.collectStatistics()).willReturn(null);
        given(systemCollector.collectStatistics()).willReturn(null);

        // when
        final StatisticsPayload encodedStatistics1 = statGateway2.getStatisticsPayload();

        // then
        assertThat(encodedStatistics1).isNull();
    }

    @Test
    public void shouldReturnPayloadWhenAtLeastOneOfStatisticCollectorsReturnDataAndItIsTimeToGenerateData() {
        // given
        final Map<String, Map<String, Object>> businessStats = new HashMap<>();
        businessStats.put("backOfficeUsers", new HashMap<String, Object>());
        businessStats.get("backOfficeUsers").put("employees", Integer.valueOf(5));
        businessStats.get("backOfficeUsers").put("customers", Integer.valueOf(10));

        final Map<String, Map<String, Object>> systemStats = new HashMap<>();
        systemStats.put("cpu", new HashMap<String, Object>());
        systemStats.get("cpu").put("numCPU", Integer.valueOf(8));

        generateStats = true;
        given(businessCollector.collectStatistics()).willReturn(businessStats);
        given(systemCollector.collectStatistics()).willReturn(systemStats);
        given(encryptor.encrypt(
                eq("{\"system\":{\"cpu\":{\"numCPU\":8}},\"business\":{\"backOfficeUsers\":{\"customers\":10,"
                        + "\"employees\":5}}}"),
                anyString())).willReturn(statisticsPayload);

        // when
        final StatisticsPayload encodedStatistics1 = statGateway2.getStatisticsPayload();

        // then
        assertThat(encodedStatistics1).isNotNull();
    }

    @Test
    public void shouldUpdateLoggedInBackOfficeUsersConcurrently() throws InterruptedException {
        // given
        generateStats = true;
        given(businessCollector.collectStatistics()).willReturn(null);
        given(systemCollector.collectStatistics()).willReturn(null);
        given(encryptor.encrypt(eq("{\"session\":{\"backOfficeOverallUsers\":{\"hac\":500}}}"), anyString()))
                .willReturn(statisticsPayload);
        final CountDownLatch latch = new CountDownLatch(500);

        // when
        for (int i = 0; i < 500; i++) {
            new Thread(new Runnable() {

                @Override
                public void run() {
                    try {
                        statGateway.updateLoggedInUsersStats("/hac");
                    } finally {
                        latch.countDown();
                    }
                }
            }).start();
        }
        latch.await();
        final StatisticsPayload encodedStatistics = statGateway.getStatisticsPayload();

        // then
        assertThat(encodedStatistics).isNotNull();
    }
}