1 files changed, 61 insertions, 0 deletions
diff --git a/compiler/rustc_ast/src/util/comments/tests.rs b/compiler/rustc_ast/src/util/comments/tests.rs
new file mode 100644
index 000000000..11d50603a
--- /dev/null
+++ b/compiler/rustc_ast/src/util/comments/tests.rs
@@ -0,0 +1,61 @@
+use super::*;
+use rustc_span::create_default_session_globals_then;
+
+#[test]
+fn test_block_doc_comment_1() {
+    create_default_session_globals_then(|| {
+        let comment = "\n * Test \n **  Test\n *   Test\n";
+        let stripped = beautify_doc_string(Symbol::intern(comment), CommentKind::Block);
+        assert_eq!(stripped.as_str(), " Test \n*  Test\n   Test");
+    })
+}
+
+#[test]
+fn test_block_doc_comment_2() {
+    create_default_session_globals_then(|| {
+        let comment = "\n * Test\n *  Test\n";
+        let stripped = beautify_doc_string(Symbol::intern(comment), CommentKind::Block);
+        assert_eq!(stripped.as_str(), " Test\n  Test");
+    })
+}
+
+#[test]
+fn test_block_doc_comment_3() {
+    create_default_session_globals_then(|| {
+        let comment = "\n let a: *i32;\n *a = 5;\n";
+        let stripped = beautify_doc_string(Symbol::intern(comment), CommentKind::Block);
+        assert_eq!(stripped.as_str(), "let a: *i32;\n*a = 5;");
+    })
+}
+
+#[test]
+fn test_line_doc_comment() {
+    create_default_session_globals_then(|| {
+        let stripped = beautify_doc_string(Symbol::intern(" test"), CommentKind::Line);
+        assert_eq!(stripped.as_str(), " test");
+        let stripped = beautify_doc_string(Symbol::intern("! test"), CommentKind::Line);
+        assert_eq!(stripped.as_str(), "! test");
+        let stripped = beautify_doc_string(Symbol::intern("test"), CommentKind::Line);
+        assert_eq!(stripped.as_str(), "test");
+        let stripped = beautify_doc_string(Symbol::intern("!test"), CommentKind::Line);
+        assert_eq!(stripped.as_str(), "!test");
+    })
+}
+
+#[test]
+fn test_doc_blocks() {
+    create_default_session_globals_then(|| {
+        let stripped =
+            beautify_doc_string(Symbol::intern(" # Returns\n     *\n     "), CommentKind::Block);
+        assert_eq!(stripped.as_str(), " # Returns\n\n");
+
+        let stripped = beautify_doc_string(
+            Symbol::intern("\n     * # Returns\n     *\n     "),
+            CommentKind::Block,
+        );
+        assert_eq!(stripped.as_str(), " # Returns\n\n");
+
+        let stripped = beautify_doc_string(Symbol::intern("\n *     a\n "), CommentKind::Block);
+        assert_eq!(stripped.as_str(), "     a\n");
+    })
+}