Add support for asserting with promises