Add tests for `VirtualKeyboard`