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()