Skip to content

Commit

Permalink
test: create_rai_ws
Browse files Browse the repository at this point in the history
  • Loading branch information
maciejmajek committed Sep 3, 2024
1 parent 88a84c6 commit de28123
Showing 1 changed file with 27 additions and 0 deletions.
27 changes: 27 additions & 0 deletions tests/core/test_rai_cli.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,27 @@
import os
from pathlib import Path
from tempfile import TemporaryDirectory
from unittest.mock import MagicMock, patch

from rai.cli.rai_cli import create_rai_ws


def test_create_rai_ws():
with TemporaryDirectory() as directory:
# Mock ArgumentParser and its methods
mock_parser = MagicMock()
mock_args = MagicMock()
mock_args.name = "test_package"
mock_args.destination_directory = directory
mock_parser.parse_args.return_value = mock_args

# Patch argparse.ArgumentParser to return our mock
with patch("argparse.ArgumentParser", return_value=mock_parser):
create_rai_ws()

whoami_directory = Path(directory) / "test_package_whoami"

assert os.path.exists(whoami_directory), "Description folder is missing"

description_files = os.listdir(whoami_directory / "description")
assert "robot_constitution.txt" in description_files

0 comments on commit de28123

Please sign in to comment.