Explorar o código

Integration test: exit if no prompt match (#1772)

Boxuan Li hai 1 ano
pai
achega
b84f25ab35
Modificáronse 1 ficheiros con 5 adicións e 3 borrados
  1. 5 3
      tests/integration/conftest.py

+ 5 - 3
tests/integration/conftest.py

@@ -1,6 +1,7 @@
 import io
 import os
 import re
+import sys
 from functools import partial
 
 import pytest
@@ -79,9 +80,10 @@ def mock_completion(*args, test_name, **kwargs):
     for message in messages:
         message_str += message['content']
     mock_response = get_mock_response(test_name, message_str)
-    assert mock_response is not None, (
-        'Mock response for prompt is not found:\n\n' + message_str
-    )
+    if mock_response is None:
+        print('Mock response for prompt is not found:\n\n' + message_str)
+        print('Exiting...')
+        sys.exit(1)
     response = completion(**kwargs, mock_response=mock_response)
     return response