def test_assert_num_tasks(): a_mixin = AsyncTestMixin() a_mixin.sub_setUp() a_mixin.assertNumCrosstownTasks(0) def some_task(): pass through_to_you = crosstown_traffic() through_to_you(some_task) a_mixin.assertNumCrosstownTasks(1)
def test_no_more_tasks(): a_mixin = AsyncTestMixin() a_mixin.sub_setUp() a_mixin.assertNumCrosstownTasks(0) def some_task(): pass through_to_you = crosstown_traffic() through_to_you(some_task) same_task = a_mixin.next_task() # That will be the only (and thus last) task. with pytest.raises(StopIteration): a_mixin.next_task()
def test_next_task(self): a_mixin = AsyncTestMixin() a_mixin.sub_setUp() a_mixin.assertNumCrosstownTasks(0) def some_task(): pass through_to_you = crosstown_traffic() through_to_you(some_task) self.assertIs(some_task, a_mixin.next_task())