From 6aa6383fb73bdecae063f20f14f4d7204baeb5e8 Mon Sep 17 00:00:00 2001
From: Marek Wolan <marek.wolan@methods.co.uk>
Date: Wed, 31 Jan 2024 11:44:56 +0000
Subject: [PATCH] Fix broken test configs

---
 tests/assets/configs/bad_primaite_session.yaml        | 3 ++-
 tests/assets/configs/eval_only_primaite_session.yaml  | 3 ++-
 tests/assets/configs/multi_agent_session.yaml         | 4 +++-
 tests/assets/configs/test_primaite_session.yaml       | 3 ++-
 tests/assets/configs/train_only_primaite_session.yaml | 3 ++-
 5 files changed, 11 insertions(+), 5 deletions(-)

diff --git a/tests/assets/configs/bad_primaite_session.yaml b/tests/assets/configs/bad_primaite_session.yaml
index f5d7850c..552351b2 100644
--- a/tests/assets/configs/bad_primaite_session.yaml
+++ b/tests/assets/configs/bad_primaite_session.yaml
@@ -66,7 +66,7 @@ agents:
         nodes:
         - node_name: client_1
           applications:
-            - application_name: data_manipulation_bot
+            - application_name: DataManipulationBot
         max_folders_per_node: 1
         max_files_per_folder: 1
         max_services_per_node: 1
@@ -155,6 +155,7 @@ agents:
         - type: NODE_SERVICE_RESTART
         - type: NODE_SERVICE_DISABLE
         - type: NODE_SERVICE_ENABLE
+        - type: NODE_SERVICE_PATCH
         - type: NODE_FILE_SCAN
         - type: NODE_FILE_CHECKHASH
         - type: NODE_FILE_DELETE
diff --git a/tests/assets/configs/eval_only_primaite_session.yaml b/tests/assets/configs/eval_only_primaite_session.yaml
index b46c2f8d..d49562c8 100644
--- a/tests/assets/configs/eval_only_primaite_session.yaml
+++ b/tests/assets/configs/eval_only_primaite_session.yaml
@@ -71,7 +71,7 @@ agents:
         nodes:
         - node_name: client_1
           applications:
-            - application_name: data_manipulation_bot
+            - application_name: DataManipulationBot
         max_folders_per_node: 1
         max_files_per_folder: 1
         max_services_per_node: 1
@@ -159,6 +159,7 @@ agents:
         - type: NODE_SERVICE_RESTART
         - type: NODE_SERVICE_DISABLE
         - type: NODE_SERVICE_ENABLE
+        - type: NODE_SERVICE_PATCH
         - type: NODE_FILE_SCAN
         - type: NODE_FILE_CHECKHASH
         - type: NODE_FILE_DELETE
diff --git a/tests/assets/configs/multi_agent_session.yaml b/tests/assets/configs/multi_agent_session.yaml
index 23bd46c2..29f0ae7f 100644
--- a/tests/assets/configs/multi_agent_session.yaml
+++ b/tests/assets/configs/multi_agent_session.yaml
@@ -77,7 +77,7 @@ agents:
         nodes:
         - node_name: client_1
           applications:
-            - application_name: data_manipulation_bot
+            - application_name: DataManipulationBot
         max_folders_per_node: 1
         max_files_per_folder: 1
         max_services_per_node: 1
@@ -166,6 +166,7 @@ agents:
         - type: NODE_SERVICE_RESTART
         - type: NODE_SERVICE_DISABLE
         - type: NODE_SERVICE_ENABLE
+        - type: NODE_SERVICE_PATCH
         - type: NODE_FILE_SCAN
         - type: NODE_FILE_CHECKHASH
         - type: NODE_FILE_DELETE
@@ -609,6 +610,7 @@ agents:
         - type: NODE_SERVICE_RESTART
         - type: NODE_SERVICE_DISABLE
         - type: NODE_SERVICE_ENABLE
+        - type: NODE_SERVICE_PATCH
         - type: NODE_FILE_SCAN
         - type: NODE_FILE_CHECKHASH
         - type: NODE_FILE_DELETE
diff --git a/tests/assets/configs/test_primaite_session.yaml b/tests/assets/configs/test_primaite_session.yaml
index 1aa6ad71..0c70840d 100644
--- a/tests/assets/configs/test_primaite_session.yaml
+++ b/tests/assets/configs/test_primaite_session.yaml
@@ -75,7 +75,7 @@ agents:
         nodes:
         - node_name: client_1
           applications:
-            - application_name: data_manipulation_bot
+            - application_name: DataManipulationBot
         max_folders_per_node: 1
         max_files_per_folder: 1
         max_services_per_node: 1
@@ -166,6 +166,7 @@ agents:
         - type: NODE_SERVICE_RESTART
         - type: NODE_SERVICE_DISABLE
         - type: NODE_SERVICE_ENABLE
+        - type: NODE_SERVICE_PATCH
         - type: NODE_FILE_SCAN
         - type: NODE_FILE_CHECKHASH
         - type: NODE_FILE_DELETE
diff --git a/tests/assets/configs/train_only_primaite_session.yaml b/tests/assets/configs/train_only_primaite_session.yaml
index 115b6c85..0466a5ac 100644
--- a/tests/assets/configs/train_only_primaite_session.yaml
+++ b/tests/assets/configs/train_only_primaite_session.yaml
@@ -78,7 +78,7 @@ agents:
         nodes:
         - node_name: client_1
           applications:
-            - application_name: data_manipulation_bot
+            - application_name: DataManipulationBot
         max_folders_per_node: 1
         max_files_per_folder: 1
         max_services_per_node: 1
@@ -167,6 +167,7 @@ agents:
         - type: NODE_SERVICE_RESTART
         - type: NODE_SERVICE_DISABLE
         - type: NODE_SERVICE_ENABLE
+        - type: NODE_SERVICE_PATCH
         - type: NODE_FILE_SCAN
         - type: NODE_FILE_CHECKHASH
         - type: NODE_FILE_DELETE