[Kitaka] Refactored the functional test.