def testSetupError(self, mock_subprocess_call):
"""Tests that the specified directory is used if created."""
mock_subprocess_call.return_value = -1
test_state = state.DFTimewolfState(config.Config)
scp_exporter = scp_ex.SCPExporter(test_state)
scp_exporter.SetUp('/path1,/path2', '/destination', 'fakeuser',
'fakehost', 'fakeid', True)
self.assertEqual(test_state.errors[0],
('Unable to connect to host.', True))
def testProcess(self, mock_subprocess_call):
"""Tests that the specified directory is used if created."""
mock_subprocess_call.return_value = 0
test_state = state.DFTimewolfState(config.Config)
scp_exporter = scp_ex.SCPExporter(test_state)
scp_exporter.SetUp('/path1,/path2', '/destination', 'fakeuser',
'fakehost', 'fakeid', True)
scp_exporter.Process()
mock_subprocess_call.assert_called_with(
['scp', '/path1', '/path2', 'fakeuser@fakehost:/destination'])
def testSetupError(self, mock_subprocess_call):
"""Tests that the specified directory is used if created."""
mock_subprocess_call.return_value = -1
test_state = state.DFTimewolfState(config.Config)
scp_exporter = scp_ex.SCPExporter(test_state)
with self.assertRaises(errors.DFTimewolfError) as error:
scp_exporter.SetUp('/path1,/path2', '/destination', 'fakeuser',
'fakehost', 'fakeid', True)
self.assertEqual(test_state.errors[0], error.exception)
self.assertEqual(error.exception.message, 'Unable to connect to host.')
self.assertTrue(error.exception.critical)
def testSetupError(self, mock_subprocess_call):
"""Tests that recipe errors out if connection check fails."""
mock_subprocess_call.return_value = -1
test_state = state.DFTimewolfState(config.Config)
scp_exporter = scp_ex.SCPExporter(test_state)
with self.assertRaises(errors.DFTimewolfError) as error:
scp_exporter.SetUp('/path1,/path2', '/destination', 'fakeuser',
'fakehost', 'fakeid', 'upload', False, True)
self.assertEqual(test_state.errors[0], error.exception)
self.assertEqual(error.exception.message, 'Unable to connect to fakehost.')
self.assertTrue(error.exception.critical)
def testProcessError(self, mock_subprocess_call):
"""Tests that the specified directory is used if created."""
mock_subprocess_call.return_value = 0
test_state = state.DFTimewolfState(config.Config)
scp_exporter = scp_ex.SCPExporter(test_state)
scp_exporter.SetUp('/path1,/path2', '/destination', 'fakeuser',
'fakehost', 'fakeid', True)
mock_subprocess_call.return_value = -1
scp_exporter.Process()
self.assertEqual(test_state.errors[0],
("Failed copying ['/path1', '/path2']", True))
def testProcessDownloadMultiplex(self, mock_subprocess_call):
"""Tests that the specified directory is used if created."""
mock_subprocess_call.return_value = 0
test_state = state.DFTimewolfState(config.Config)
scp_exporter = scp_ex.SCPExporter(test_state)
scp_exporter.SetUp('/path1,/path2', '/destination', 'fakeuser',
'fakehost', 'fakeid', 'download', True, True)
scp_exporter.Process()
mock_subprocess_call.assert_called_with(
['scp',
'-o', 'ControlMaster=auto',
'-o', 'ControlPath=~/.ssh/ctrl-%C',
'-i', 'fakeid',
'fakeuser@fakehost:/path1 /path2', '/destination'])
def testSetup(self, mock_subprocess_call):
"""Tests that the specified directory is used if created."""
mock_subprocess_call.return_value = 0
test_state = state.DFTimewolfState(config.Config)
scp_exporter = scp_ex.SCPExporter(test_state)
scp_exporter.SetUp('/path1,/path2', '/destination', 'fakeuser',
'fakehost', 'fakeid', 'upload', False, True)
mock_subprocess_call.assert_called_with(
['ssh', '-q', '-l', 'fakeuser', 'fakehost', 'true', '-i', 'fakeid'])
# pylint: disable=protected-access
self.assertEqual(scp_exporter._destination, '/destination')
self.assertEqual(scp_exporter._hostname, 'fakehost')
self.assertEqual(scp_exporter._id_file, 'fakeid')
self.assertEqual(scp_exporter._paths, ['/path1', '/path2'])
self.assertEqual(scp_exporter._user, 'fakeuser')
def testCreateDestinationDirectory(self, mock_subprocess_call):
"""Tests that the remote directory is created as expected."""
mock_subprocess_call.return_value = 0
test_state = state.DFTimewolfState(config.Config)
scp_exporter = scp_ex.SCPExporter(test_state)
scp_exporter.SetUp('/path1,/path2', '/destination', 'fakeuser',
'fakehost', 'fakeid', 'upload', False, False)
# pylint: disable=protected-access
scp_exporter._CreateDestinationDirectory(remote=True)
mock_subprocess_call.assert_called_with(
['ssh', 'fakeuser@fakehost', 'mkdir', '-p', '/destination']
)
scp_exporter._CreateDestinationDirectory(remote=False)
mock_subprocess_call.assert_called_with(
['mkdir', '-p', '/destination']
)
def testProcessError(self, mock_subprocess_call):
"""Tests that failures creating directories are properly caught."""
mock_subprocess_call.return_value = 0
test_state = state.DFTimewolfState(config.Config)
scp_exporter = scp_ex.SCPExporter(test_state)
# pylint: disable=protected-access
scp_exporter._CreateDestinationDirectory = mock.Mock()
scp_exporter.SetUp('/path1,/path2', '/destination', 'fakeuser',
'fakehost', 'fakeid', 'upload', False, True)
mock_subprocess_call.return_value = -1
with self.assertRaises(errors.DFTimewolfError) as error:
scp_exporter.Process()
self.assertEqual(test_state.errors[0], error.exception)
self.assertEqual(error.exception.message,
"Failed copying ['/path1', '/path2']")
self.assertTrue(error.exception.critical)
def testInitialization(self):
"""Tests that the exporter can be initialized."""
test_state = state.DFTimewolfState(config.Config)
local_scp_exporter = scp_ex.SCPExporter(test_state)
self.assertIsNotNone(local_scp_exporter)
