Selaa lähdekoodia

HUE-7248 [adls] fix test_export_result

jdesjean 8 vuotta sitten
vanhempi
commit
15519f5883
1 muutettua tiedostoa jossa 15 lisäystä ja 1 poistoa
  1. 15 1
      desktop/libs/notebook/src/notebook/tests.py

+ 15 - 1
desktop/libs/notebook/src/notebook/tests.py

@@ -22,7 +22,7 @@ from nose.tools import assert_equal, assert_true, assert_false
 
 from django.contrib.auth.models import User
 from django.core.urlresolvers import reverse
-
+from azure.conf import is_adls_enabled
 from desktop.lib.django_test_util import make_logged_in_client
 from desktop.lib.test_utils import grant_access, add_permission
 from desktop.models import Directory, Document, Document2
@@ -35,6 +35,7 @@ from notebook.connectors.base import Notebook, QueryError, Api
 from notebook.decorators import api_error_handler
 from notebook.conf import get_ordered_interpreters, INTERPRETERS_SHOWN_ON_WHEEL, INTERPRETERS
 
+
 try:
   from collections import OrderedDict
 except ImportError:
@@ -406,6 +407,19 @@ class TestNotebookApiMocked(object):
     assert_equal(0, data['status'], data)
     assert_equal('/user/hue/path.csv', data['watch_url']['destination'], data)
 
+    if is_adls_enabled():
+        response = self.client.post(reverse('notebook:export_result'), {
+            'notebook': notebook_json,
+            'snippet': json.dumps(json.loads(notebook_json)['snippets'][0]),
+            'format': json.dumps('hdfs-file'),
+            'destination': json.dumps('adl:/user/hue/path.csv'),
+            'overwrite': json.dumps(False)
+        })
+
+        data = json.loads(response.content)
+        assert_equal(0, data['status'], data)
+        assert_equal('adl:/user/hue/path.csv', data['watch_url']['destination'], data)
+
 
 def test_get_interpreters_to_show():
   default_interpreters = OrderedDict((