Fix positional info of cr+lf in mdx

1 files changed, 88 insertions, 0 deletions

diff --git a/src/util/location.rs b/src/util/location.rs
index 0c9c426..deff307 100644
--- a/src/util/location.rs
+++ b/src/util/location.rs
@@ -32,6 +32,7 @@ impl Location {
             if bytes[index] == b'\r' {
                 if index + 1 < bytes.len() && bytes[index + 1] == b'\n' {
                     location_index.indices.push(index + 2);
+                    index += 1;
                 } else {
                     location_index.indices.push(index + 1);
                 }
@@ -109,3 +110,90 @@ impl Location {
         }
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_location_lf() {
+        let location = Location::new("ab\nc".as_bytes());
+        assert_eq!(
+            location.to_point(0), // `a`
+            Some(Point::new(1, 1, 0)),
+            "should support some points (1)"
+        );
+        assert_eq!(
+            location.to_point(1), // `b`
+            Some(Point::new(1, 2, 1)),
+            "should support some points (2)"
+        );
+        assert_eq!(
+            location.to_point(2), // `\n`
+            Some(Point::new(1, 3, 2)),
+            "should support some points (3)"
+        );
+        assert_eq!(
+            location.to_point(3), // `c`
+            Some(Point::new(2, 1, 3)),
+            "should support some points (4)"
+        );
+        assert_eq!(
+            location.to_point(4), // EOF
+            // Still gets a point, so that we can represent positions of things
+            // that end at the last character, the `c` end at `2:2`.
+            Some(Point::new(2, 2, 4)),
+            "should support some points (5)"
+        );
+        assert_eq!(
+            location.to_point(5), // Out of bounds
+            None,
+            "should support some points (6)"
+        );
+    }
+
+    #[test]
+    fn test_location_cr() {
+        let location = Location::new("a\rb".as_bytes());
+        assert_eq!(
+            location.to_point(0), // `a`
+            Some(Point::new(1, 1, 0)),
+            "should support some points (1)"
+        );
+        assert_eq!(
+            location.to_point(1), // `\r`
+            Some(Point::new(1, 2, 1)),
+            "should support some points (2)"
+        );
+        assert_eq!(
+            location.to_point(2), // `b`
+            Some(Point::new(2, 1, 2)),
+            "should support some points (3)"
+        );
+    }
+
+    #[test]
+    fn test_location_cr_lf() {
+        let location = Location::new("a\r\nb".as_bytes());
+        assert_eq!(
+            location.to_point(0), // `a`
+            Some(Point::new(1, 1, 0)),
+            "should support some points (1)"
+        );
+        assert_eq!(
+            location.to_point(1), // `\r`
+            Some(Point::new(1, 2, 1)),
+            "should support some points (2)"
+        );
+        assert_eq!(
+            location.to_point(2), // `\n`
+            Some(Point::new(1, 3, 2)),
+            "should support some points (3)"
+        );
+        assert_eq!(
+            location.to_point(3), // `b`
+            Some(Point::new(2, 1, 3)),
+            "should support some points (4)"
+        );
+    }
+}