Tests: Update set_location test in stateserver