1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
# ClusterShell task resource consumption/limits test suite
# Written by S. Thiell
"""Unit test for ClusterShell Task (resource limits)"""
import resource
import unittest
from TLib import HOSTNAME
from ClusterShell.Task import *
from ClusterShell.Worker.Pdsh import WorkerPdsh
class TaskRLimitsTest(unittest.TestCase):
def setUp(self):
"""set soft nofile resource limit to 100"""
self.soft, self.hard = resource.getrlimit(resource.RLIMIT_NOFILE)
resource.setrlimit(resource.RLIMIT_NOFILE, (100, self.hard))
def tearDown(self):
"""restore original resource limits"""
resource.setrlimit(resource.RLIMIT_NOFILE, (self.soft, self.hard))
def _testPopen(self, stderr):
task = task_self()
task.set_info("fanout", 10)
for i in range(2000):
worker = task.shell("/bin/hostname", stderr=stderr)
# run task
task.resume()
def testPopen(self):
"""test resource usage with local task.shell(stderr=False)"""
self._testPopen(False)
def testPopenStderr(self):
"""test resource usage with local task.shell(stderr=True)"""
self._testPopen(True)
def _testRemote(self, stderr):
task = task_self()
task.set_info("fanout", 10)
for i in range(400):
worker = task.shell("/bin/hostname", nodes=HOSTNAME,
stderr=stderr)
# run task
task.resume()
def testRemote(self):
"""test resource usage with remote task.shell(stderr=False)"""
self._testRemote(False)
def testRemoteStderr(self):
"""test resource usage with remote task.shell(stderr=True)"""
self._testRemote(True)
def _testRemotePdsh(self, stderr):
task = task_self()
task.set_info("fanout", 10)
for i in range(200):
worker = WorkerPdsh(HOSTNAME, handler=None,
timeout=0,
command="/bin/hostname",
stderr=stderr)
task.schedule(worker)
# run task
task.resume()
def testRemotePdsh(self):
"""test resource usage with WorkerPdsh(stderr=False)"""
self._testRemotePdsh(False)
def testRemotePdshStderr(self):
"""test resource usage with WorkerPdsh(stderr=True)"""
self._testRemotePdsh(True)
|
