Add missing unit tests.