Improve test coverage

[jeff-cohere]

Our test coverage is hovering in the low 70%s. Ideally, we'd get it to above 90%.

This is kind of an aspirational goal, because striving for complete test coverage in systems that have nontrivial error handling logic can be expensive, and hitting a high number doesn't always deliver actual benefits. Nevertheless, we aspire!