Refactored test suit to use widget-tester