(self, wf_ex_id, status=None)
| 791 | self.assertEqual(task_ex_db.status, wf_statuses.RUNNING) |
| 792 | |
| 793 | def assert_workflow_completed(self, wf_ex_id, status=None): |
| 794 | wf_ex_db = wf_db_access.WorkflowExecution.get_by_id(wf_ex_id) |
| 795 | self.assertIn(wf_ex_db.status, wf_statuses.COMPLETED_STATUSES) |
| 796 | |
| 797 | if status: |
| 798 | self.assertIn(status, wf_statuses.COMPLETED_STATUSES) |
| 799 | self.assertEqual(wf_ex_db.status, status) |
| 800 | |
| 801 | |
| 802 | class FakeResponse(object): |