/*
 * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
 * or more contributor license agreements. Licensed under the Elastic License;
 * you may not use this file except in compliance with the Elastic License.
 */
package org.elasticsearch.xpack.monitoring.exporter.http;

import org.apache.http.entity.ContentType;
import org.apache.http.entity.StringEntity;
import org.elasticsearch.Version;
import org.elasticsearch.client.Response;
import org.elasticsearch.client.RestClient;
import org.elasticsearch.test.ESTestCase;

import java.io.IOException;

import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

/**
 * Tests {@link VersionHttpResource}.
 */
public class VersionHttpResourceTests extends ESTestCase {

    private final String owner = getClass().getSimpleName();
    private final RestClient client = mock(RestClient.class);

    public void testDoCheckAndPublishSuccess() throws IOException {
        final Version minimumVersion =
                randomFrom(Version.V_2_0_0, Version.V_2_0_0_beta1, Version.V_2_0_0_rc1, Version.V_2_3_3, Version.CURRENT);
        final Version version = randomFrom(minimumVersion, Version.CURRENT);
        final Response response = responseForVersion(version);

        final VersionHttpResource resource = new VersionHttpResource(owner, minimumVersion);

        assertTrue(resource.doCheckAndPublish(client));

        verify(response).getEntity();
    }

    public void testDoCheckAndPublishFailedParsing() throws IOException {
        // malformed JSON
        final Response response = responseForJSON("{");

        final VersionHttpResource resource = new VersionHttpResource(owner, Version.CURRENT);

        assertFalse(resource.doCheckAndPublish(client));

        verify(response).getEntity();
    }

    public void testDoCheckAndPublishFailedFieldMissing() throws IOException {
        // malformed response; imagining that we may change it in the future or someone breaks filter_path
        final Response response = responseForJSON("{\"version.number\":\"" + Version.CURRENT + "\"}");

        final VersionHttpResource resource = new VersionHttpResource(owner, Version.CURRENT);

        assertFalse(resource.doCheckAndPublish(client));

        verify(response).getEntity();
    }

    public void testDoCheckAndPublishFailedFieldWrongType() throws IOException {
        // malformed response (should be {version: { number : ... }})
        final Response response = responseForJSON("{\"version\":\"" + Version.CURRENT + "\"}");

        final VersionHttpResource resource = new VersionHttpResource(owner, Version.CURRENT);

        assertFalse(resource.doCheckAndPublish(client));

        verify(response).getEntity();
    }

    public void testDoCheckAndPublishFailedWithIOException() throws IOException {
        // request fails for some reason
        when(client.performRequest("GET", "/", VersionHttpResource.PARAMETERS)).thenThrow(new IOException("expected"));

        final VersionHttpResource resource = new VersionHttpResource(owner, Version.CURRENT);

        assertFalse(resource.doCheckAndPublish(client));
    }

    private Response responseForJSON(final String json) throws IOException {
        final StringEntity entity = new StringEntity(json, ContentType.APPLICATION_JSON);

        final Response response = mock(Response.class);
        when(response.getEntity()).thenReturn(entity);

        when(client.performRequest("GET", "/", VersionHttpResource.PARAMETERS)).thenReturn(response);

        return response;
    }

    private Response responseForVersion(final Version version) throws IOException {
        return responseForJSON("{\"version\":{\"number\":\"" + version + "\"}}");
    }

}

@@ -542,16 +542,60 @@
 estCase;
+%0Aimport org.elasticsearch.test.VersionUtils;
 %0A%0Aimport
@@ -1080,129 +1080,44 @@
 on =
-%0A                randomFrom(Version.V_2_0_0, Version.V_2_0_0_beta1, Version.V_2_0_0_rc1, Version.V_2_3_3, Version.CURRENT
+ VersionUtils.randomVersion(random()
 );%0A 

Remove uses of 2.0 prerelease constants