Add tests for Widget helper