Add simple tests for Supply.wait