added switch button with test
added expected assertions