Troubleshoot unit test failure

NeonGeckoCom · Nov 6, 2024 · 274c92d · 274c92d
1 parent 8144bd1
commit 274c92d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_diana_utils.py b/tests/test_diana_utils.py
@@ -212,7 +212,7 @@ def test_generate_hana_config(self, confirm, prompt):
         self.assertEqual(hana_config['auth_requests_per_minute'], 6)
 
         # Test none confirmed
-        confirm.return_value = False
+        confirm.side_effect = [False, True]
         prompt.side_effect = [30, 3]
         hana_config = generate_hana_config()
         self.assertIsInstance(hana_config['hana'], dict)