Python generate_workflows Examples

Example #1

File: data_builder_test.py Project: zhengge2017/pinball

    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)

Example #2

File: data_builder_test.py Project: zhengge2017/pinball

 def _add_tokens(self, _, __):
     generate_workflows(2, 2, 2, 2, 2, self._store)