Adding AwaitData query flag in tailable tests