diff options
| -rw-r--r-- | src/utils/message_builder.rs | 3 | ||||
| -rw-r--r-- | tests/test_msg_builder.rs | 8 |
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") |