Ver Fonte

[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 há 10 anos atrás
pai
commit
6d30a7dda6
1 ficheiros alterados com 2 adições e 0 exclusões
  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):