瀏覽代碼

HUE-8758 [model] Test checking get document history with connector

Romain 5 年之前
父節點
當前提交
ad570cdf25
共有 1 個文件被更改,包括 29 次插入1 次删除
  1. 29 1
      desktop/core/src/desktop/models_tests.py

+ 29 - 1
desktop/core/src/desktop/models_tests.py

@@ -32,6 +32,7 @@ from notebook.models import import_saved_beeswax_query
 from useradmin.models import get_default_user_group, User
 
 from desktop.converters import DocumentConverter
+from desktop.lib.connectors.models import Connector
 from desktop.lib.django_test_util import make_logged_in_client
 from desktop.lib.fs import ProxyFS
 from desktop.lib.test_utils import grant_access
@@ -54,7 +55,6 @@ class TestDocument2(object):
   def setUp(self):
     self.client = make_logged_in_client(username="doc2", groupname="doc2", recreate=True, is_superuser=False)
     self.user = User.objects.get(username="doc2")
-    grant_access("doc2", "doc2", "beeswax")
 
     # This creates the user directories for the new user
     response = self.client.get('/desktop/api2/doc/')
@@ -419,6 +419,34 @@ class TestDocument2(object):
       query.delete()
 
 
+  def test_get_history_with_connector(self):
+    connector = Connector.objects.create(
+        name='MySql',
+        dialect='mysql'
+    )
+
+    query = Document2.objects.create(
+        name='test_get_history',
+        type='query-hive',
+        owner=self.user,
+        is_history=False,
+        connector=connector
+    )
+
+    try:
+      history = Document2.objects.get_history(user=self.user, doc_type='query-hive', connector_id=connector.id)
+      assert_false(history.filter(name='test_get_history').exists())
+
+      query.is_history = True
+      query.save()
+
+      history = Document2.objects.get_history(user=self.user, doc_type='query-hive', connector_id=connector.id)
+      assert_true(history.filter(name='test_get_history').exists())
+    finally:
+      query.delete()
+      connector.delete()
+
+
   def test_validate_immutable_user_directories(self):
     # Test that home and Trash directories cannot be recreated or modified
     test_dir = Directory.objects.create(name='test_dir', owner=self.user, parent_directory=self.home_dir)