Wrap entire test widget in .test