On keyboard completion, the autocomplete did not re-focus on the input. This change fixes that, and prevents any regression by adding asserts to the test suite. Bug: Issue 6253 Change-Id: I76bae30e069d28c00d3029fe36ff9d731fd7631b