def test_save_job_writes_to_file(self): mock_job = self._create_mock_job() return_object = self.MockContextManager() self.mock_builtins_open.return_value = return_object job_bundler = JobBundler('fake_name', {}, mock_job, None) job_bundler._save_job() return_object.write.assert_called_with('something')
def test_save_job_opens_file(self): mock_job = self._create_mock_job() job_bundler = JobBundler('fake_name', {}, mock_job, None) job_bundler._save_job() self.mock_builtins_open.assert_called_with('fake_name.bin', 'w+b')