org.piwik.java.tracking.PiwikTrackerTest.java Source code

Introduction

Here is the source code for org.piwik.java.tracking.PiwikTrackerTest.java

Source

/* 
 * Piwik Java Tracker
 * 
 * @link https://github.com/piwik/piwik-java-tracker
 * @license https://github.com/piwik/piwik-java-tracker/blob/master/LICENSE BSD-3 Clause
 */
package org.piwik.java.tracking;

import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;
import org.apache.http.HttpResponse;
import org.apache.http.client.HttpClient;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.methods.HttpPost;
import org.junit.After;
import org.junit.AfterClass;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.fail;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Test;
import org.mockito.ArgumentMatcher;
import static org.mockito.Matchers.argThat;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.spy;

/**
 *
 * @author brettcsorba
 */
public class PiwikTrackerTest {
    PiwikTracker piwikTracker;

    public PiwikTrackerTest() {
    }

    @BeforeClass
    public static void setUpClass() {
    }

    @AfterClass
    public static void tearDownClass() {
    }

    @Before
    public void setUp() {
        piwikTracker = spy(new PiwikTracker("http://test.com"));
    }

    @After
    public void tearDown() {
    }

    /**
     * Test of addParameter method, of class PiwikTracker.
     */
    @Test
    public void testAddParameter() {
    }

    /**
     * Test of sendRequest method, of class PiwikTracker.
     */
    @Test
    public void testSendRequest() throws Exception {
        PiwikRequest request = mock(PiwikRequest.class);
        HttpClient client = mock(HttpClient.class);
        HttpResponse response = mock(HttpResponse.class);

        doReturn(client).when(piwikTracker).getHttpClient();
        doReturn("query").when(request).getUrlEncodedQueryString();
        doReturn(response).when(client).execute(argThat(new CorrectGetRequest("http://test.com?query")));

        assertEquals(response, piwikTracker.sendRequest(request));
    }

    class CorrectGetRequest extends ArgumentMatcher<HttpGet> {
        String url;

        public CorrectGetRequest(String url) {
            this.url = url;
        }

        @Override
        public boolean matches(Object argument) {
            HttpGet get = (HttpGet) argument;
            return url.equals(get.getURI().toString());
        }
    }

    /**
     * Test of sendBulkRequest method, of class PiwikTracker.
     */
    @Test
    public void testSendBulkRequest_Iterable() throws Exception {
        List<PiwikRequest> requests = new ArrayList<>();
        HttpResponse response = mock(HttpResponse.class);

        doReturn(response).when(piwikTracker).sendBulkRequest(requests, null);

        assertEquals(response, piwikTracker.sendBulkRequest(requests));
    }

    /**
     * Test of sendBulkRequest method, of class PiwikTracker.
     */
    @Test
    public void testSendBulkRequest_Iterable_StringTT() throws Exception {
        try {
            List<PiwikRequest> requests = new ArrayList<>();
            HttpClient client = mock(HttpClient.class);
            PiwikRequest request = mock(PiwikRequest.class);

            doReturn("query").when(request).getQueryString();
            requests.add(request);
            doReturn(client).when(piwikTracker).getHttpClient();

            piwikTracker.sendBulkRequest(requests, "1");
            fail("Exception should have been thrown.");
        } catch (IllegalArgumentException e) {
            assertEquals("1 is not 32 characters long.", e.getLocalizedMessage());
        }
    }

    @Test
    public void testSendBulkRequest_Iterable_StringFF() throws Exception {
        List<PiwikRequest> requests = new ArrayList<>();
        HttpClient client = mock(HttpClient.class);
        PiwikRequest request = mock(PiwikRequest.class);
        HttpResponse response = mock(HttpResponse.class);

        doReturn("query").when(request).getQueryString();
        requests.add(request);
        doReturn(client).when(piwikTracker).getHttpClient();
        doReturn(response).when(client).execute(argThat(new CorrectPostRequest("{\"requests\":[\"?query\"]}")));

        assertEquals(response, piwikTracker.sendBulkRequest(requests, null));
    }

    @Test
    public void testSendBulkRequest_Iterable_StringFT() throws Exception {
        List<PiwikRequest> requests = new ArrayList<>();
        HttpClient client = mock(HttpClient.class);
        PiwikRequest request = mock(PiwikRequest.class);
        HttpResponse response = mock(HttpResponse.class);

        doReturn("query").when(request).getQueryString();
        requests.add(request);
        doReturn(client).when(piwikTracker).getHttpClient();
        doReturn(response).when(client).execute(argThat(new CorrectPostRequest(
                "{\"requests\":[\"?query\"],\"token_auth\":\"12345678901234567890123456789012\"}")));

        assertEquals(response, piwikTracker.sendBulkRequest(requests, "12345678901234567890123456789012"));
    }

    class CorrectPostRequest extends ArgumentMatcher<HttpPost> {
        String body;

        public CorrectPostRequest(String body) {
            this.body = body;
        }

        @Override
        public boolean matches(Object argument) {
            try {
                HttpPost post = (HttpPost) argument;
                InputStream bais = post.getEntity().getContent();
                byte[] bytes = new byte[bais.available()];
                bais.read(bytes);
                String str = new String(bytes);
                return body.equals(str);
            } catch (IOException e) {
                fail("Exception should not have been throw.");
            }
            return false;
        }
    }

    /**
     * Test of getHttpClient method, of class PiwikTracker.
     */
    @Test
    public void testGetHttpClient() {
        assertNotNull(piwikTracker.getHttpClient());
    }

}