adding check for optional segment_ids message to tests