Add key unittest