Add missing unit test file