Update unit tests to resolve test failures

NeonGeckoCom · Jun 25, 2024 · 30cc129 · 30cc129
1 parent e63c2b7
commit 30cc129
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/test/test_skill.py b/test/test_skill.py
@@ -74,12 +74,12 @@ def test_handle_exit_shutdown_intent(self):
         # Exit Confirmed
         message = Message("valid_intent", {"exit": "exit"})
 
-        def get_response(*args):
+        def get_response(*args, **kwargs):
             self.assertEqual(args[0], "ask_exit_shutdown")
             dialog = args[0]
-            dialog_data = args[1]
-            validator = args[2]
-            on_fail = args[3]
+            dialog_data = kwargs.get('data') or args[1]
+            validator = kwargs.get('validator') or args[2]
+            on_fail = kwargs.get('on_fail') or args[3]
 
             self.assertEqual(dialog, "ask_exit_shutdown")
             self.assertEqual(dialog_data["action"], "stop Neon")