2020def test_create_agent_with_options ():
2121 runner = CliRunner ()
2222 with runner .isolated_filesystem () as temp_dir :
23- result = runner .invoke (create , ["test-agent " , "--ark-api-key" , "test-key" ])
23+ result = runner .invoke (create , ["test_agent " , "--ark-api-key" , "test-key" ])
2424 assert result .exit_code == 0
2525
26- agent_folder = Path (temp_dir ) / "test-agent "
26+ agent_folder = Path (temp_dir ) / "test_agent "
2727 assert agent_folder .exists ()
2828
2929 config_path = agent_folder / ".env"
@@ -42,31 +42,31 @@ def test_create_agent_overwrite_existing_directory():
4242 runner = CliRunner ()
4343 with runner .isolated_filesystem () as temp_dir :
4444 # First, create the agent
45- runner .invoke (create , ["test-agent " , "--ark-api-key" , "test-key" ])
45+ runner .invoke (create , ["test_agent " , "--ark-api-key" , "test-key" ])
4646
4747 # Attempt to create it again, but cancel the overwrite
4848 result = runner .invoke (
4949 create ,
50- ["test-agent " , "--ark-api-key" , "test-key" ],
50+ ["test_agent " , "--ark-api-key" , "test-key" ],
5151 input = "n\n " ,
5252 )
5353 assert "Operation cancelled" in result .output
5454
5555 # Attempt to create it again, and confirm the overwrite
5656 result = runner .invoke (
5757 create ,
58- ["test-agent " , "--ark-api-key" , "new-key" ],
58+ ["test_agent " , "--ark-api-key" , "new-key" ],
5959 input = "y\n " ,
6060 )
6161 assert result .exit_code == 0
62- agent_folder = Path (temp_dir ) / "test-agent "
62+ agent_folder = Path (temp_dir ) / "test_agent "
6363 config_path = agent_folder / ".env"
6464 config_content = config_path .read_text ()
6565 assert "MODEL_AGENT_API_KEY=new-key" in config_content
6666
6767
6868def test_generate_files (tmp_path : Path ):
69- agent_name = "test-agent "
69+ agent_name = "test_agent "
7070 api_key = "test-key"
7171 target_dir = tmp_path / agent_name
7272
@@ -87,26 +87,26 @@ def test_generate_files(tmp_path: Path):
8787def test_prompt_for_ark_api_key_enter_now ():
8888 runner = CliRunner ()
8989 with runner .isolated_filesystem ():
90- result = runner .invoke (create , input = "test-agent \n 1\n my-secret-key\n " )
90+ result = runner .invoke (create , input = "test_agent \n 1\n my-secret-key\n " )
9191 assert result .exit_code == 0
92- assert "my-secret-key" in (Path ("test-agent " ) / ".env" ).read_text ()
92+ assert "my-secret-key" in (Path ("test_agent " ) / ".env" ).read_text ()
9393
9494
9595def test_prompt_for_ark_api_key_configure_later ():
9696 runner = CliRunner ()
9797 with runner .isolated_filesystem ():
98- result = runner .invoke (create , input = "test-agent \n 2\n " )
98+ result = runner .invoke (create , input = "test_agent \n 2\n " )
9999 assert result .exit_code == 0
100- assert "MODEL_AGENT_API_KEY=" in (Path ("test-agent " ) / ".env" ).read_text ()
100+ assert "MODEL_AGENT_API_KEY=" in (Path ("test_agent " ) / ".env" ).read_text ()
101101
102102
103103def test_create_agent_with_prompts ():
104104 runner = CliRunner ()
105105 with runner .isolated_filesystem () as temp_dir :
106- result = runner .invoke (create , input = "test-agent \n 1\n test-key\n " )
106+ result = runner .invoke (create , input = "test_agent \n 1\n test-key\n " )
107107 assert result .exit_code == 0
108108
109- agent_folder = Path (temp_dir ) / "test-agent "
109+ agent_folder = Path (temp_dir ) / "test_agent "
110110 assert agent_folder .exists ()
111111
112112 config_path = agent_folder / ".env"
0 commit comments