Add dialog tests