github-ops
86 lines
2.8 KiB
Python
86 lines
2.8 KiB
Python
import os
|
|
import subprocess
|
|
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
|
|
ROOT = Path(__file__).resolve().parents[2]
|
|
CHECKER = ROOT / "scripts" / "check-rust-code-health.sh"
|
|
|
|
|
|
class RustCodeHealthCheckTests(unittest.TestCase):
|
|
def run_checker(self, root: Path, max_lines: int = 20) -> subprocess.CompletedProcess[str]:
|
|
env = os.environ.copy()
|
|
env["CRANK_RUST_HEALTH_ROOT"] = str(root)
|
|
env["CRANK_RUST_HEALTH_DEFAULT_MAX_LINES"] = str(max_lines)
|
|
return subprocess.run(
|
|
[str(CHECKER)],
|
|
check=False,
|
|
text=True,
|
|
env=env,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE,
|
|
)
|
|
|
|
def prepare_root(self, root: Path) -> None:
|
|
(root / "apps" / "demo" / "src").mkdir(parents=True)
|
|
(root / "crates" / "demo" / "src").mkdir(parents=True)
|
|
|
|
def test_passes_for_small_rust_files(self) -> None:
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
root = Path(tmp)
|
|
self.prepare_root(root)
|
|
(root / "crates" / "demo" / "src" / "lib.rs").write_text(
|
|
"pub fn ok() -> bool {\n true\n}\n",
|
|
encoding="utf-8",
|
|
)
|
|
|
|
result = self.run_checker(root)
|
|
|
|
self.assertEqual(result.returncode, 0, result.stderr)
|
|
self.assertIn("checking file size limits", result.stdout)
|
|
|
|
def test_fails_for_new_large_rust_file(self) -> None:
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
root = Path(tmp)
|
|
self.prepare_root(root)
|
|
(root / "apps" / "demo" / "src" / "large.rs").write_text(
|
|
"\n".join("// line" for _ in range(21)) + "\n",
|
|
encoding="utf-8",
|
|
)
|
|
|
|
result = self.run_checker(root)
|
|
|
|
self.assertNotEqual(result.returncode, 0)
|
|
self.assertIn("apps/demo/src/large.rs has 21 lines", result.stderr)
|
|
|
|
def test_fails_for_large_inline_test_module(self) -> None:
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
root = Path(tmp)
|
|
self.prepare_root(root)
|
|
source = "\n".join(
|
|
[
|
|
"pub fn ok() -> bool { true }",
|
|
"#[cfg(test)]",
|
|
"mod tests {",
|
|
" #[test]",
|
|
" fn works() { assert!(super::ok()); }",
|
|
*[" // test debt" for _ in range(8)],
|
|
"}",
|
|
]
|
|
)
|
|
(root / "crates" / "demo" / "src" / "lib.rs").write_text(
|
|
source + "\n",
|
|
encoding="utf-8",
|
|
)
|
|
|
|
result = self.run_checker(root)
|
|
|
|
self.assertNotEqual(result.returncode, 0)
|
|
self.assertIn("contains inline tests", result.stderr)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|