Skip to content

Commit 1230675

Browse files
committed
add tests
1 parent 432e04d commit 1230675

1 file changed

Lines changed: 6 additions & 23 deletions

File tree

tests/code_utils/test_js_workflow_helpers.py

Lines changed: 6 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -131,30 +131,13 @@ def test_returns_empty_when_is_dependency(self) -> None:
131131

132132
assert result == ""
133133

134-
def test_npm_global_install(self) -> None:
135-
"""Should generate npm global install when not a dependency."""
136-
result = get_js_codeflash_install_step(JsPackageManager.NPM, is_dependency=False)
134+
def test_uv_tool_install_when_not_dependency(self) -> None:
135+
"""Should generate uv tool install when not a dependency, regardless of package manager."""
136+
for pkg_manager in (JsPackageManager.NPM, JsPackageManager.YARN, JsPackageManager.PNPM, JsPackageManager.BUN):
137+
result = get_js_codeflash_install_step(pkg_manager, is_dependency=False)
137138

138-
assert "Install Codeflash" in result
139-
assert "npm install -g codeflash" in result
140-
141-
def test_yarn_global_install(self) -> None:
142-
"""Should generate yarn global install when not a dependency."""
143-
result = get_js_codeflash_install_step(JsPackageManager.YARN, is_dependency=False)
144-
145-
assert "yarn global add codeflash" in result
146-
147-
def test_pnpm_global_install(self) -> None:
148-
"""Should generate pnpm global install when not a dependency."""
149-
result = get_js_codeflash_install_step(JsPackageManager.PNPM, is_dependency=False)
150-
151-
assert "pnpm add -g codeflash" in result
152-
153-
def test_bun_global_install(self) -> None:
154-
"""Should generate bun global install when not a dependency."""
155-
result = get_js_codeflash_install_step(JsPackageManager.BUN, is_dependency=False)
156-
157-
assert "bun add -g codeflash" in result
139+
assert "Install Codeflash" in result
140+
assert "uv tool install codeflash" in result
158141

159142

160143
class TestGetJsCodeflashRunCommand:

0 commit comments

Comments
 (0)