io.trino.tests.product.deltalake.BaseTestDeltaLakeMinioReads Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of trino-product-tests Show documentation
Trino - Product tests
There is a newer version: 464
/*
 * Licensed 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.
 */
package io.trino.tests.product.deltalake;

import io.trino.tempto.AfterMethodWithContext;
import io.trino.tempto.BeforeMethodWithContext;
import io.trino.tempto.ProductTest;
import io.trino.testing.minio.MinioClient;
import org.testng.annotations.Test;

import static io.minio.messages.EventType.OBJECT_ACCESSED_GET;
import static io.minio.messages.EventType.OBJECT_ACCESSED_HEAD;
import static io.trino.tempto.assertions.QueryAssert.Row.row;
import static io.trino.tests.product.TestGroups.DELTA_LAKE_MINIO;
import static io.trino.tests.product.TestGroups.PROFILE_SPECIFIC_TESTS;
import static io.trino.tests.product.utils.MinioNotificationsAssertions.assertNotificationsCount;
import static io.trino.tests.product.utils.MinioNotificationsAssertions.createNotificationsTable;
import static io.trino.tests.product.utils.MinioNotificationsAssertions.deleteNotificationsTable;
import static io.trino.tests.product.utils.MinioNotificationsAssertions.recordNotification;
import static io.trino.tests.product.utils.QueryExecutors.onTrino;
import static java.lang.String.format;
import static java.util.Objects.requireNonNull;
import static org.assertj.core.api.Assertions.assertThat;

public abstract class BaseTestDeltaLakeMinioReads
        extends ProductTest
{
    protected static final String BUCKET_NAME = "delta-test-basic-reads";
    protected static final String NOTIFICATIONS_TABLE = "read_region_notifications";

    protected MinioClient client;

    protected String tableName;
    protected String regionResourcePath;

    public BaseTestDeltaLakeMinioReads(String regionTableName, String regionResourcePath)
    {
        this.tableName = requireNonNull(regionTableName, "regionTableName is null");
        this.regionResourcePath = requireNonNull(regionResourcePath, "regionResourcePath is null");
    }

    @BeforeMethodWithContext
    public void setUp()
    {
        client = new MinioClient();

        deleteNotificationsTable(NOTIFICATIONS_TABLE);
        createNotificationsTable(NOTIFICATIONS_TABLE);

        client.copyResourcePath(BUCKET_NAME, regionResourcePath, tableName);
        client.captureBucketNotifications(BUCKET_NAME, notification -> recordNotification(NOTIFICATIONS_TABLE, notification));
    }

    @AfterMethodWithContext
    public void tearDown()
    {
        deleteNotificationsTable(NOTIFICATIONS_TABLE);
        client.close();
        client = null;
    }

    @Test(groups = {DELTA_LAKE_MINIO, PROFILE_SPECIFIC_TESTS})
    public void testReadRegionTable()
    {
        onTrino().executeQuery(format("CALL delta.system.register_table('default', '%1$s', 's3://%2$s/%1$s')", tableName, BUCKET_NAME));

        assertThat(onTrino().executeQuery(
                format("SELECT count(name) FROM delta.default.\"%s\"", tableName)))
                .containsOnly(row(5L));

        assertNotificationsCount(NOTIFICATIONS_TABLE, OBJECT_ACCESSED_HEAD, tableName + "/_delta_log/00000000000000000000.json", 0);
        assertNotificationsCount(NOTIFICATIONS_TABLE, OBJECT_ACCESSED_GET, tableName + "/_delta_log/00000000000000000000.json", 1);
        onTrino().executeQuery(format("DROP TABLE delta.default.\"%s\"", tableName));
    }
}