spring-projects#2168 No TemplateVariable generated for null Pageable

Author: reda-alaoui

pom.xml

@@ -34,6 +34,8 @@
 		<xmlbeam>1.4.26</xmlbeam>
 		<java-module-name>spring.data.commons</java-module-name>
 		<kotlin.api.target>1.8</kotlin.api.target>
+
+		<spring-hateoas>2.2.0.1.cosium</spring-hateoas>
 	</properties>
 
 	<dependencies>

src/main/java/org/springframework/data/web/HateoasPageableHandlerMethodArgumentResolver.java

@@ -39,6 +39,7 @@
  * @since 1.6
  * @author Oliver Gierke
  * @author Nick Williams
+ * @author Réda Housni Alaoui
  */
 @SuppressWarnings("null")
 public class HateoasPageableHandlerMethodArgumentResolver extends PageableHandlerMethodArgumentResolver
@@ -122,6 +123,11 @@ public void enhance(UriComponentsBuilder builder, @Nullable MethodParameter para
 		this.sortResolver.enhance(builder, parameter, pageable.getSort());
 	}
 
+	@Override
+	public TemplateVariables enhance(TemplateVariables templateVariables, UriComponents uriComponents, MethodParameter parameter) {
+		return templateVariables.concat(getPaginationTemplateVariables(parameter, uriComponents));
+	}
+
 	private static HateoasSortHandlerMethodArgumentResolver getDefaultedSortResolver(
 			@Nullable HateoasSortHandlerMethodArgumentResolver sortResolver) {
 		return sortResolver == null ? DEFAULT_SORT_RESOLVER : sortResolver;

src/main/java/org/springframework/data/web/HateoasSortHandlerMethodArgumentResolver.java

@@ -36,6 +36,7 @@
  * @author Thomas Darimont
  * @author Nick Williams
  * @author Julien Béti
+ * @author Réda Housni Alaoui
  */
 public class HateoasSortHandlerMethodArgumentResolver extends SortHandlerMethodArgumentResolver
 		implements UriComponentsContributor {
@@ -77,4 +78,9 @@ public void enhance(UriComponentsBuilder builder, @Nullable MethodParameter para
 			builder.queryParam(sortParameter, expression);
 		}
 	}
+
+	@Override
+	public TemplateVariables enhance(TemplateVariables templateVariables, UriComponents uriComponents, MethodParameter parameter) {
+		return templateVariables.concat(getSortTemplateVariables(parameter, uriComponents));
+	}
 }

src/test/java/org/springframework/data/web/HateoasPageableHandlerMethodArgumentResolverUnitTests.java

@@ -21,14 +21,17 @@
 import org.springframework.core.MethodParameter;
 import org.springframework.data.domain.PageRequest;
 import org.springframework.data.domain.Pageable;
+import org.springframework.hateoas.TemplateVariables;
 import org.springframework.hateoas.server.mvc.UriComponentsContributor;
+import org.springframework.web.util.UriComponents;
 import org.springframework.web.util.UriComponentsBuilder;
 
 /**
  * Unit tests for {@link HateoasPageableHandlerMethodArgumentResolver}.
  *
  * @author Oliver Gierke
  * @author Julien Béti
+ * @author Réda Housni Alaoui
  */
 class HateoasPageableHandlerMethodArgumentResolverUnitTests
 		extends PageableHandlerMethodArgumentResolverUnitTests {
@@ -116,6 +119,18 @@ void enhancesUnpaged() {
 		assertThat(builder).isEqualTo(builder);
 	}
 
+	@Test // DATACMNS-1752
+	void enhanceTemplateVariables() {
+
+		UriComponents uriComponents = UriComponentsBuilder.fromPath("/foo").build();
+
+		HateoasPageableHandlerMethodArgumentResolver resolver = getResolver();
+		resolver.setPageParameterName("foo");
+		String variables = resolver.enhance(TemplateVariables.NONE, uriComponents, null).toString();
+
+		assertThat(variables).isEqualTo("{?foo,size,sort*}");
+	}
+
 	@Override
 	protected HateoasPageableHandlerMethodArgumentResolver getResolver() {
 

src/test/java/org/springframework/data/web/HateoasSortHandlerMethodArgumentResolverUnitTests.java

@@ -22,13 +22,16 @@
 
 import org.junit.jupiter.api.Test;
 import org.springframework.data.domain.Sort;
+import org.springframework.hateoas.TemplateVariables;
+import org.springframework.web.util.UriComponents;
 import org.springframework.web.util.UriComponentsBuilder;
 
 /**
  * Unit tests for {@link HateoasSortHandlerMethodArgumentResolver}
  *
  * @author Oliver Gierke
  * @author Julien Béti
+ * @author Réda Housni Alaoui
  */
 class HateoasSortHandlerMethodArgumentResolverUnitTests extends SortHandlerMethodArgumentResolverUnitTests {
 
@@ -56,6 +59,15 @@ void returnCorrectTemplateVariables() {
 		assertThat(resolver.getSortTemplateVariables(null, uriComponents).toString()).isEqualTo("{?sort*}");
 	}
 
+	@Test // DATACMNS-1752
+	void enhanceTemplateVariables(){
+
+		UriComponents uriComponents = UriComponentsBuilder.fromPath("/").build();
+
+		HateoasSortHandlerMethodArgumentResolver resolver = new HateoasSortHandlerMethodArgumentResolver();
+		assertThat(resolver.enhance(TemplateVariables.NONE, uriComponents, null).toString()).isEqualTo("{?sort*}");
+	}
+
 	private void assertUriStringFor(Sort sort, String expected) throws Exception {
 		assertUriStringFor(sort, expected, "/");
 	}