change IMA tests

tests/redirects/google-ima3.test.js

@@ -88,45 +88,29 @@ test('Ima - run requestAds function twice', (assert) => {
     });
 });
 
-test('Ima - check if google.ima.dai was NOT overwritten', (assert) => {
-    // Test for https://github.com/AdguardTeam/Scriptlets/issues/331
-    const done = assert.async();
-
+// https://github.com/AdguardTeam/Scriptlets/issues/331
+test('Ima - check if DAI API is not discarded by the redirect', (assert) => {
+    // Define some dummy DAI API which cannot be discarded by the redirect
     window.google = {
         ima: {
+            // should be kept
             dai: {
                 api: {
-                    StreamManager() { },
+                    foo: 'bar',
                 },
             },
-            shouldBeDiscarded: true,
+            // should be discarded
+            bar: 'foo',
         },
     };
 
     runRedirect(name);
 
-    let number = 0;
-    const test = () => {
-        number += 1;
-    };
-    const { ima } = window.google;
-    const AdsLoader = new ima.AdsLoader();
-    AdsLoader.addEventListener(ima.AdsManagerLoadedEvent.Type.ADS_MANAGER_LOADED, test);
-
-    assert.ok(
-        typeof window.google.ima.dai.api.StreamManager === 'function',
-        'window.google.ima.dai.api.StreamManager was not overwritten',
-    );
+    // check if DAI API is not discarded by the redirect
+    assert.ok(window.google.ima.dai, 'window.google.ima.dai exists');
+    assert.ok(window.google.ima.dai.api, 'window.google.ima.dai.api exists');
+    assert.strictEqual(window.google.ima.dai.api.foo, 'bar', 'window.google.ima.dai.api.foo is equal to "bar"');
 
-    // Only google.ima.dai should not be overwritten
-    assert.strictEqual(
-        window.google.ima.shouldBeDiscarded, undefined,
-        'google.ima.shouldBeDiscarded is undefined',
-    );
-
-    AdsLoader.requestAds();
-    requestAnimationFrame(() => {
-        assert.strictEqual(number, 1, 'number is equal to 1');
-        done();
-    });
+    // other properties should be discarded
+    assert.strictEqual(window.google.ima.bar, undefined, 'window.google.ima.bar is undefined');
 });