package org.infinispan.rest.search;

import static org.infinispan.commons.api.CacheContainerAdmin.AdminFlag.VOLATILE;
import static org.infinispan.commons.dataconversion.MediaType.APPLICATION_PROTOSTREAM_TYPE;
import static org.infinispan.configuration.cache.IndexStorage.LOCAL_HEAP;
import static org.testng.AssertJUnit.assertFalse;

import java.util.concurrent.CompletionStage;

import org.infinispan.client.rest.RestCacheClient;
import org.infinispan.client.rest.RestClient;
import org.infinispan.client.rest.RestEntity;
import org.infinispan.client.rest.RestResponse;
import org.infinispan.client.rest.configuration.RestClientConfigurationBuilder;
import org.infinispan.commons.configuration.JsonWriter;
import org.infinispan.commons.dataconversion.MediaType;
import org.infinispan.commons.test.TestResourceTracker;
import org.infinispan.configuration.cache.ConfigurationBuilder;
import org.infinispan.configuration.global.GlobalConfigurationBuilder;
import org.infinispan.manager.EmbeddedCacheManager;
import org.infinispan.query.Indexer;
import org.infinispan.query.Search;
import org.infinispan.rest.assertion.ResponseAssertion;
import org.infinispan.rest.helper.RestServerHelper;
import org.infinispan.test.SingleCacheManagerTest;
import org.infinispan.test.fwk.TestCacheManagerFactory;
import org.testng.annotations.AfterClass;
import org.testng.annotations.Test;

/**
 * @since 12.0
 */
@Test(groups = "functional", testName = "rest.search.IndexedCacheNonIndexedEntityTest")
public class IndexedCacheNonIndexedEntityTest extends SingleCacheManagerTest {
   private static final String CACHE_NAME = "IndexedCacheNonIndexedEntitiesTest";
   private static final String SCHEMA = "message NonIndexed { optional string name = 1; }";

   protected RestClient client;
   protected RestCacheClient cacheClient;
   private RestServerHelper restServer;

   @Override
   protected EmbeddedCacheManager createCacheManager() {
      GlobalConfigurationBuilder gcb = new GlobalConfigurationBuilder().nonClusteredDefault();
      EmbeddedCacheManager cm = TestCacheManagerFactory.createCacheManager(gcb, getDefaultStandaloneCacheConfig(false));
      restServer = new RestServerHelper(cm);
      restServer.start(TestResourceTracker.getCurrentTestShortName() + "-" + cm.getAddress());

      RestClientConfigurationBuilder clientConfigurationBuilder = new RestClientConfigurationBuilder();
      clientConfigurationBuilder.addServer().host(restServer.getHost()).port(restServer.getPort());
      client = RestClient.forConfiguration(clientConfigurationBuilder.build());

      return cm;
   }

   @AfterClass
   public void tearDown() throws Exception {
      client.close();
      restServer.stop();
   }

   @Test
   public void shouldPreventNonIndexedEntities() {
      CompletionStage<RestResponse> response = client.schemas().post("customer", SCHEMA);
      ResponseAssertion.assertThat(response).isOk();

      ConfigurationBuilder configurationBuilder = getDefaultStandaloneCacheConfig(false);
      configurationBuilder.statistics().enable();
      configurationBuilder.encoding().mediaType(APPLICATION_PROTOSTREAM_TYPE).indexing().enable()
            .storage(LOCAL_HEAP)
            .addIndexedEntity("NonIndexed");
      String config = new JsonWriter().toJSON(configurationBuilder.build());
      RestEntity configEntity = RestEntity.create(MediaType.APPLICATION_JSON, config);

      RestCacheClient cacheClient = client.cache(CACHE_NAME);
      response = cacheClient.createWithConfiguration(configEntity, VOLATILE);
      // The SearchMapping is started lazily, creating and starting the cache will not throw errors
      ResponseAssertion.assertThat(response).isOk();

      // Force initialization of the SearchMapping
      response = cacheClient.query("FROM NonIndexed");
      ResponseAssertion.assertThat(response).isBadRequest();
      String errorText = "The configured indexed-entity type 'NonIndexed' must be indexed. Please annotate it with @Indexed";

      ResponseAssertion.assertThat(response).containsReturnedText(errorText);

      // Call Indexer operations
      response = cacheClient.clearIndex();
      ResponseAssertion.assertThat(response).containsReturnedText(errorText);

      // The Indexer should not have "running" status
      Indexer indexer = Search.getIndexer(cacheManager.getCache(CACHE_NAME));
      assertFalse(indexer.isRunning());
   }
}
