add tests for function class and function enum

rokucommunity · Jan 10, 2025 · 7e8ae0c · 7e8ae0c
1 parent c86ed13
commit 7e8ae0c
Showing 1 changed file with 32 additions and 0 deletions.
diff --git a/src/Formatter.spec.ts b/src/Formatter.spec.ts
@@ -935,6 +935,38 @@ end sub`;
             );
         });
 
+        it('correctly indents function class modifiers', () => {
+            expect(formatter.format(
+                'function class()\nend function'
+            )).to.equal(
+                'function class()\nend function'
+            );
+        });
+
+        it('correctly indents function class modifiers with print statement', () => {
+            expect(formatter.format(
+                'function class()\nprint "hello"\nend function'
+            )).to.equal(
+                'function class()\n    print "hello"\nend function'
+            );
+        });
+
+        it('correctly indents function enum modifiers', () => {
+            expect(formatter.format(
+                'function enum()\nend function'
+            )).to.equal(
+                'function enum()\nend function'
+            );
+        });
+
+        it('correctly indents function enum modifiers with print statement', () => {
+            expect(formatter.format(
+                'function enum()\nprint "hello"\nend function'
+            )).to.equal(
+                'function enum()\n    print "hello"\nend function'
+            );
+        });
+
         it('trims empty lines', () => {
             expect(formatter.format(`sub a()\n    \nend sub`)).to.equal(`sub a()\n\nend sub`);
         });