indentation: Account for trailing comments

1 files changed, 28 insertions, 0 deletions

diff --git a/snakemake-test.el b/snakemake-test.el
index 80850d7..f70b284 100644
--- a/snakemake-test.el
+++ b/snakemake-test.el
@@ -394,6 +394,19 @@ rule abc:
    (string=
     "
 rule abc:
+    output:  # comment
+        "
+    (snakemake-with-temp-text
+        "
+rule abc:
+    output:  # comment
+              <point>"
+      (snakemake-indent-line)
+      (buffer-string))))
+  (should
+   (string=
+    "
+rule abc:
     output:
         'file{}{}'.format('one',
                           'two'"
@@ -502,6 +515,21 @@ rule abc:
    (string=
     "
 rule abc:
+    output:  # comment
+        'file'
+        'text'"
+    (snakemake-with-temp-text
+        "
+rule abc:
+    output:  # comment
+        'file'
+<point>            'text'"
+      (snakemake-indent-line)
+      (buffer-string))))
+  (should
+   (string=
+    "
+rule abc:
     run:
         with this:
             "