def test_get_file_content(self, open_mock, _): generate_workflows(2, 2, 2, 2, 2, self._store) file_mock = mock.MagicMock() open_mock.return_value = file_mock file_mock.__enter__.return_value = file_mock file_mock.read.return_value = 'some_content' content = self._data_builder.get_file_content('workflow_0', 'instance_0', 'job_0', 0, 'info') self.assertEqual('some_content', content)
def _add_tokens(self, _, __): generate_workflows(2, 2, 2, 2, 2, self._store)