Update extension tests.
(cherry picked from commit 5b145062a78ab7bc2b3ffe7dc9c64f039f734b1c)