Implement unit testing for existing components

[jbkolze]

Add unit tests for any existing components that lack them.

Use pytest framework. Mock as needed.

Unit tests should run quickly and without any external dependencies.