Browse Source

[impala] Mock calls to Beeswax in the tests

Remove unsued import to liboozie in valid configuration
Romain Rigaux 13 years ago
parent
commit
ed20709
2 changed files with 19 additions and 6 deletions
  1. 18 5
      apps/impala/src/impala/tests.py
  2. 1 1
      desktop/libs/liboozie/src/liboozie/conf.py

+ 18 - 5
apps/impala/src/impala/tests.py

@@ -19,17 +19,30 @@ import re
 
 from nose.tools import assert_true, assert_equal, assert_false
 
+from beeswax.server import dbms
 from desktop.lib.django_test_util import make_logged_in_client
 
 
+class MockDbms:
+  def get_databases(self):
+    return ['db1', 'db2']
+
+
 class TestImpala:
   def setUp(self):
     self.client = make_logged_in_client()
 
   def test_basic_flow(self):
-    response = self.client.get("/impala/")
-    assert_true(re.search('<li id="impalaIcon"\W+class="active', response.content), response.content)
-    assert_true('Query Editor' in response.content)
+    try:
+      # Mock DB calls as we don't need the real ones
+      self.prev_dbms = dbms.get
+      dbms.get = lambda a, b: MockDbms()
+  
+      response = self.client.get("/impala/")
+      assert_true(re.search('<li id="impalaIcon"\W+class="active', response.content), response.content)
+      assert_true('Query Editor' in response.content)
 
-    response = self.client.get("/impala/execute/")
-    assert_true('Query Editor' in response.content)
+      response = self.client.get("/impala/execute/")
+      assert_true('Query Editor' in response.content)
+    finally:
+      dbms.get = self.prev_dbms

+ 1 - 1
desktop/libs/liboozie/src/liboozie/conf.py

@@ -46,7 +46,7 @@ def config_validator():
   Called by core check_config() view.
   """
   from hadoop.cluster import get_all_hdfs
-  from liboozie.oozie_api import get_oozie
+#  from liboozie.oozie_api import get_oozie
 
   res = []