def test_unbundle_opens_correct_file(self):
job_bundler = JobBundler('fake_name', {}, None, None)
job_bundler.unbundle()
self.mock_tarfile_open.assert_called_with(self.temp_directory + '/fake_name.tgz', 'r:gz')
def test_unbundle_extracts_from_tarfile(self):
return_object = self.MockContextManager()
self.mock_tarfile_open.return_value = return_object
job_bundler = JobBundler('fake_name', {}, None, None)
job_bundler.unbundle()
return_object.extractall.assert_called()