def test_remove_bbcp(self, mock_run_ignoring_errors): cloud = mock.Mock() cloud.hosts_with_bbcp = {'fake_host_1', 'fake_host_2'} cloud.position = 'src' bbcp_copier.remove_bbcp(cloud) self.assertEqual(2, mock_run_ignoring_errors.call_count)
def test_remove_bbcp(self, mock_run_ignoring_errors): cloud = mock.Mock() cloud.hosts_with_bbcp = {'fake_host_1', 'fake_host_2'} cloud.position = 'src' bbcp_copier.remove_bbcp(cloud) self.assertEqual(2, mock_run_ignoring_errors.call_count)
def run(self, **kwargs): bbcp_copier.remove_bbcp(self.src_cloud) bbcp_copier.remove_bbcp(self.dst_cloud)