Add unit tests for more of main firmware class.