From 17997ef617e6ba5d9b9c70e7e5e8484cd82dbc53 Mon Sep 17 00:00:00 2001
From: Zegveld <41897697+Zegveld@users.noreply.github.com>
Date: Sat, 20 Aug 2022 13:01:47 +0200
Subject: [PATCH] #2901: Fix `@TargetType` annotation on a `@Condition`
 annotated method for a Collection value

---
 .../ap/internal/model/macro/CommonMacros.ftl  |  3 ++-
 ...itionWithTargetTypeOnCollectionMapper.java | 23 +++++++++++++++++++
 .../ap/test/bugs/_2901/Issue2901Test.java     | 22 ++++++++++++++++++
 .../mapstruct/ap/test/bugs/_2901/Source.java  | 21 +++++++++++++++++
 .../mapstruct/ap/test/bugs/_2901/Target.java  | 21 +++++++++++++++++
 5 files changed, 89 insertions(+), 1 deletion(-)
 create mode 100644 processor/src/test/java/org/mapstruct/ap/test/bugs/_2901/ConditionWithTargetTypeOnCollectionMapper.java
 create mode 100644 processor/src/test/java/org/mapstruct/ap/test/bugs/_2901/Issue2901Test.java
 create mode 100644 processor/src/test/java/org/mapstruct/ap/test/bugs/_2901/Source.java
 create mode 100644 processor/src/test/java/org/mapstruct/ap/test/bugs/_2901/Target.java

diff --git a/processor/src/main/resources/org/mapstruct/ap/internal/model/macro/CommonMacros.ftl b/processor/src/main/resources/org/mapstruct/ap/internal/model/macro/CommonMacros.ftl
index e5fb97060..bce28ebe1 100644
--- a/processor/src/main/resources/org/mapstruct/ap/internal/model/macro/CommonMacros.ftl
+++ b/processor/src/main/resources/org/mapstruct/ap/internal/model/macro/CommonMacros.ftl
@@ -60,7 +60,8 @@
 <#macro handleLocalVarNullCheck needs_explicit_local_var>
   <#if sourcePresenceCheckerReference??>
     if ( <@includeModel object=sourcePresenceCheckerReference
-           targetPropertyName=ext.targetPropertyName/> ) {
+           targetType=ext.targetType
+           targetPropertyName=ext.targetPropertyName /> ) {
       <#if needs_explicit_local_var>
         <@includeModel object=nullCheckLocalVarType/> ${nullCheckLocalVarName} = <@lib.handleAssignment/>;
         <#nested>
diff --git a/processor/src/test/java/org/mapstruct/ap/test/bugs/_2901/ConditionWithTargetTypeOnCollectionMapper.java b/processor/src/test/java/org/mapstruct/ap/test/bugs/_2901/ConditionWithTargetTypeOnCollectionMapper.java
new file mode 100644
index 000000000..7e833cfc5
--- /dev/null
+++ b/processor/src/test/java/org/mapstruct/ap/test/bugs/_2901/ConditionWithTargetTypeOnCollectionMapper.java
@@ -0,0 +1,23 @@
+/*
+ * Copyright MapStruct Authors.
+ *
+ * Licensed under the Apache License version 2.0, available at http://www.apache.org/licenses/LICENSE-2.0
+ */
+package org.mapstruct.ap.test.bugs._2901;
+
+import java.util.List;
+
+import org.mapstruct.Condition;
+import org.mapstruct.Mapper;
+import org.mapstruct.TargetType;
+
+@Mapper
+public interface ConditionWithTargetTypeOnCollectionMapper {
+
+    Target map(Source source);
+
+    @Condition
+    default boolean check(List<String> test, @TargetType Class<?> type) {
+        return type.isInstance( test );
+    }
+}
diff --git a/processor/src/test/java/org/mapstruct/ap/test/bugs/_2901/Issue2901Test.java b/processor/src/test/java/org/mapstruct/ap/test/bugs/_2901/Issue2901Test.java
new file mode 100644
index 000000000..3ae2c3db8
--- /dev/null
+++ b/processor/src/test/java/org/mapstruct/ap/test/bugs/_2901/Issue2901Test.java
@@ -0,0 +1,22 @@
+/*
+ * Copyright MapStruct Authors.
+ *
+ * Licensed under the Apache License version 2.0, available at http://www.apache.org/licenses/LICENSE-2.0
+ */
+package org.mapstruct.ap.test.bugs._2901;
+
+import org.mapstruct.ap.testutil.IssueKey;
+import org.mapstruct.ap.testutil.ProcessorTest;
+import org.mapstruct.ap.testutil.WithClasses;
+
+/**
+ * @author Ben Zegveld
+ */
+@IssueKey( "2901" )
+class Issue2901Test {
+
+    @ProcessorTest
+    @WithClasses( { Source.class, Target.class, ConditionWithTargetTypeOnCollectionMapper.class } )
+    void shouldCompile() {
+    }
+}
diff --git a/processor/src/test/java/org/mapstruct/ap/test/bugs/_2901/Source.java b/processor/src/test/java/org/mapstruct/ap/test/bugs/_2901/Source.java
new file mode 100644
index 000000000..4eb50d58f
--- /dev/null
+++ b/processor/src/test/java/org/mapstruct/ap/test/bugs/_2901/Source.java
@@ -0,0 +1,21 @@
+/*
+ * Copyright MapStruct Authors.
+ *
+ * Licensed under the Apache License version 2.0, available at http://www.apache.org/licenses/LICENSE-2.0
+ */
+package org.mapstruct.ap.test.bugs._2901;
+
+import java.util.List;
+
+public class Source {
+
+    private List<String> field;
+
+    public List<String> getField() {
+        return field;
+    }
+
+    public void setField(List<String> field) {
+        this.field = field;
+    }
+}
diff --git a/processor/src/test/java/org/mapstruct/ap/test/bugs/_2901/Target.java b/processor/src/test/java/org/mapstruct/ap/test/bugs/_2901/Target.java
new file mode 100644
index 000000000..b4c5299f3
--- /dev/null
+++ b/processor/src/test/java/org/mapstruct/ap/test/bugs/_2901/Target.java
@@ -0,0 +1,21 @@
+/*
+ * Copyright MapStruct Authors.
+ *
+ * Licensed under the Apache License version 2.0, available at http://www.apache.org/licenses/LICENSE-2.0
+ */
+package org.mapstruct.ap.test.bugs._2901;
+
+import java.util.List;
+
+public class Target {
+
+    private List<Integer> field;
+
+    public List<Integer> getField() {
+        return field;
+    }
+
+    public void setField(List<Integer> field) {
+        this.field = field;
+    }
+}