Эх сурвалжийг харах

HUE-8758 [editor] Update test to provide an execute status code

Romain 5 жил өмнө
parent
commit
c058b051d0

+ 2 - 3
desktop/libs/notebook/src/notebook/connectors/base_tests.py

@@ -20,9 +20,8 @@ from builtins import object
 import json
 import sys
 
-from nose.tools import assert_equal, assert_true, assert_false
-
 from django.urls import reverse
+from nose.tools import assert_equal, assert_true, assert_false
 
 from desktop.lib.django_test_util import make_logged_in_client
 from useradmin.models import User
@@ -45,7 +44,7 @@ class TestNotebook(object):
   def test_execute_and_wait(self):
     query = Notebook()
 
-    query.execute = Mock(return_value={'history_uuid': 1})
+    query.execute = Mock(return_value={'history_uuid': 1, 'status': 0})
     query.check_status = Mock(
       side_effect=check_status_side_effect
     )