WM: Fix input dispatch timeout for instrumented processes

Input dispatch timeout varies based on the process
state. This state was queried when the ActivityRecord
was reparented. But this happens too early and as a
result the ActivityRecord always got a default
timeout.

Fix this by updating the timeout when attaching a
process to ActivityRecord.

Test: atest WmTests:ActivityRecordTests#testKeyDispatchTimeout
Bug: 162542125
Change-Id: I85db6b652790affc85d3464cca98ff8da3bab73b