Add input handler unit tests