[tests] Add $message as third parameter for our custom assertions.