فهرست منبع

[beeswax] Reset the cache when using the Mocked API in the tests

If not the Mocked API might not be returned and used
Romain Rigaux 10 سال پیش
والد
کامیت
6d30a7dda6
1فایلهای تغییر یافته به همراه2 افزوده شده و 0 حذف شده
  1. 2 0
      apps/beeswax/src/beeswax/tests.py

+ 2 - 0
apps/beeswax/src/beeswax/tests.py

@@ -2325,6 +2325,7 @@ class TestWithMockedServer(object):
     # Beware: Monkey patch Beeswax/Hive server with Mock API
     # Beware: Monkey patch Beeswax/Hive server with Mock API
     if not hasattr(dbms, 'OriginalBeeswaxApi'):
     if not hasattr(dbms, 'OriginalBeeswaxApi'):
       dbms.OriginalBeeswaxApi = dbms.HiveServer2Dbms
       dbms.OriginalBeeswaxApi = dbms.HiveServer2Dbms
+    dbms.DBMS_CACHE = {}
     dbms.HiveServer2Dbms = MockDbms
     dbms.HiveServer2Dbms = MockDbms
 
 
     self.client = make_logged_in_client(is_superuser=False)
     self.client = make_logged_in_client(is_superuser=False)
@@ -2334,6 +2335,7 @@ class TestWithMockedServer(object):
     grant_access("test", "test", "beeswax")
     grant_access("test", "test", "beeswax")
 
 
   def tearDown(self):
   def tearDown(self):
+    dbms.DBMS_CACHE = {}
     dbms.HiveServer2Dbms = dbms.OriginalBeeswaxApi
     dbms.HiveServer2Dbms = dbms.OriginalBeeswaxApi
 
 
   def test_bulk_query_trash(self):
   def test_bulk_query_trash(self):