Преглед изворни кода

HUE-4148 [editor] Improve import testing of beeswax queries to notebook format

Romain Rigaux пре 9 година
родитељ
комит
f02ad203bd
1 измењених фајлова са 10 додато и 2 уклоњено
  1. 10 2
      desktop/core/src/desktop/tests_converter.py

+ 10 - 2
desktop/core/src/desktop/tests_converter.py

@@ -57,7 +57,7 @@ class TestDocumentConverter(object):
     ]
     file_resources = [{'type': 'jar', 'path': '/tmp/doc2/test.jar'}]
     functions = [{'name': 'myUpper', 'class_name': 'org.hue.udf.MyUpper'}]
-    design = hql_query(sql, settings=settings, file_resources=file_resources, functions=functions)
+    design = hql_query(sql, database='etl', settings=settings, file_resources=file_resources, functions=functions)
 
     query = SavedQuery.objects.create(
         type=SavedQuery.TYPES_MAPPING['hql'],
@@ -88,11 +88,15 @@ class TestDocumentConverter(object):
       assert_equal('ready', doc2.data_dict['snippets'][0]['status'])
       assert_equal(sql, doc2.data_dict['snippets'][0]['statement'])
       assert_equal(sql, doc2.data_dict['snippets'][0]['statement_raw'])
+      assert_equal('etl', doc2.data_dict['snippets'][0]['database'])
 
       # Verify snippet properties
       assert_equal(settings, doc2.data_dict['snippets'][0]['properties']['settings'])
       assert_equal(file_resources, doc2.data_dict['snippets'][0]['properties']['files'])
       assert_equal(functions, doc2.data_dict['snippets'][0]['properties']['functions'])
+
+      # Verify default properties
+      assert_true(doc2.data_dict['isSaved'])
     finally:
       query.delete()
 
@@ -103,7 +107,7 @@ class TestDocumentConverter(object):
         {'key': 'EXPLAIN_LEVEL', 'value': '2'},
         {'key': 'ABORT_ON_ERROR', 'value': '1'}
     ]
-    design = hql_query(sql, settings=settings)
+    design = hql_query(sql, database='etl', settings=settings)
 
     query = SavedQuery.objects.create(
         type=SavedQuery.TYPES_MAPPING['impala'],
@@ -134,9 +138,13 @@ class TestDocumentConverter(object):
       assert_equal('ready', doc2.data_dict['snippets'][0]['status'])
       assert_equal(sql, doc2.data_dict['snippets'][0]['statement'])
       assert_equal(sql, doc2.data_dict['snippets'][0]['statement_raw'])
+      assert_equal('etl', doc2.data_dict['snippets'][0]['database'])
 
       # Verify snippet properties
       assert_equal(settings, doc2.data_dict['snippets'][0]['properties']['settings'])
+
+      # Verify default properties
+      assert_true(doc2.data_dict['isSaved'])
     finally:
       query.delete()