61 lines
1.8 KiB
Python
61 lines
1.8 KiB
Python
 
|
from nanobot.utils.helpers import split_message
|
||
|
|
|
||
|
|
|
||
|
|
def test_split_message_no_code_blocks_unchanged():
|
||
|
|
content = "alpha beta gamma delta"
|
||
|
|
|
||
|
|
assert split_message(content, max_len=12) == ["alpha beta", "gamma delta"]
|
||
|
|
|
||
|
|
|
||
|
|
def test_split_message_outside_code_block_unchanged():
|
||
|
|
content = "alpha beta gamma delta\n```python\nx = 1\n```\ndone"
|
||
|
|
|
||
|
|
chunks = split_message(content, max_len=12)
|
||
|
|
|
||
|
|
assert chunks[0] == "alpha beta"
|
||
|
|
assert chunks[1].startswith("gamma")
|
||
|
|
|
||
|
|
|
||
|
|
def test_split_message_inside_code_block_moves_before_fence():
|
||
|
|
content = "Intro paragraph.\n```python\nprint('a')\nprint('b')\n```\nDone"
|
||
|
|
|
||
|
|
chunks = split_message(content, max_len=35)
|
||
|
|
|
||
|
|
assert chunks[0] == "Intro paragraph.\n"
|
||
|
|
assert chunks[1].startswith("```python\nprint('a')")
|
||
|
|
assert all(chunk.count("```") % 2 == 0 for chunk in chunks[1:])
|
||
|
|
|
||
|
|
|
||
|
|
def test_split_message_code_block_longer_than_max_len_closes_and_reopens():
|
||
|
|
content = "```python\n" + ("print('line one')\n" * 6) + "```\nDone"
|
||
|
|
|
||
|
|
chunks = split_message(content, max_len=60)
|
||
|
|
|
||
|
|
assert len(chunks) > 1
|
||
|
|
assert all(len(chunk) <= 60 for chunk in chunks)
|
||
|
|
assert all(chunk.count("```") % 2 == 0 for chunk in chunks)
|
||
|
|
assert chunks[0].startswith("```python\n")
|
||
|
|
assert chunks[0].endswith("\n```")
|
||
|
|
assert chunks[1].startswith("```python\n")
|
||
|
|
|
||
|
|
|
||
|
|
def test_split_message_multiple_code_blocks_moves_second_block_to_next_chunk():
|
||
|
|
content = (
|
||
|
|
"First\n"
|
||
|
|
"```js\n"
|
||
|
|
"one();\n"
|
||
|
|
"```\n"
|
||
|
|
"Middle paragraph here\n"
|
||
|
|
"```py\n"
|
||
|
|
"two()\n"
|
||
|
|
"three()\n"
|
||
|
|
"```\n"
|
||
|
|
"End"
|
||
|
|
)
|
||
|
|
|
||
|
|
chunks = split_message(content, max_len=55)
|
||
|
|
|
||
|
|
assert chunks[0].endswith("Middle paragraph here\n")
|
||
|
|
assert chunks[1].startswith("```py\n")
|
||
|
|
assert all(chunk.count("```") % 2 == 0 for chunk in chunks)
|