TelemetryInterceptorTest.java

package com.auth0.net;

import okhttp3.OkHttpClient;
import okhttp3.Request;
import okhttp3.mockwebserver.MockResponse;
import okhttp3.mockwebserver.MockWebServer;
import okhttp3.mockwebserver.RecordedRequest;
import org.junit.Before;
import org.junit.Test;

import static org.hamcrest.CoreMatchers.is;
import static org.hamcrest.CoreMatchers.nullValue;
import static org.junit.Assert.assertThat;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

public class TelemetryInterceptorTest {
    private Telemetry telemetry;

    @Before
    public void setUp() throws Exception {
        telemetry = mock(Telemetry.class);
        when(telemetry.getValue()).thenReturn("ClientInfo");
    }

    @Test
    public void shouldBeEnabledByDefault() throws Exception {
        TelemetryInterceptor interceptor = new TelemetryInterceptor(telemetry);
        assertThat(interceptor.isEnabled(), is(true));
    }

    @Test
    public void shouldDisable() throws Exception {
        TelemetryInterceptor interceptor = new TelemetryInterceptor(telemetry);
        interceptor.setEnabled(false);
        assertThat(interceptor.isEnabled(), is(false));
    }

    @Test
    public void shouldEnable() throws Exception {
        TelemetryInterceptor interceptor = new TelemetryInterceptor(telemetry);
        interceptor.setEnabled(true);
        assertThat(interceptor.isEnabled(), is(true));
    }

    @Test
    public void shouldAddTelemetryHeaderIfEnabled() throws Exception {
        TelemetryInterceptor interceptor = new TelemetryInterceptor(telemetry);
        interceptor.setEnabled(true);
        OkHttpClient client = new OkHttpClient.Builder()
                .addInterceptor(interceptor)
                .build();

        MockWebServer server = new MockWebServer();
        server.start();
        server.enqueue(new MockResponse());

        Request request = new Request.Builder()
                .get()
                .url(server.url("/"))
                .build();
        client.newCall(request).execute();

        RecordedRequest finalRequest = server.takeRequest();
        assertThat(finalRequest.getHeader("Auth0-Client"), is("ClientInfo"));

        server.shutdown();
    }

    @Test
    public void shouldNotAddTelemetryHeaderIfDisabled() throws Exception {
        TelemetryInterceptor interceptor = new TelemetryInterceptor(telemetry);
        interceptor.setEnabled(false);
        OkHttpClient client = new OkHttpClient.Builder()
                .addInterceptor(interceptor)
                .build();

        MockWebServer server = new MockWebServer();
        server.start();
        server.enqueue(new MockResponse());

        Request request = new Request.Builder()
                .get()
                .url(server.url("/"))
                .build();
        client.newCall(request).execute();

        RecordedRequest finalRequest = server.takeRequest();
        assertThat(finalRequest.getHeader("Auth0-Client"), is(nullValue()));

        server.shutdown();
    }


}