This is expected and is a constraint of macros in Swift. The presence of the await keyword signals to the compiler that some subexpression of the expectation is asynchronous, but it is impossible to know exactly which subexpression from the AST alone. This means it is not possible to correctly decompose the expression as we can with synchronous, non-throwing ones.

@grynspan Thanks! If I understand the issue correctly, for a test case like

@Test func testAsyncExpectation() {
  func value() async -> Int { 1 }
  #expect(await value() == 2)
}

the expect macro is going to resolve to a __checkValue (generic boolean expression) and not a __checkBinaryOperation (where we would be able to decompose actual and expected values on lhs and rhs respectively).

Can't we work around this in private func _parseCondition by not just looking for InfixOperatorExprSyntax but anything embedded in an AwaitExprSyntax? In this case:

AwaitExprSyntax
├─awaitKeyword: keyword(SwiftSyntax.Keyword.await)
╰─expression: InfixOperatorExprSyntax
  ├─leftOperand: FunctionCallExprSyntax
  │ ├─calledExpression: DeclReferenceExprSyntax
  │ │ ╰─baseName: identifier("value")
  │ ├─leftParen: leftParen
  │ ├─arguments: LabeledExprListSyntax
  │ ├─rightParen: rightParen
  │ ╰─additionalTrailingClosures: MultipleTrailingClosureElementListSyntax
  ├─operator: BinaryOperatorExprSyntax
  │ ╰─operator: binaryOperator("==")
  ╰─rightOperand: IntegerLiteralExprSyntax
    ╰─literal: integerLiteral("2")

Also, how is this issue different from matching a try expression?

The same issue exists for a try expression.

Imagine we add support for an AwaitExprSyntax path. What would the implementation look like? It would need to preserve await semantics on all parts of the expanded expression, but it is impossible to know which parts need to be awaited (and which are synchronous) just from the syntax tree alone.

This means we'd need additional overloads of every __check function that are async, throws, and async throws. This quadruples the number of overloads that need to be resolved (which has a worse-than-linear impact on compile times), and it is semantically incorrect because we'd have to introduce multiple fake suspension points that could affect the correctness of a test.

The expanded form of #expect(await x == y) would look something like:

__checkBinaryOperation(
  await (x, Testing.__requiringAwait).0,
  { await $0 == $1() },
  await (y, Testing.__requiringAwait).0,
  ...
).__expected()

And would have at least 4 and at most 6 suspension points when the original expression had as few as 1.

Now, it may be possible to simplify that expansion a bit given that the outermost call to #expect() would need an await keyword applied to it, but the effect keywords on a macro are not visible to the macro during expansion, so we can't know if the developer typed it or not. We could place our own await keyword on the call to __check(), and that could let us simplify part of (but not all of) the macro expansion, but we're still left with four times as many overloads as before. rethrows (and a hypothetical reasync) doesn't help us because we must express the infix operator (among other possible subexpressions) as a closure, and that requires us to explicitly write try and/or await within the closure body, and that defeats the purpose of rethrows/reasync.

Thank you for the detailed explanation. I understand the tradeoffs now.

We could place our own await keyword on the call to __check(), and that could let us simplify part of (but not all of) the macro expansion, but we're still left with four times as many overloads as before.

This seems to be the only viable solution to the issue.

The issue can be avoided by extracting the async subexpression out into a separate expression:

let y = await foo()
#expect(x == y)

So I'd steer developers toward that solution as preferable.

I appreciate the writeup and the warning in #302. Thank you @grynspan

Reopening. Now that we have #isolation, it may be possible to avoid the unnecessary hops.

Blocked by swiftlang/swift#76930.

Tracked internally as rdar://137583415.