39 lines
1022 B
Python
39 lines
1022 B
Python
|
import pytest
|
||
|
|
from typer.testing import CliRunner
|
||
|
|
from guardrails.cli.hub import hub_command
|
||
|
|
import os
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def mock_post_validator_submit(mocker):
|
||
|
|
return mocker.patch(
|
||
|
|
"guardrails.cli.hub.submit.post_validator_submit",
|
||
|
|
new=lambda *args, **kwargs: None,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def test_submit_command_success(mock_post_validator_submit):
|
||
|
|
runner = CliRunner()
|
||
|
|
package_name = "test_validator"
|
||
|
|
|
||
|
|
with runner.isolated_filesystem():
|
||
|
|
package_path = f"{package_name}.py"
|
||
|
|
|
||
|
|
# Create a temporary file with the package name
|
||
|
|
with open(package_path, "w+") as file:
|
||
|
|
file.write("# Test validator content")
|
||
|
|
|
||
|
|
result = runner.invoke(hub_command, ["submit", package_name])
|
||
|
|
|
||
|
|
assert result.exit_code == 0
|
||
|
|
|
||
|
|
os.remove(package_path)
|
||
|
|
|
||
|
|
|
||
|
|
def test_submit_command_failure(mock_post_validator_submit):
|
||
|
|
runner = CliRunner()
|
||
|
|
package_name = "test_validator"
|
||
|
|
|
||
|
|
result = runner.invoke(hub_command, ["submit", package_name])
|
||
|
|
assert result.exit_code != 0
|