def test_search_connections_negative(*mocks): """Test list localConnections negative response.""" app = create_app() response = app.get("api/connection/query") assert response.status_code == 400 result = json.loads(response.get_data(as_text=True)) expected_result = "Failed to search items." assert result["detail"] == expected_result
def test_js(): """Test that the home-page exits.""" app = create_app() # sends HTTP GET request to the application # on the specified path result = app.get("/static/js/home.js") # assert the status code of the response assert result.status_code == 200
def test_list_connections(*mocks): """Test list localConnections.""" app = create_app() response = app.get("api/agent/agent_name/connection") assert response.status_code == 200 result = json.loads(response.get_data(as_text=True)) expected_result = [{"name": "some-connection"}] assert result == expected_result
def test_create_agent_fail(*mocks): """Test creating an agent and failing.""" app = create_app() agent_name = "test_agent_id" response_create = app.post("api/agent", content_type="application/json", data=json.dumps(agent_name)) assert response_create.status_code == 400 data = json.loads(response_create.get_data(as_text=True)) assert data["detail"] == "Failed to create Agent. Message"
def test_home_page_exits(): """Test that the home-page exits.""" app = create_app() # sends HTTP GET request to the application # on the specified path result = app.get("/") # assert the status code of the response assert result.status_code == 200 assert "AEA GUI" in str(result.data)
def test_create_agent(*mocks): """Test creating an agent.""" app = create_app() agent_name = "test_agent_id" # Ensure there is now one agent response_create = app.post("api/agent", content_type="application/json", data=json.dumps(agent_name)) assert response_create.status_code == 201 data = json.loads(response_create.get_data(as_text=True)) assert data == agent_name
def test_delete_agent(*mocks): """Test creating an agent.""" app = create_app() agent_name = "test_agent_id" # Ensure there is now one agent response_delete = app.delete("api/agent/" + agent_name, data=None, content_type="application/json") assert response_delete.status_code == 200 data = json.loads(response_delete.get_data(as_text=True)) assert data == "Agent {} deleted".format(agent_name)
def test_delete_agent_fail(*mocks): """Test creating an agent and failing.""" app = create_app() agent_name = "test_agent_id" response_delete = app.delete("api/agent/" + agent_name, data=None, content_type="application/json") assert response_delete.status_code == 400 data = json.loads(response_delete.get_data(as_text=True)) assert data[ "detail"] == "Failed to delete Agent {} - it may not exist".format( agent_name)
def test_fetch_agent(*mocks): """Test fetch an agent.""" app = create_app() agent_name = "test_agent_name" agent_id = "author/{}:0.1.0".format(agent_name) # Ensure there is now one agent resp = app.post( "api/fetch-agent", content_type="application/json", data=json.dumps(agent_id), ) assert resp.status_code == 201 data = json.loads(resp.get_data(as_text=True)) assert data == agent_name
def test_search_connections(*mocks): """Test list localConnections.""" app = create_app() response = app.get("api/connection/query") assert response.status_code == 200 result = json.loads(response.get_data(as_text=True)) expected_result = { "item_type": "connection", "search_result": [], "search_term": "query", } assert result == expected_result
def test_fetch_agent_fail(*mocks): """Test fetch agent fail.""" app = create_app() agent_name = "test_agent_name" agent_id = "author/{}:0.1.0".format(agent_name) resp = app.post( "api/fetch-agent", content_type="application/json", data=json.dumps(agent_id), ) assert resp.status_code == 400 data = json.loads(resp.get_data(as_text=True)) assert data["detail"] == "Failed to fetch an agent {}. {}".format( agent_id, "Message" )
def test_remove_item(*mocks): """Test remove a skill/connection/protocol. Actually we just do connection as code coverage is the same. """ app = create_app() agent_name = "test_agent_id" connection_name = "fetchai/test_connection:0.1.0" response_remove = app.post( "api/agent/" + agent_name + "/connection/remove", content_type="application/json", data=json.dumps(connection_name), ) assert response_remove.status_code == 201 data = json.loads(response_remove.get_data(as_text=True)) assert data == agent_name
def test_real_create_local(*mocks): """Really create an agent (have to test the call_aea at some point).""" # Set up a temporary current working directory to make agents in with TempCWD() as temp_cwd: app = create_app() # copy the 'packages' directory in the parent of the agent folder. shutil.copytree(Path(CUR_PATH, "..", "packages"), Path(temp_cwd.temp_dir, "packages")) agent_id = "test_agent_id" # Make an agent # We do it programmatically as we need to create an agent with default author # that was prevented from GUI. ctx = Context(cwd=temp_cwd.temp_dir) create_aea(ctx, agent_id, local=True, author=DEFAULT_AUTHOR) # Give it a bit of time so the polling funcionts get called time.sleep(1) # Check that we can actually see this agent too response_agents = app.get( "api/agent", data=None, content_type="application/json", ) data = json.loads(response_agents.get_data(as_text=True)) assert response_agents.status_code == 200 assert len(data) == 1 assert data[0]["public_id"] == agent_id assert data[0]["description"] == "placeholder description" # do same but this time find that this is not an agent directory. with patch("os.path.isdir", return_value=False): response_agents = app.get( "api/agent", data=None, content_type="application/json", ) data = json.loads(response_agents.get_data(as_text=True)) assert response_agents.status_code == 200 assert len(data) == 0
def test_remove_item_fail(*mocks): """Test remove a skill/connection/protocol when it fails. Actually we just do connection as code coverage is the same. """ app = create_app() agent_name = "test_agent_id" connection_name = "fetchai/test_connection:0.1.0" response_remove = app.post( "api/agent/" + agent_name + "/connection/remove", content_type="application/json", data=json.dumps(connection_name), ) assert response_remove.status_code == 400 data = json.loads(response_remove.get_data(as_text=True)) assert data[ "detail"] == "Failed to remove connection {} from agent {}".format( connection_name, agent_name)
def test_create_and_run_agent(): """Test for running and agent, reading TTY and errors.""" # Set up a temporary current working directory in which to make agents with TempCWD() as temp_cwd: app = create_app() # copy the 'packages' directory in the parent of the agent folder. shutil.copytree(Path(CUR_PATH, "..", "packages"), Path(temp_cwd.temp_dir, "packages")) agent_id = "test_agent" # Make an agent # We do it programmatically as we need to create an agent with default author # that was prevented from GUI. ctx = Context(cwd=temp_cwd.temp_dir) ctx.set_config("is_local", True) create_aea(ctx, agent_id, local=True, author=DEFAULT_AUTHOR) # Add the local connection with patch("aea.cli_gui.app_context.local", True): response_add = app.post( "api/agent/" + agent_id + "/connection", content_type="application/json", data=json.dumps(str(LOCAL_PUBLIC_ID)), ) assert response_add.status_code == 201 # Get the running status before we have run it response_status = app.get( "api/agent/" + agent_id + "/run", data=None, content_type="application/json", ) assert response_status.status_code == 200 data = json.loads(response_status.get_data(as_text=True)) assert "NOT_STARTED" in data["status"] # run the agent with a non existent connection response_run = app.post( "api/agent/" + agent_id + "/run", content_type="application/json", data=json.dumps("author/non-existent-connection:0.1.0"), ) assert response_run.status_code == 400 # run the agent with default connection - should be something in the error output? response_run = app.post( "api/agent/" + agent_id + "/run", content_type="application/json", data=json.dumps(""), ) assert response_run.status_code == 201 time.sleep(2) # Stop the agent running response_stop = app.delete( "api/agent/" + agent_id + "/run", data=None, content_type="application/json", ) assert response_stop.status_code == 200 time.sleep(2) # run the agent with stub connection response_run = app.post( "api/agent/" + agent_id + "/run", content_type="application/json", data=json.dumps(str(STUB_CONNECTION_PUBLIC_ID)), ) assert response_run.status_code == 201 time.sleep(2) # Try running it again (this should fail) response_run = app.post( "api/agent/" + agent_id + "/run", content_type="application/json", data=json.dumps(str(STUB_CONNECTION_PUBLIC_ID)), ) assert response_run.status_code == 400 # Get the running status response_status = app.get( "api/agent/" + agent_id + "/run", data=None, content_type="application/json", ) assert response_status.status_code == 200 data = json.loads(response_status.get_data(as_text=True)) assert data["error"] == "" assert "RUNNING" in data["status"] app.delete( "api/agent/" + agent_id + "/run", data=None, content_type="application/json", ) time.sleep(1) # Get the running status response_status = app.get( "api/agent/" + agent_id + "/run", data=None, content_type="application/json", ) assert response_status.status_code == 200 data = json.loads(response_status.get_data(as_text=True)) assert "process terminate" in data["error"] assert "NOT_STARTED" in data["status"] # run the agent again (takes a different path through code) response_run = app.post( "api/agent/" + agent_id + "/run", content_type="application/json", data=json.dumps(str(STUB_CONNECTION_PUBLIC_ID)), ) assert response_run.status_code == 201 time.sleep(2) # Get the running status response_status = app.get( "api/agent/" + agent_id + "/run", data=None, content_type="application/json", ) assert response_status.status_code == 200 data = json.loads(response_status.get_data(as_text=True)) assert data["error"] == "" assert "RUNNING" in data["status"] # Stop the agent running response_stop = app.delete( "api/agent/" + agent_id + "/run", data=None, content_type="application/json", ) assert response_stop.status_code == 200 time.sleep(2) # Get the running status response_status = app.get( "api/agent/" + agent_id + "/run", data=None, content_type="application/json", ) assert response_status.status_code == 200 data = json.loads(response_status.get_data(as_text=True)) assert "process terminate" in data["error"] assert "NOT_STARTED" in data["status"] # Stop a none existent agent running response_stop = app.delete( "api/agent/" + agent_id + "_NOT" + "/run", data=None, content_type="application/json", ) assert response_stop.status_code == 400 time.sleep(2) genuine_func = aea.cli_gui.call_aea_async def _dummycall_aea_async(param_list, dir_arg): assert param_list[0] == sys.executable assert param_list[1] == "-m" assert param_list[2] == "aea.cli" if param_list[3] == "run": return None else: return genuine_func(param_list, dir_arg) # Run when process files (but other call - such as status should not fail) with patch("aea.cli_gui.call_aea_async", _dummycall_aea_async): response_run = app.post( "api/agent/" + agent_id + "/run", content_type="application/json", data=json.dumps(str(STUB_CONNECTION_PUBLIC_ID)), ) assert response_run.status_code == 400
def setUp(self): """Set up test case.""" self.app = create_app()