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)