Refactor WidgetTestCase :)