Wrap Queries Containing TYPEOF

octoberswimmer · Jan 14, 2025 · 68ec2e3 · 68ec2e3
1 parent 8e2a540
commit 68ec2e3
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/formatter/chain.go b/formatter/chain.go
@@ -124,6 +124,9 @@ func (v *ChainVisitor) VisitSelectEntry(ctx *parser.SelectEntryContext) interfac
 		// estimate complexity; probably close enough
 		return 5
 	}
+	if ctx.TypeOf() != nil {
+		return 3
+	}
 	return 1
 }
 

diff --git a/formatter/format_test.go b/formatter/format_test.go
@@ -352,6 +352,32 @@ b);`,
 			.save();
 }`,
 			},
+			{
+				`return Database.getQueryLocator(
+   [
+      SELECT
+         TYPEOF What
+            WHEN Account_Profile__c THEN Account__c, Marketing_Team__c
+         END
+      FROM Event
+      WHERE WhatId IN :this.oldAccountProfileIds
+   ]
+);`,
+				`return Database.getQueryLocator([
+	SELECT
+		TYPEOF What
+		WHEN
+			Account_Profile__c
+		THEN
+			Account__c,
+			Marketing_Team__c
+		END
+	FROM
+		Event
+	WHERE
+		WhatId IN :this.oldAccountProfileIds
+]);`,
+			},
 		}
 	dmp := diffmatchpatch.New()
 	for i, tt := range tests {