From 9cf4bd8251ad56f58ae5c802d2fdb57f970350d4 Mon Sep 17 00:00:00 2001
From: Alex <alex.coolen@duo.nl>
Date: Thu, 5 Sep 2024 09:19:02 +0200
Subject: [PATCH] HHH-18560: added unittest

---
 .../hibernate/dialect/DB2iDialectTest.java    | 69 +++++++++++++++++++
 1 file changed, 69 insertions(+)
 create mode 100644 hibernate-core/src/test/java/org/hibernate/dialect/DB2iDialectTest.java

diff --git a/hibernate-core/src/test/java/org/hibernate/dialect/DB2iDialectTest.java b/hibernate-core/src/test/java/org/hibernate/dialect/DB2iDialectTest.java
new file mode 100644
index 0000000000..4167d5d9dd
--- /dev/null
+++ b/hibernate-core/src/test/java/org/hibernate/dialect/DB2iDialectTest.java
@@ -0,0 +1,69 @@
+package org.hibernate.dialect;
+
+import org.hibernate.testing.orm.junit.JiraKey;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+
+class DB2iDialectTest {
+
+    private static final String EXPECTED_FOR_UPDATE = " for update with rs";
+    private static final String EXPECTED_FOR_UPDATE_SKIP_LOCK = " for update with rs skip locked data";
+
+    private DB2iDialect dialect;
+
+    @BeforeEach
+    void setUp() {
+        dialect = new DB2iDialect();
+    }
+
+    @Test
+    @JiraKey("HHH-18560")
+    void getForUpdateString() {
+        String actual = dialect.getForUpdateString();
+        assertEquals(EXPECTED_FOR_UPDATE, actual);
+    }
+
+    @Test
+    @JiraKey("HHH-18560")
+    void getForUpdateSkipLockedString() {
+        String actual = dialect.getForUpdateSkipLockedString();
+        assertEquals(EXPECTED_FOR_UPDATE_SKIP_LOCK, actual);
+    }
+
+    @Test
+    @JiraKey("HHH-18560")
+    void testGetForUpdateSkipLockedString() {
+        String actual = dialect.getForUpdateSkipLockedString("alias");
+        assertEquals(EXPECTED_FOR_UPDATE_SKIP_LOCK, actual);
+    }
+
+    @Test
+    @JiraKey("HHH-18560")
+    void getWriteLockString_skiplocked() {
+        String actual = dialect.getWriteLockString(-2);
+        assertEquals(EXPECTED_FOR_UPDATE_SKIP_LOCK, actual);
+    }
+
+    @Test
+    @JiraKey("HHH-18560")
+    void getWriteLockString() {
+        String actual = dialect.getWriteLockString(0);
+        assertEquals(EXPECTED_FOR_UPDATE, actual);
+    }
+
+    @Test
+    @JiraKey("HHH-18560")
+    void getReadLockString() {
+        String actual = dialect.getReadLockString(0);
+        assertEquals(EXPECTED_FOR_UPDATE, actual);
+    }
+
+    @Test
+    @JiraKey("HHH-18560")
+    void getReadLockString_skipLocked() {
+        String actual = dialect.getReadLockString(-2);
+        assertEquals(EXPECTED_FOR_UPDATE_SKIP_LOCK, actual);
+    }
+}