Add remaining tests

co-authored-by: Conrad <conrad@zed.dev>