| 829 | calls = [0] |
| 830 | |
| 831 | class _MockNode(BaseNode): |
| 832 | wait_for_output: bool = True |
| 833 | rerun_on_resume: bool = True |
| 834 | |
| 835 | async def _run_impl( |
| 836 | self, *, ctx: Context, node_input: Any |
| 837 | ) -> AsyncGenerator[Any, None]: |
| 838 | calls[0] += 1 |
| 839 | if calls[0] == 2: |
| 840 | yield 'success' |
| 841 | |
| 842 | class _ParentNode(BaseNode): |
| 843 | rerun_on_resume: bool = True |