Improve unit test coverage.

* [ ] Ensure that we are not counting generated code in coverage metrics.
* [ ] Add simple tests for public APIs to the specified package where they are only implemented in integration tests currently.
* [ ] Add additional stories for test gaps discovered from the above.