Update context-lost and context-lost-restore tests to include extensions