From 26f148789c6ce150577286c80f348da7b59909a7 Mon Sep 17 00:00:00 2001
From: Albert Meltzer <7529386+kitbellew@users.noreply.github.com>
Date: Sat, 28 Sep 2024 08:43:47 -0700
Subject: [PATCH] Test partial function within apply

---
 .../resources/newlines/source_classic.stat    | 14 +++++++++++
 .../test/resources/newlines/source_fold.stat  | 14 +++++++++++
 .../test/resources/newlines/source_keep.stat  | 23 +++++++++++++++++++
 .../resources/newlines/source_unfold.stat     | 16 +++++++++++++
 4 files changed, 67 insertions(+)

diff --git a/scalafmt-tests/shared/src/test/resources/newlines/source_classic.stat b/scalafmt-tests/shared/src/test/resources/newlines/source_classic.stat
index 855af52694..e6e1efbe63 100644
--- a/scalafmt-tests/shared/src/test/resources/newlines/source_classic.stat
+++ b/scalafmt-tests/shared/src/test/resources/newlines/source_classic.stat
@@ -9662,3 +9662,17 @@ object a {
     }
   }
 }
+<<< #4133 partial function within apply, short
+rdd.map(
+    {case (id, count) => (count, id)})
+>>>
+rdd.map({ case (id, count) =>
+  (count, id)
+})
+<<< #4133 partial function within apply, long
+maxColumn = 80
+===
+rdd.map(
+    {case (id, count) => (count, id)})
+>>>
+rdd.map({ case (id, count) => (count, id) })
diff --git a/scalafmt-tests/shared/src/test/resources/newlines/source_fold.stat b/scalafmt-tests/shared/src/test/resources/newlines/source_fold.stat
index 7ffd2bd256..731b841de5 100644
--- a/scalafmt-tests/shared/src/test/resources/newlines/source_fold.stat
+++ b/scalafmt-tests/shared/src/test/resources/newlines/source_fold.stat
@@ -9062,3 +9062,17 @@ object a {
       }
   }
 }
+<<< #4133 partial function within apply, short
+rdd.map(
+    {case (id, count) => (count, id)})
+>>>
+rdd.map({ case (id, count) =>
+  (count, id)
+})
+<<< #4133 partial function within apply, long
+maxColumn = 80
+===
+rdd.map(
+    {case (id, count) => (count, id)})
+>>>
+rdd.map({ case (id, count) => (count, id) })
diff --git a/scalafmt-tests/shared/src/test/resources/newlines/source_keep.stat b/scalafmt-tests/shared/src/test/resources/newlines/source_keep.stat
index b9a84b2ca5..f849f2c014 100644
--- a/scalafmt-tests/shared/src/test/resources/newlines/source_keep.stat
+++ b/scalafmt-tests/shared/src/test/resources/newlines/source_keep.stat
@@ -9459,3 +9459,26 @@ object a {
     }
   }
 }
+<<< #4133 partial function within apply, short
+rdd.map(
+    {case (id, count) => (count, id)})
+>>>
+Idempotency violated
+=> Diff (- obtained, + expected)
+ rdd.map({ case (id, count) =>
+-  (count, id)
+-})
++    (count, id)
++  }
++)
+<<< #4133 partial function within apply, long
+maxColumn = 80
+===
+rdd.map(
+    {case (id, count) => (count, id)})
+>>>
+Idempotency violated
+=> Diff (- obtained, + expected)
+-rdd.map({ case (id, count) => (count, id) })
++rdd.map({ case (id, count) => (count, id) }
++)
diff --git a/scalafmt-tests/shared/src/test/resources/newlines/source_unfold.stat b/scalafmt-tests/shared/src/test/resources/newlines/source_unfold.stat
index 92f0435190..9d30b5cf52 100644
--- a/scalafmt-tests/shared/src/test/resources/newlines/source_unfold.stat
+++ b/scalafmt-tests/shared/src/test/resources/newlines/source_unfold.stat
@@ -9739,3 +9739,19 @@ object a {
       }
   }
 }
+<<< #4133 partial function within apply, short
+rdd.map(
+    {case (id, count) => (count, id)})
+>>>
+rdd.map({ case (id, count) =>
+  (count, id)
+})
+<<< #4133 partial function within apply, long
+maxColumn = 80
+===
+rdd.map(
+    {case (id, count) => (count, id)})
+>>>
+rdd.map({ case (id, count) =>
+  (count, id)
+})