hibernate-dialect/src/test/java/tech/ydb/hibernate/datetime/DataTimeTests.java - main - githubmirror/ydb-java-dialects

hibernate-dialect/src/test/java/tech/ydb/hibernate/datetime/DataTimeTests.java

63 строки2 KB

Kirill Kurdyukov

Unify Hibernate 6 and 7 dialect into single module (#228)

01 июл 2026, 13:21

01 июл 2026, 13:21eeb4e73

package tech.ydb.hibernate.datetime;

import java.time.Instant;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.time.temporal.ChronoUnit;
import org.hibernate.cfg.AvailableSettings;
import static org.junit.jupiter.api.Assertions.assertEquals;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.RegisterExtension;
import static tech.ydb.hibernate.TestUtils.SESSION_FACTORY;
import static tech.ydb.hibernate.TestUtils.basedConfiguration;
import static tech.ydb.hibernate.TestUtils.inTransaction;
import static tech.ydb.hibernate.TestUtils.jdbcUrl;
import tech.ydb.test.junit5.YdbHelperExtension;

/**
 * @author Kirill Kurdyukov
 */
public class DataTimeTests {

    @RegisterExtension
    private static final YdbHelperExtension ydb = new YdbHelperExtension();

    @Test
    void integrationTest() {
        SESSION_FACTORY = basedConfiguration()
                .addAnnotatedClass(TestEntity.class)
                .setProperty(AvailableSettings.URL, jdbcUrl(ydb))
                .buildSessionFactory();

        var date = LocalDate.now();
        var datetime = LocalDateTime.now().truncatedTo(ChronoUnit.SECONDS);
        var timestamp = Instant.now().truncatedTo(ChronoUnit.MICROS);

        var expected = new TestEntity();
        expected.setId(1);
        expected.setDate(date);
        expected.setDatetime(datetime);
        expected.setTimestamp(timestamp);
        expected.setDate32(date);
        expected.setDatetime64(datetime);
        expected.setTimestamp64(timestamp);

        inTransaction(session -> session.persist(expected));
        inTransaction(session -> assertEquals(expected, session.find(TestEntity.class, 1)));

        expected.setDatetime(datetime.plusDays(1));

        inTransaction(session -> session.merge(expected));
        inTransaction(session -> assertEquals(expected, session.find(TestEntity.class, 1)));

        expected.setDate(date.plusDays(1));

        inTransaction(session -> session.merge(expected));
        inTransaction(session -> assertEquals(expected, session.find(TestEntity.class, 1)));

        expected.setTimestamp(timestamp.plus(1, ChronoUnit.DAYS));

        inTransaction(session -> session.merge(expected));
        inTransaction(session -> assertEquals(expected, session.find(TestEntity.class, 1)));
    }
}