Improve unit test coverage.

[robshakir]

• 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.

[robshakir]

Blocked by: fix coveralls.io 🎢