def METHOD_NAME(self) -> None:
    self.dc.kill_stage("agent1")
    WAIT_FOR_KILL = 5
    for _i in range(WAIT_FOR_KILL):
        agent_data = get_agent_data(conf.make_master_url())
        if len(agent_data) == 0:
            break
        if len(agent_data) == 1 and agent_data[0]["draining"] is True:
            break
        time.sleep(1)
    else:
        pytest.fail(f"Agent is still present after {WAIT_FOR_KILL} seconds")