From 34dc13bb98ab1ffb431fd746faebc5efa0356dd0 Mon Sep 17 00:00:00 2001
From: liuminhui <huidbk@163.com>
Date: Fri, 17 Jan 2025 17:00:51 +0800
Subject: [PATCH] tools linter ut

---
 tests/metagpt/tools/libs/test_linter.py | 56 +++++++++++++++++++++++++
 1 file changed, 56 insertions(+)
 create mode 100644 tests/metagpt/tools/libs/test_linter.py

diff --git a/tests/metagpt/tools/libs/test_linter.py b/tests/metagpt/tools/libs/test_linter.py
new file mode 100644
index 000000000..4443e7c04
--- /dev/null
+++ b/tests/metagpt/tools/libs/test_linter.py
@@ -0,0 +1,56 @@
+import tempfile
+
+from metagpt.tools.libs.linter import Linter, LintResult
+
+
+def test_linter_initialization():
+    linter = Linter(encoding="utf-8", root="/test/root")
+    assert linter.encoding == "utf-8"
+    assert linter.root == "/test/root"
+    assert "python" in linter.languages
+    assert callable(linter.languages["python"])
+
+
+def test_get_abs_fname():
+    linter = Linter(root="/test/root")
+    abs_path = linter.get_abs_fname("test_file.py")
+    assert abs_path == linter.get_rel_fname("test_file.py")
+
+
+def test_py_lint():
+    linter = Linter()
+    code = "print('Hello, World!')"
+    result = linter.py_lint("test_linter.py", "test_linter.py", code)
+    assert result is None  # No errors expected for valid Python code
+
+
+def test_lint_with_python_file():
+    linter = Linter()
+    with tempfile.NamedTemporaryFile(suffix=".py", delete=True) as temp_file:
+        temp_file.write(b"def hello():\nprint('Hello')\n")  # IndentationError
+        temp_file.flush()
+        result = linter.lint(temp_file.name)
+        assert isinstance(result, LintResult)
+        assert "IndentationError" in result.text
+        assert len(result.lines) > 0
+
+
+def test_lint_with_unsupported_language():
+    linter = Linter()
+    with tempfile.NamedTemporaryFile(suffix=".unsupported", delete=True) as temp_file:
+        temp_file.write(b"This is unsupported code.")
+        temp_file.flush()
+
+        result = linter.lint(temp_file.name)
+        assert result is None  # Unsupported language should return None
+
+
+def test_run_cmd():
+    linter = Linter()
+    with tempfile.NamedTemporaryFile(suffix=".py", delete=True) as temp_file:
+        temp_file.write(b"print('Hello, World!')\n")
+        temp_file.flush()
+
+        result = linter.run_cmd("flake8", temp_file.name, "print('Hello, World!')")
+        # Since flake8 might not be installed in the test environment, we just ensure no exception is raised
+        assert result is None or isinstance(result, LintResult)