[enhancement] Added additional code coverage to unit tests.