package com.perimeterx.http;

import com.perimeterx.api.activities.BufferedActivityHandler;
import com.perimeterx.api.providers.DefaultHostnameProvider;
import com.perimeterx.api.providers.HostnameProvider;
import com.perimeterx.api.providers.IPProvider;
import com.perimeterx.api.providers.RemoteAddressIPProvider;
import com.perimeterx.models.PXContext;
import com.perimeterx.models.configuration.PXConfiguration;
import com.perimeterx.models.exceptions.PXException;
import org.junit.Test;
import org.springframework.mock.web.MockHttpServletRequest;
import org.testng.annotations.BeforeMethod;
import testutils.TestObjectUtils;

import javax.servlet.http.HttpServletRequest;
import java.io.IOException;
import java.util.List;

import static org.mockito.Mockito.*;

/**
 * Created by nitzangoldfeder on 23/02/2017.
 */
public class BufferedActivitiesTest {

    private BufferedActivityHandler bufferedActivityHandler;
    private PXHttpClient pxClient;
    private PXConfiguration configuration;
    private HttpServletRequest request;
    private IPProvider ipProvider;
    private HostnameProvider hostnameProvider;
    private PXContext context;
    private PXConfiguration config;

    @BeforeMethod
    public void setUp() {
        config = PXConfiguration.builder()
                .appId("APP_ID")
                .authToken("AUTH_123")
                .cookieKey("COOKIE_123")
                .build();
        this.pxClient = mock(PXHttpClient.class);
        this.configuration = TestObjectUtils.generateConfiguration();
        this.bufferedActivityHandler = new BufferedActivityHandler(pxClient, configuration);

        this.request = new MockHttpServletRequest();
        this.ipProvider = new RemoteAddressIPProvider();
        this.hostnameProvider = new DefaultHostnameProvider();
        this.context = new PXContext(request, this.ipProvider, this.hostnameProvider, config);

    }

    @Test
    public void testSendActivityOnMaxBuffer() throws IOException, PXException {
        for (int i = 0; i <= this.configuration.getMaxBufferLen(); i++) {
            bufferedActivityHandler.handleBlockActivity(context);
        }

        verify(pxClient, atLeastOnce()).sendBatchActivities(any(List.class));
    }

    @Test
    public void testDontSendActivityBelowMaxBuffer() throws IOException, PXException {
        for (int i = 0; i < this.configuration.getMaxBufferLen() - 1; i++) {
            bufferedActivityHandler.handleBlockActivity(context);
        }
        verify(pxClient, never()).sendBatchActivities(any(List.class));
    }
}

@@ -477,13 +477,26 @@
 org.
-junit
+testng.annotations
 .Tes

Update BufferedActivitiesTest.java