[FEATURE] Add some functional tests.