def METHOD_NAME(self) -> Assignment:
    """Create a test assignment for self.task_run using mock agents"""
    task_run = self.task_run
    assignment_id = self.db.new_assignment(
        task_run.task_id,
        task_run.db_id,
        task_run.requester_id,
        task_run.task_type,
        task_run.provider_type,
    )
    assign = Assignment.get(self.db, assignment_id)
    unit_id = self.db.new_unit(
        task_run.task_id,
        task_run.db_id,
        task_run.requester_id,
        assignment_id,
        0,
        0,
        task_run.provider_type,
        task_run.task_type,
    )
    unit = MockUnit.get(self.db, unit_id)
    worker_id = self.db.new_worker("MOCK_TEST_WORKER", MOCK_PROVIDER_TYPE)
    worker = MockWorker.get(self.db, worker_id)
    agent_id = self.db.new_agent(
        worker.db_id,
        unit_id,
        task_run.task_id,
        task_run.db_id,
        assignment_id,
        task_run.task_type,
        task_run.provider_type,
    )
    Agent = MockAgent.get(self.db, agent_id)
    return assign