Fix tests to conform to coroutines

[mccorby]

What?

What would you like to refactor?

Why?

What's your justification for a refactor? These take a lot of time... why can't this be specified into a single issue?

Breakdown

Provide a bulleted or numbered list of how you might break this refactor down into smaller issues.

• My cool issue
  • Sub-issue 1
  • Sub-issue 2
• Another high-level problem
• Et cetera

Additional Context

Add any other context about the problem here.