Tests: Update onFocusChange tests

Test Plan:
`make test`
See the onFocusChange tests passing

Auditors: charlie