Add extension functional tests