Merge pull request #51 from adangel/sosl-tolabel

Support toLabel in SOSL queries

Author: pwrightcertinia

antlr/ApexParser.g4

@@ -865,6 +865,7 @@ fieldSpec
 
 fieldList
     : soslId (COMMA fieldList)*
+    | TOLABEL LPAREN soslId RPAREN
     ;
 
 updateList

jvm/src/test/java/io/github/apexdevtools/apexparser/SOSLParserTest.java

@@ -76,4 +76,12 @@ void testWithSystemModeQuery() {
         assertNotNull(context);
         assertEquals(0, parserAndCounter.getValue().getNumErrors());
     }
+
+    @Test
+    void testToLabel() {
+        Map.Entry<ApexParser, SyntaxErrorCounter> parserAndCounter = createParser("[FIND :searchTerm IN ALL FIELDS RETURNING Account(Id, toLabel(Name)) LIMIT 10]");
+        ApexParser.SoslLiteralContext context = parserAndCounter.getKey().soslLiteral();
+        assertNotNull(context);
+        assertEquals(0, parserAndCounter.getValue().getNumErrors());
+    }
 }

npm/src/__tests__/SOSLParserTest.ts

@@ -65,17 +65,26 @@ test('testQuotesFailOnAltFormat', () => {
 test('testWithUserModeQuery', () => {
     const [parser, errorCounter] = createParser("[Find 'something' RETURNING Account WITH USER_MODE WITH METADATA='Labels']")
 
-    const context = parser.soslLiteralAlt()
+    const context = parser.soslLiteral()
 
-    expect(context).toBeInstanceOf(SoslLiteralAltContext)
-    expect(errorCounter.getNumErrors()).toEqual(1)
+    expect(context).toBeInstanceOf(SoslLiteralContext)
+    expect(errorCounter.getNumErrors()).toEqual(0)
 })
 
 test('testWithSystemModeQuery', () => {
     const [parser, errorCounter] = createParser("[Find 'something' RETURNING Account WITH METADATA='Labels' WITH SYSTEM_MODE]")
 
-    const context = parser.soslLiteralAlt()
+    const context = parser.soslLiteral()
 
-    expect(context).toBeInstanceOf(SoslLiteralAltContext)
-    expect(errorCounter.getNumErrors()).toEqual(1)
+    expect(context).toBeInstanceOf(SoslLiteralContext)
+    expect(errorCounter.getNumErrors()).toEqual(0)
+})
+
+test('testToLabel', () => {
+    const [parser, errorCounter] = createParser("[FIND :searchTerm IN ALL FIELDS RETURNING Account(Id, toLabel(Name)) LIMIT 10]")
+
+    const context = parser.soslLiteral()
+
+    expect(context).toBeInstanceOf(SoslLiteralContext)
+    expect(errorCounter.getNumErrors()).toEqual(0)
 })