def test_run_with_no_pid_file(self, mock1, mock2):
segments = [
self._get_mock_segment('seg1', '/tmp/gpseg1', 1234, 'host1',
'host1')
]
c = CleanSharedMem(segments)
c.run()
def test_run(self, mock1, mock2):
segments = [self._get_mock_segment('seg1', '/tmp/gpseg1', 1234, 'host1', 'host1')]
c = CleanSharedMem(segments)
file_contents = 'asdfads\nasdfsd asdfadsf\n12345 23456'.split()
m = MagicMock()
m.return_value.__enter__.return_value.readlines.return_value = file_contents
with patch('__builtin__.open', m, create=True):
c.run()
def test_run_with_error_in_workerpool(self, mock1, mock2, mock3):
segments = [self._get_mock_segment('seg1', '/tmp/gpseg1', 1234, 'host1', 'host1')]
c = CleanSharedMem(segments)
file_contents = 'asdfads\nasdfsd asdfadsf\n12345 23456'.split()
m = MagicMock()
m.return_value.__enter__.return_value.readlines.return_value = file_contents
with patch('__builtin__.open', m, create=True):
with self.assertRaisesRegexp(Exception, 'Unable to clean up shared memory'):
c.run()
def test_run_with_invalid_pid_file(self, mock1, mock2):
segments = [self._get_mock_segment('seg1', '/tmp/gpseg1', 1234, 'host1', 'host1')]
c = CleanSharedMem(segments)
file_contents = 'asdfadsasdfasdf'.split()
m = MagicMock()
m.return_value.__enter__.return_value.readlines.return_value = file_contents
with patch('__builtin__.open', m, create=True):
with self.assertRaisesRegexp(Exception, 'Unable to clean up shared memory for segment'):
c.run()
def test_run_with_no_pid_file(self, mock1, mock2):
segments = [self._get_mock_segment('seg1', '/tmp/gpseg1', 1234, 'host1', 'host1')]
c = CleanSharedMem(segments)
c.run()
