def test_with_subsubprocess(self):
cmd = GetMemoryUsage(None)
def bar(): # pylint: disable=blacklisted-name
create_process(foo, args=[])
def foo(): # pylint: disable=blacklisted-name
time.sleep(60)
create_process(bar, args=[])
request = et.fromstring('<get_memory_usage />')
response = et.fromstring(cmd.handle_xml(request))
processes_element = response.find('processes')
process_elements = processes_element.findall('process')
# sub-sub-processes aren't listed
self.assertTrue(len(process_elements), 2)
for process_element in process_elements:
rss_element = process_element.find('rss')
vms_element = process_element.find('vms')
shared_element = process_element.find('shared')
self.assertIsNotNone(rss_element)
self.assertIsNotNone(rss_element.text)
self.assertIsNotNone(vms_element)
self.assertIsNotNone(vms_element.text)
self.assertIsNotNone(shared_element)
self.assertIsNotNone(shared_element.text)
def test_with_main_process_only(self):
cmd = GetMemoryUsage(None)
request = et.fromstring('<get_memory_usage />')
response = et.fromstring(cmd.handle_xml(request))
processes_element = response.find('processes')
process_elements = processes_element.findall('process')
self.assertTrue(len(process_elements), 1)
main_process_element = process_elements[0]
rss_element = main_process_element.find('rss')
vms_element = main_process_element.find('vms')
shared_element = main_process_element.find('shared')
self.assertIsNotNone(rss_element)
self.assertIsNotNone(rss_element.text)
self.assertIsNotNone(vms_element)
self.assertIsNotNone(vms_element.text)
self.assertIsNotNone(shared_element)
self.assertIsNotNone(shared_element.text)