add missing unit tests