1 files changed, 22 insertions, 12 deletions
diff --git a/tests/wpt/tests/ai/language_detection/resources/iframe-helper.html b/tests/wpt/tests/ai/language_detection/resources/iframe-helper.html
index fe07de3f8ab..35ba8525587 100644
--- a/tests/wpt/tests/ai/language_detection/resources/iframe-helper.html
+++ b/tests/wpt/tests/ai/language_detection/resources/iframe-helper.html
@@ -1,18 +1,28 @@
 <!DOCTYPE HTML>
 <meta charset="utf-8">
+<script src="/resources/testdriver.js"></script>
+<script src="/resources/testdriver-vendor.js"></script>
+<body></body>
 <script>
+test_driver.set_test_context(parent);
+
 window.onmessage = async message => {
-  switch (message.data.type) {
-    case 'LanguageDetectorCreate':
-      LanguageDetector.create()
-        .then(t => parent.postMessage({success: 'Success'}))
-        .catch(err => parent.postMessage({err}));
-      break;
-    case 'LanguageDetectorAvailability':
-      LanguageDetector.availability()
-        .then(t => parent.postMessage({success: availability}))
-        .catch(err => parent.postMessage({err}));
-      break;
-  };
+  const {id, type} = message.data;
+  try {
+    switch (type) {
+      case 'LanguageDetectorCreate':
+        await test_driver.bless("Create LanguageDetector", null, window);
+        await LanguageDetector.create();
+        parent.postMessage({id, success: 'Success'}, '*');
+        break;
+      case 'LanguageDetectorAvailability':
+        await test_driver.bless("Check LanguageDetector Availability", null, window);
+        const availability = await LanguageDetector.availability();
+        parent.postMessage({id, success: availability}, '*');
+        break;
+    }
+  } catch (err) {
+    parent.postMessage({id, err: err}, '*');
+  }
 };
 </script>