|
|
@@ -390,8 +390,8 @@ class TestMapReduce2NoHadoop:
|
|
|
|
|
|
def setUp(self):
|
|
|
# Beware: Monkey patching
|
|
|
- if not hasattr(resource_manager_api, 'old_get_resource_manager_pool'):
|
|
|
- resource_manager_api.old_get_resource_manager_pool = resource_manager_api.get_resource_manager_pool
|
|
|
+ if not hasattr(resource_manager_api, 'old_get_resource_manager_api'):
|
|
|
+ resource_manager_api.old_get_resource_manager = resource_manager_api.get_resource_manager
|
|
|
if not hasattr(mapreduce_api, 'old_get_mapreduce_api'):
|
|
|
mapreduce_api.old_get_mapreduce_api = mapreduce_api.get_mapreduce_api
|
|
|
if not hasattr(history_server_api, 'old_get_history_server_api'):
|
|
|
@@ -405,7 +405,7 @@ class TestMapReduce2NoHadoop:
|
|
|
grant_access("test2", "test2", "jobbrowser")
|
|
|
self.user2 = User.objects.get(username='test2')
|
|
|
|
|
|
- resource_manager_api.get_resource_manager_pool = lambda: MockResourceManagerApiPool()
|
|
|
+ resource_manager_api.get_resource_manager = lambda username: MockResourceManagerApi(username)
|
|
|
mapreduce_api.get_mapreduce_api = lambda username: MockMapreduceApi(username)
|
|
|
history_server_api.get_history_server_api = lambda username: HistoryServerApi(username)
|
|
|
|
|
|
@@ -417,7 +417,7 @@ class TestMapReduce2NoHadoop:
|
|
|
|
|
|
|
|
|
def tearDown(self):
|
|
|
- resource_manager_api.get_resource_manager_pool = getattr(resource_manager_api, 'old_get_resource_manager_pool')
|
|
|
+ resource_manager_api.get_resource_manager = getattr(resource_manager_api, 'old_get_resource_manager')
|
|
|
mapreduce_api.get_mapreduce_api = getattr(mapreduce_api, 'old_get_mapreduce_api')
|
|
|
history_server_api.get_history_server_api = getattr(history_server_api, 'old_get_history_server_api')
|
|
|
|
|
|
@@ -502,23 +502,6 @@ class TestMapReduce2NoHadoop:
|
|
|
finally:
|
|
|
MockResourceManagerApi.APPS[job_id]['state'] = 'RUNNING'
|
|
|
|
|
|
-class MockResourceManagerApiPool:
|
|
|
-
|
|
|
- def __init__(self):
|
|
|
- pool_size = 10
|
|
|
- self.rmobj_pool = Queue.LifoQueue()
|
|
|
- for i in range(pool_size):
|
|
|
- rm_instance = MockResourceManagerApi()
|
|
|
- self.rmobj_pool.put(rm_instance)
|
|
|
-
|
|
|
- def get(self, user):
|
|
|
- rmobj = self.rmobj_pool.get()
|
|
|
- rmobj.setuser(user)
|
|
|
- return rmobj
|
|
|
-
|
|
|
- def put(self, rmobj):
|
|
|
- self.rmobj_pool.put(rmobj)
|
|
|
-
|
|
|
class MockResourceManagerApi:
|
|
|
APPS = {
|
|
|
'application_1356251510842_0054': {
|
|
|
@@ -620,7 +603,7 @@ class MockResourceManagerApi:
|
|
|
}
|
|
|
}
|
|
|
|
|
|
- def __init__(self, rm_url=None): pass
|
|
|
+ def __init__(self, user, rm_url=None): pass
|
|
|
|
|
|
def apps(self, **kwargs):
|
|
|
return {
|
|
|
@@ -643,17 +626,6 @@ class MockResourceManagerApi:
|
|
|
u'app': MockResourceManagerApi.APPS[job_id]
|
|
|
}
|
|
|
|
|
|
- def setuser(self, user):
|
|
|
- self.user = user
|
|
|
- return user
|
|
|
-
|
|
|
- @property
|
|
|
- def user(self):
|
|
|
- return self.username
|
|
|
-
|
|
|
- @property
|
|
|
- def username(self):
|
|
|
- return self.user.username
|
|
|
|
|
|
class MockMapreduce2Api(object):
|
|
|
"""
|