#2405 add e2e test for application install and uninstall, refactor input params

tests/e2e_integration_tests/test_uc2_data_manipulation_scenario.py

@@ -1,8 +1,13 @@
+import yaml
+
+from primaite.game.game import PrimaiteGame
+from primaite.session.environment import PrimaiteGymEnv
 from primaite.simulator.network.hardware.nodes.host.computer import Computer
 from primaite.simulator.network.hardware.nodes.host.server import Server
 from primaite.simulator.system.applications.database_client import DatabaseClient
 from primaite.simulator.system.applications.red_applications.data_manipulation_bot import DataManipulationBot
 from primaite.simulator.system.services.database.database_service import DatabaseService
+from tests import TEST_ASSETS_ROOT
 
 
 def test_data_manipulation(uc2_network):
@@ -32,3 +37,27 @@ def test_data_manipulation(uc2_network):
 
     # Now check that the DB client on the web_server can successfully query the users table on the database
     assert db_client.query("SELECT")
+
+
+def test_application_install_uninstall_on_uc2():
+    """Test Application install and uninstall via agent actions mid episode."""
+    with open(TEST_ASSETS_ROOT / "configs/test_application_install.yaml", "r") as f:
+        cfg = yaml.safe_load(f)
+
+    env = PrimaiteGymEnv(game_config=cfg)
+    env.agent.flatten_obs = False
+    env.reset()
+
+    _, _, _, _, _ = env.step(0)
+    domcon = env.game.simulation.network.get_node_by_hostname("domain_controller")
+
+    _, _, _, _, _ = env.step(78)
+    assert "DoSBot" in domcon.software_manager.software
+
+    _, _, _, _, _ = env.step(79)
+
+    assert "DoSBot" not in domcon.software_manager.software
+    assert "WebBrowser" in domcon.software_manager.software
+
+    _, _, _, _, _ = env.step(80)
+    assert "WebBrowser" not in domcon.software_manager.software