Added missing unit tests.