Merge pull request #327 from PvdBerg1998/v0.6.xv0.6.x

Fix push_codeblock_safe

2 files changed, 10 insertions, 1 deletions

diff --git a/src/utils/message_builder.rs b/src/utils/message_builder.rs
index 9e90c53..be3c153 100644
--- a/src/utils/message_builder.rs
+++ b/src/utils/message_builder.rs
@@ -491,12 +491,13 @@ impl MessageBuilder {
             self.0.push_str(language);
         }
 
+        self.0.push('\n');
         {
             let mut c = content.into();
             c.inner = normalize(&c.inner).replace("```", " ");
             self.0.push_str(&c.to_string());
         }
-        self.0.push_str("```");
+        self.0.push_str("\n```");
 
         self
     }
diff --git a/tests/test_msg_builder.rs b/tests/test_msg_builder.rs
index c291bf5..eca8aef 100644
--- a/tests/test_msg_builder.rs
+++ b/tests/test_msg_builder.rs
@@ -16,6 +16,14 @@ fn code_blocks() {
 }
 
 #[test]
+fn safe_code_blocks() {
+    let content = MessageBuilder::new()
+        .push_codeblock_safe("test", Some("rb"))
+        .build();
+    assert_eq!(content, "```rb\ntest\n```");
+}
+
+#[test]
 fn safe_content() {
     let content = MessageBuilder::new()
         .push_safe("@everyone discord.gg/discord-api")