/* SPDX-License-Identifier: Apache-2.0
*
* The OpenSearch Contributors require contributions made to
* this file be licensed under the Apache-2.0 license or a
* compatible open source license.
*/
/*
* Modifications Copyright OpenSearch Contributors. See
* GitHub history for details.
*
*  Licensed to Elasticsearch B.V. under one or more contributor
*  license agreements. See the NOTICE file distributed with
*  this work for additional information regarding copyright
*  ownership. Elasticsearch B.V. licenses this file to you under
*  the Apache License, Version 2.0 (the "License"); you may
*  not use this file except in compliance with the License.
*  You may obtain a copy of the License at
*
* 	http://www.apache.org/licenses/LICENSE-2.0
*
*  Unless required by applicable law or agreed to in writing,
*  software distributed under the License is distributed on an
*  "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
*  KIND, either express or implied.  See the License for the
*  specific language governing permissions and limitations
*  under the License.
*/

using System;
using System.Threading.Tasks;
using OpenSearch.OpenSearch.Xunit;
using OpenSearch.OpenSearch.Xunit.XunitPlumbing;
using FluentAssertions;
using OpenSearch.Client;
using OpenSearch.Client.Specification.IndicesApi;
using Tests.Core.Client;
using Tests.Core.ManagedOpenSearch.Clusters;
using Tests.Core.Serialization;
using Tests.Framework.EndpointTests.TestState;

namespace Tests.Analysis
{
	public interface IAnalysisAssertion
	{
		object Json { get; }
		string Name { get; }
	}

	public interface IAnalysisAssertion<out TComponent, out TContainer, in TDescriptor> : IAnalysisAssertion
		where TContainer : class
	{
		Func<string, TDescriptor, IPromise<TContainer>> Fluent { get; }
		TComponent Initializer { get; }
	}

	[IntegrationTestCluster(typeof(WritableCluster))]
	public abstract class AnalysisComponentTestBase<TAssertion, TComponent, TContainer, TDescriptor>
		: IAnalysisAssertion<TComponent, TContainer, TDescriptor>
		where TAssertion : AnalysisComponentTestBase<TAssertion, TComponent, TContainer, TDescriptor>, new()
		where TContainer : class
	{
		private static readonly SingleEndpointUsage<CreateIndexResponse> Usage = new SingleEndpointUsage<CreateIndexResponse>
		(
			(s, c) => c.Indices.Create(s, AssertionSetup.FluentCall),
			(s, c) => c.Indices.CreateAsync(s, AssertionSetup.FluentCall),
			(s, c) => c.Indices.Create(AssertionSetup.InitializerCall(s)),
			(s, c) => c.Indices.CreateAsync(AssertionSetup.InitializerCall(s)),
			$"test-{typeof(TAssertion).Name.ToLowerInvariant()}"
		)
		{
			OnAfterCall = c => c.Indices.Delete(Usage.CallUniqueValues.Value)
		};

		protected AnalysisComponentTestBase()
		{
			Client = (OpenSearchXunitRunner.CurrentCluster as IOpenSearchClientTestCluster)?.Client ?? TestClient.DefaultInMemoryClient;
			Usage.KickOffOnce(Client, true);
		}

		public abstract Func<string, TDescriptor, IPromise<TContainer>> Fluent { get; }
		public abstract TComponent Initializer { get; }
		public abstract object Json { get; }

		public abstract string Name { get; }

		protected abstract object AnalysisJson { get; }
		protected static TAssertion AssertionSetup { get; } = new TAssertion();

		private IOpenSearchClient Client { get; }

		private Func<CreateIndexDescriptor, ICreateIndexRequest> FluentCall => i => i.Settings(s => s.Analysis(FluentAnalysis));

		protected abstract IAnalysis FluentAnalysis(AnalysisDescriptor an);

		private CreateIndexRequest InitializerCall(string index) => new CreateIndexRequest(index)
		{
			Settings = new IndexSettings { Analysis = InitializerAnalysis() }
		};

		protected abstract OpenSearch.Client.Analysis InitializerAnalysis();

		[U] public virtual async Task TestPutSettingsRequest() => await Usage.AssertOnAllResponses(r =>
		{
			var json = new { settings = new { analysis = AnalysisJson } };
			SerializationTestHelper.Expect(json).FromRequest(r);
		});

		[I] public virtual async Task TestPutSettingsResponse() => await Usage.AssertOnAllResponses(r =>
		{
			r.ApiCall.HttpStatusCode.Should().Be(200);
		});
	}
}