key_extension_test came from @apenwarr's "continue map poll" commit
(6a822dc) which was since reverted, and various APIs used by the tests
were altered. This updates the test so that go vet/go check pass.
Signed-off-by: Luke Kosewski <[email protected]>