EME: Allow temporary sessions to be removed for ClearKey only.

third_party/WebKit/LayoutTests/media/encrypted-media/encrypted-media-utils.js

 var consoleDiv = null;
 
 function consoleWrite(text)
 {
     if (!consoleDiv && document.body) {
         consoleDiv = document.createElement('div');
         document.body.appendChild(consoleDiv);
     }
     var span = document.createElement('span');
     span.appendChild(document.createTextNode(text));
@@ skipping 164 matching lines @@
     for (var entry of keyStatuses.entries()) {
         consoleWrite(arrayBufferAsString(entry[0]) + ": " + entry[1]);
     }
     consoleWrite("keyStatuses.forEach()");
     keyStatuses.forEach(function(status, keyId) {
         consoleWrite(arrayBufferAsString(keyId) + ": " + status);
     });
 }
 
 // Verify that |keyStatuses| contains just the keys in |keys.expected|
-// and none of the keys in |keys.unexpected|. All keys should have status
-// 'usable'. Example call: verifyKeyStatuses(mediaKeySession.keyStatuses,
-// { expected: [key1], unexpected: [key2] });
-function verifyKeyStatuses(keyStatuses, keys)
-{
-    var expected = keys.expected || [];
-    var unexpected = keys.unexpected || [];
+// and none of the keys in |keys.unexpected|. All expected keys should have
+// status |status|. Example call: verifyKeyStatuses(mediaKeySession.keyStatuses,
+// { expected: [key1], unexpected: [key2] }, 'usable');
+function verifyKeyStatuses(keyStatuses, keys, status) {
+  var expected = keys.expected || [];
+  var unexpected = keys.unexpected || [];
+  status = status || 'usable';
 
-    // |keyStatuses| should have same size as number of |keys.expected|.
-    assert_equals(keyStatuses.size, expected.length);
+  // |keyStatuses| should have same size as number of |keys.expected|.
+  assert_equals(keyStatuses.size, expected.length);
 
-    // All |keys.expected| should be found.
-    expected.map(function(key) {
-        assert_true(keyStatuses.has(key));
-        assert_equals(keyStatuses.get(key), 'usable');
-    });
+  // All |keys.expected| should be found.
+  expected.map(function(key) {
+    assert_true(keyStatuses.has(key));
+    assert_equals(keyStatuses.get(key), status);
+  });
 
-    // All |keys.unexpected| should not be found.
-    unexpected.map(function(key) {
-        assert_false(keyStatuses.has(key));
-        assert_equals(keyStatuses.get(key), undefined);
-    });
+  // All |keys.unexpected| should not be found.
+  unexpected.map(function(key) {
+    assert_false(keyStatuses.has(key));
+    assert_equals(keyStatuses.get(key), undefined);
+  });
 }
 
 // Encodes |data| into base64url string. There is no '=' padding, and the
 // characters '-' and '_' must be used instead of '+' and '/', respectively.
 function base64urlEncode(data)
 {
     var result = btoa(String.fromCharCode.apply(null, data));
     return result.replace(/=+$/g, '').replace(/\+/g, "-").replace(/\//g, "_");
 }
 
@@ skipping 90 matching lines @@
         mediaKeys = result;
         mediaKeySession = mediaKeys.createSession();
         return mediaKeySession.generateRequest('keyids', request);
     }).then(function() {
         return mediaKeySession.update(jwkSet);
     }).then(function() {
         return Promise.resolve(mediaKeys);
     });
 }
 
+// Convert an event into a promise. When |event| is fired on |object|,
+// call |func| to handle the event and either resolve or reject the promise.
+// The event is only fired once.
+function waitForSingleEvent(object, event, func) {
+  return new Promise(function(resolve, reject) {
+    object.addEventListener(event, function listener(e) {
+      object.removeEventListener(event, listener);
+      func(e, resolve, reject);
+    });
+  });
+};
+
 // Play the specified |content| on |video|. Returns a promise that is resolved
 // after the video plays for |duration| seconds.
 function playVideoAndWaitForTimeupdate(video, content, duration)
 {
     video.src = content;
     video.play();
     return new Promise(function(resolve) {
         video.addEventListener('timeupdate', function listener(event) {
             if (event.target.currentTime < duration)
                 return;
             video.removeEventListener('timeupdate', listener);
             resolve('success');
         });
     });
 }
 
 // Verifies that the number of existing MediaKey and MediaKeySession objects
 // match what is expected.
 function verifyMediaKeyAndMediaKeySessionCount(
     expectedMediaKeysCount, expectedMediaKeySessionCount, description)
 {
     assert_equals(window.internals.mediaKeysCount(),
                   expectedMediaKeysCount,
                   description + ', MediaKeys:');
     assert_equals(window.internals.mediaKeySessionCount(),
                   expectedMediaKeySessionCount,
                   description + ', MediaKeySession:');
 }