CavalliumDBEngine-Server/src/test/java/it/cavallium/dbengine/database/server/LLQuicConnectionTest.java

package it.cavallium.dbengine.database.server;

import static org.junit.jupiter.api.Assertions.assertDoesNotThrow;
import static org.junit.jupiter.api.Assertions.assertEquals;

import io.micrometer.core.instrument.composite.CompositeMeterRegistry;
import io.netty5.buffer.BufferAllocator;
import io.netty5.buffer.DefaultBufferAllocators;
import it.cavallium.data.generator.nativedata.Nullableboolean;
import it.cavallium.data.generator.nativedata.Nullabledouble;
import it.cavallium.data.generator.nativedata.Nullableint;
import it.cavallium.dbengine.client.DefaultDatabaseOptions;
import it.cavallium.dbengine.client.IndicizerAnalyzers;
import it.cavallium.dbengine.client.IndicizerSimilarities;
import it.cavallium.dbengine.database.ColumnUtils;
import it.cavallium.dbengine.database.LLDatabaseConnection;
import it.cavallium.dbengine.database.LLUtils;
import it.cavallium.dbengine.lucene.LuceneUtils;
import it.cavallium.dbengine.rpc.current.data.ByteBuffersDirectory;
import it.cavallium.dbengine.rpc.current.data.LuceneOptions;
import java.time.Duration;
import java.util.List;
import java.util.Map;
import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;

public class LLQuicConnectionTest {

	private BufferAllocator allocator;
	private CompositeMeterRegistry meterRegistry;
	private TestDbServer server;
	private LLDatabaseConnection client;

	@BeforeEach
	void setUp() {
		this.allocator = DefaultBufferAllocators.preferredAllocator();
		this.meterRegistry = new CompositeMeterRegistry();
		this.server = TestDbServer.create(allocator, meterRegistry);
		server.bind().transform(LLUtils::handleDiscard).block();
		this.client = TestDbClient.create(allocator, meterRegistry, server.address()).connect().block();
	}

	@AfterEach
	void tearDown() {
		if (client != null) {
			client.disconnect().block();
		}
		if (server != null) {
			server.dispose().block();
		}
	}

	@Test
	void getAllocator() {
		assertEquals(allocator, client.getAllocator());
	}

	@Test
	void getMeterRegistry() {
		assertEquals(meterRegistry, client.getMeterRegistry());
	}

	@Test
	void getDatabase() {
		var dbName = "test-temp-db";
		var singletonsColumnName = "singletons";
		var db = client.getDatabase(dbName,
				List.of(ColumnUtils.special(singletonsColumnName)),
				DefaultDatabaseOptions.DEFAULT_DATABASE_OPTIONS
		).blockOptional().orElseThrow();
		assertEquals(dbName, db.getDatabaseName());
		assertEquals(allocator, db.getAllocator());
		assertEquals(meterRegistry, db.getMeterRegistry());
		assertDoesNotThrow(() -> db.close().block());
	}

	@Test
	void getLuceneIndex() {
		var shardName = "test-lucene-shard";
		var index = client.getLuceneIndex(shardName,
				LuceneUtils.singleStructure(),
				IndicizerAnalyzers.of(),
				IndicizerSimilarities.of(),
				new LuceneOptions(Map.of(),
						Duration.ofSeconds(1),
						Duration.ofSeconds(1),
						false,
						new ByteBuffersDirectory(),
						Nullableboolean.empty(),
						Nullabledouble.empty(),
						Nullableint.empty(),
						Nullableboolean.empty(),
						Nullableboolean.empty(),
						false,
						100,
						LuceneUtils.getDefaultMergePolicy()
				),
				null).blockOptional().orElseThrow();
		assertEquals(shardName, index.getLuceneIndexName());
		assertDoesNotThrow(index::close);
	}
}