def test_update_open_files_ulimit(self):
config = AmbariConfig()
open_files_ulimit = 10240
config.set_ulimit_open_files(open_files_ulimit)
main.update_open_files_ulimit(config)
(soft_limit, hard_limit) = resource.getrlimit(resource.RLIMIT_NOFILE)
self.assertEquals(hard_limit, open_files_ulimit)
def test_update_open_files_ulimit(self):
# get the current soft and hard limits
(soft_limit, hard_limit) = resource.getrlimit(resource.RLIMIT_NOFILE)
# update will be successful only if the new value is >= soft limit
if hard_limit != resource.RLIM_INFINITY:
open_files_ulimit = soft_limit + (hard_limit - soft_limit) / 2
else:
open_files_ulimit = soft_limit
config = AmbariConfig()
config.set_ulimit_open_files(open_files_ulimit)
main.update_open_files_ulimit(config)
(soft_limit, hard_limit) = resource.getrlimit(resource.RLIMIT_NOFILE)
self.assertEquals(hard_limit, open_files_ulimit)
def test_ambari_config_get(self):
config = AmbariConfig()
#default
self.assertEqual(config.get("security", "keysdir"), "/tmp/ambari-agent")
#non-default
config.set("security", "keysdir", "/tmp/non-default-path")
self.assertEqual(config.get("security", "keysdir"), "/tmp/non-default-path")
#whitespace handling
config.set("security", "keysdir", " /tmp/non-stripped")
self.assertEqual(config.get("security", "keysdir"), "/tmp/non-stripped")
# test default value
open_files_ulimit = config.get_ulimit_open_files()
self.assertEqual(open_files_ulimit, 0)
# set a value
open_files_ulimit = 128000
config.set_ulimit_open_files(open_files_ulimit)
self.assertEqual(config.get_ulimit_open_files(), open_files_ulimit)
