def test_concurrency_set(self, Popen, cpu_count): """ Test for the case where PULP_CONCURRENCY is set in /etc/default/pulp_workers. """ class MockPipe(object): def communicate(self): """ Return 32\n as the output of the subprocess, to simulate PULP_CONCURRENCY being defined as 32. """ return ('32\n', ) Popen.return_value = MockPipe() concurrency = manage_workers._get_concurrency() # Since we mocked the output of the subprocess call to be 32\n, concurrency should be 32 self.assertEqual(concurrency, 32) # Make sure that Popen was called correctly Popen.assert_called_once_with('. %s; echo $PULP_CONCURRENCY' % manage_workers._ENVIRONMENT_FILE, stdout=subprocess.PIPE, shell=True) # Since the shell call was successful, no call to cpu_count() should have been made self.assertEqual(cpu_count.call_count, 0)
def test_concurrency_not_set(self, Popen, cpu_count): """ Test for the case where PULP_CONCURRENCY is not set in /etc/default/pulp_workers. """ class MockPipe(object): def communicate(self): """ Return \n as the output of the subprocess, to simulate PULP_CONCURRENCY not being defined. """ return ('\n',) Popen.return_value = MockPipe() concurrency = manage_workers._get_concurrency() # Since we mocked the cpu_count to be 16, concurrency should be 16 self.assertEqual(concurrency, 16) # Make sure that Popen and cpu_count were called correctly Popen.assert_called_once_with( '. %s; echo $PULP_CONCURRENCY' % manage_workers._ENVIRONMENT_FILE, stdout=subprocess.PIPE, shell=True) cpu_count.assert_called_once_with()
def test_concurrency_set(self, Popen, cpu_count): """ Test for the case where PULP_CONCURRENCY is set in /etc/default/pulp_workers. """ class MockPipe(object): def communicate(self): """ Return 32\n as the output of the subprocess, to simulate PULP_CONCURRENCY being defined as 32. """ return ('32\n',) Popen.return_value = MockPipe() concurrency = manage_workers._get_concurrency() # Since we mocked the output of the subprocess call to be 32\n, concurrency should be 32 self.assertEqual(concurrency, 32) # Make sure that Popen was called correctly Popen.assert_called_once_with( '. %s; echo $PULP_CONCURRENCY' % manage_workers._ENVIRONMENT_FILE, stdout=subprocess.PIPE, shell=True) # Since the shell call was successful, no call to cpu_count() should have been made self.assertEqual(cpu_count.call_count, 0)
def test_concurrency_not_set(self, Popen, cpu_count): """ Test for the case where PULP_CONCURRENCY is not set in /etc/default/pulp_workers. """ class MockPipe(object): def communicate(self): """ Return \n as the output of the subprocess, to simulate PULP_CONCURRENCY not being defined. """ return ('\n', ) Popen.return_value = MockPipe() concurrency = manage_workers._get_concurrency() # Since we mocked the cpu_count to be 16, concurrency should be 16 self.assertEqual(concurrency, 16) # Make sure that Popen and cpu_count were called correctly Popen.assert_called_once_with('. %s; echo $PULP_CONCURRENCY' % manage_workers._ENVIRONMENT_FILE, stdout=subprocess.PIPE, shell=True) cpu_count.assert_called_once_with()