Add functional tests for image