def METHOD_NAME(self):
    self.auth.servers.running.return_value = {
        1: "server1",
        2: "server2",
    }
    self.auth.served["server1"].add("chall1")
    self.auth.served["server2"].update(["chall2", "chall3"])
    self.auth.cleanup(["chall1"])
    assert self.auth.served == {
        "server1": set(), "server2": {"chall2", "chall3"}}
    self.auth.servers.stop.assert_called_once_with(1)
    self.auth.servers.running.return_value = {
        2: "server2",
    }
    self.auth.cleanup(["chall2"])
    assert self.auth.served == {
        "server1": set(), "server2": {"chall3"}}
    assert 1 == self.auth.servers.stop.call_count
    self.auth.cleanup(["chall3"])
    assert self.auth.served == {
        "server1": set(), "server2": set()}
    self.auth.servers.stop.assert_called_with(2)