ws: Update test cases.

Fixes #940