From 919d8c663fba1787b73dd03a3ee3d81956044066 Mon Sep 17 00:00:00 2001
From: Jens Schauder <jschauder@pivotal.io>
Date: Thu, 7 May 2020 09:37:09 +0200
Subject: [PATCH] #364 - Add lock() to Dialect implementation for tests.

---
 .../r2dbc/dialect/DialectResolverUnitTests.java | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/src/test/java/org/springframework/data/r2dbc/dialect/DialectResolverUnitTests.java b/src/test/java/org/springframework/data/r2dbc/dialect/DialectResolverUnitTests.java
index b38f984..993ed99 100644
--- a/src/test/java/org/springframework/data/r2dbc/dialect/DialectResolverUnitTests.java
+++ b/src/test/java/org/springframework/data/r2dbc/dialect/DialectResolverUnitTests.java
@@ -22,6 +22,8 @@ import org.junit.Test;
 import org.reactivestreams.Publisher;
 
 import org.springframework.data.relational.core.dialect.LimitClause;
+import org.springframework.data.relational.core.dialect.LockClause;
+import org.springframework.data.relational.core.sql.LockOptions;
 import org.springframework.data.relational.core.sql.render.SelectRenderContext;
 
 import com.github.jasync.r2dbc.mysql.JasyncConnectionFactory;
@@ -107,6 +109,21 @@ public class DialectResolverUnitTests {
 			return null;
 		}
 
+		@Override
+		public LockClause lock() {
+			return new LockClause() {
+				@Override
+				public String getLock(LockOptions lockOptions) {
+					return "FOR UPDATE";
+				}
+
+				@Override
+				public Position getClausePosition() {
+					return Position.AFTER_ORDER_BY;
+				}
+			};
+		}
+
 		@Override
 		public SelectRenderContext getSelectContext() {
 			return null;