adding more rlang tests to match pre-existing lua and python ones

Signed-off-by: Elijah Cúchulainn Reid <[email protected]>

Author: Wizzzzzzard

client/test/components/util/SASCodeDocument.test.ts

@@ -38,6 +38,40 @@ run;
     assert.equal(sasCodeDoc.getWrappedCode(), expected);
   });
 
+  it("wrap rlang code", () => {
+    const parameters: SASCodeDocumentParameters = {
+      languageId: "rlang",
+      code: `for (x in 1:6) {
+  print(x)
+}
+print("test")`,
+      selectedCode: "",
+      htmlStyle: "Illuminate",
+      outputHtml: true,
+      uuid: "519058ad-d33b-4b5c-9d23-4cc8d6ffb163",
+      checkKeyword: async () => false,
+    };
+
+    const sasCodeDoc = new SASCodeDocument(parameters);
+
+    const expected = `/** LOG_START_INDICATOR **/
+title;footnote;ods _all_ close;
+ods graphics on;
+ods html5(id=vscode) style=Illuminate options(bitmap_mode='inline' svg_mode='inline') body="519058ad-d33b-4b5c-9d23-4cc8d6ffb163.htm";
+proc rlang;
+submit;
+for (x in 1:6) {
+  print(x)
+}
+print("test")
+endsubmit;
+run;
+;*';*";*/;run;quit;ods html5(id=vscode) close;
+`;
+
+    assert.equal(sasCodeDoc.getWrappedCode(), expected);
+  });
+
   it("wrap sql code", () => {
     const parameters: SASCodeDocumentParameters = {
       languageId: "sql",

client/testFixture/formatter/expected.sas

@@ -46,6 +46,16 @@ def my_function():
 endsubmit;
 run;
 
+proc rlang;
+submit;
+# Reference to variable defined in previous PROC RLANG call
+print(paste("x =", x))
+my_function <- function() {
+    print("Inside the proc step")
+}
+endsubmit;
+run;
+
 proc lua;
 submit;
 local dsid = sas.open("sashelp.company") -- open for input            
@@ -131,6 +141,17 @@ print('first statement after for loop')
 endinteractive;
 run;
 
+proc rlang;
+submit;
+fruits <- c("apple", "banana", "cherry")
+for (x in fruits) {
+   print(x)
+}
+
+print('first statement after for loop')
+endsubmit;
+run;
+
 proc lua;
 submit;
 

client/testFixture/formatter/unformatted.sas

@@ -40,6 +40,15 @@ def my_function():
     print("Inside the proc step")
 endsubmit;
 run;
+proc rlang;
+submit;
+# Reference to variable defined in previous PROC RLANG call
+print(paste("x =", x))
+my_function <- function() {
+    print("Inside the proc step")
+}
+endsubmit;
+run;
 proc lua;
 submit;
 local dsid = sas.open("sashelp.company") -- open for input            
@@ -126,6 +135,17 @@ print('first statement after for loop')
 endinteractive;
 run;
 
+proc rlang;
+submit;
+fruits <- c("apple", "banana", "cherry")
+for (x in fruits) {
+   print(x)
+}
+
+print('first statement after for loop')
+endsubmit;
+run;
+
 proc lua;
 submit;