Chore/widget unit tests