From 1bf9e023e4f682aa825f1f0a2d321aebe2f1637b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=88=98=E6=A3=92=E6=A3=92?= <liubangbang@fuzhi.ai>
Date: Mon, 22 Jan 2024 18:53:40 +0800
Subject: [PATCH] add new test.

---
 tests/metagpt/actions/test_execute_code.py | 40 +++++++++++++++++++---
 1 file changed, 36 insertions(+), 4 deletions(-)

diff --git a/tests/metagpt/actions/test_execute_code.py b/tests/metagpt/actions/test_execute_code.py
index ecddccf6f..21627e6f9 100644
--- a/tests/metagpt/actions/test_execute_code.py
+++ b/tests/metagpt/actions/test_execute_code.py
@@ -1,7 +1,6 @@
 import pytest
 
 from metagpt.actions.execute_code import ExecutePyCode, truncate
-from metagpt.schema import Message
 
 
 @pytest.mark.asyncio
@@ -11,9 +10,6 @@ async def test_code_running():
     assert output[1] is True
     output = await pi.run({"code": "print('hello world!')", "language": "python"})
     assert output[1] is True
-    code_msg = Message("print('hello world!')")
-    output = await pi.run(code_msg)
-    assert output[1] is True
 
 
 @pytest.mark.asyncio
@@ -67,6 +63,15 @@ def test_truncate():
     output, is_success = truncate("hello world", 5, False)
     assert "Truncated to show only last 5 characters\nworld" in output
     assert not is_success
+    # 异步
+    output, is_success = truncate("<coroutine object", 5, True)
+    assert not is_success
+    assert "await" in output
+    # 重复的desc
+    result = "Executed code successfully. Truncated to show only first 5 characters\nhello"
+    output, is_success = truncate(result, 5, True)
+    assert is_success
+    assert output == result
 
 
 @pytest.mark.asyncio
@@ -87,3 +92,30 @@ async def test_run_code_text():
     message, success = await pi.run(code="# This is a code!", language="markdown")
     assert success
     assert message == "# This is a code!"
+    mix_text = "# Title!\n ```python\n print('This is a code!')```"
+    message, success = await pi.run(code=mix_text, language="markdown")
+    assert success
+    assert message == mix_text
+
+
+@pytest.mark.asyncio
+async def test_terminate():
+    pi = ExecutePyCode()
+    await pi.run(code='print("This is a code!")', language="python")
+    is_kernel_alive = await pi.nb_client.km.is_alive()
+    assert is_kernel_alive
+    await pi.terminate()
+    import time
+
+    time.sleep(2)
+    assert pi.nb_client.km is None
+
+
+@pytest.mark.asyncio
+async def test_reset():
+    pi = ExecutePyCode()
+    await pi.run(code='print("This is a code!")', language="python")
+    is_kernel_alive = await pi.nb_client.km.is_alive()
+    assert is_kernel_alive
+    await pi.reset()
+    assert pi.nb_client.km is None