File: common.py

package info (click to toggle)
firefox 147.0.2-1
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 4,683,484 kB
  • sloc: cpp: 7,607,246; javascript: 6,533,185; ansic: 3,775,227; python: 1,415,393; xml: 634,561; asm: 438,951; java: 186,241; sh: 62,752; makefile: 18,079; objc: 13,092; perl: 12,808; yacc: 4,583; cs: 3,846; pascal: 3,448; lex: 1,720; ruby: 1,003; php: 436; lisp: 258; awk: 247; sql: 66; sed: 54; csh: 10; exp: 6
file content (84 lines) | stat: -rw-r--r-- 2,655 bytes parent folder | download | duplicates (17)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
GPT_SHIM_MSG = "Google Publisher Tags is being shimmed by Firefox"
GAGTA_SHIM_MSG = "Google Analytics and Tag Manager is being shimmed by Firefox"


async def is_gtag_placeholder_displayed(client, url, finder, **kwargs):
    await client.navigate(url, **kwargs)
    client.execute_async_script(
        """
        const done = arguments[0];
        if (window.dataLayer?.push?.toString() === [].push.toString()) {
            return done();
        }
        setTimeout(() => {
            dataLayer.push({
                event: "datalayerReady",
                eventTimeout: 1,
                eventCallback: done,
            });
        }, 100);
    """
    )
    return client.is_displayed(client.find_element(finder))


async def clicking_link_navigates(client, url, finder, **kwargs):
    await client.navigate(url, **kwargs)
    elem = client.await_element(finder)
    return client.session.execute_async_script(
        """
        const elem = arguments[0],
              done = arguments[1];
        window.onbeforeunload = function() {
            done(true);
        };
        elem.click();
        setTimeout(() => {
            done(false);
        }, 1000);
    """,
        args=[elem],
    )


async def verify_redirectors(client, urls, expected="REDIRECTED"):
    await client.navigate(client.inline("<html>"))
    for url, type in urls.items():
        assert expected == client.execute_async_script(
            """
            const [url, type, resolve] = arguments;
            fetch(url).then(async response => {
              if (!response.ok) {
                return resolve("FAILED");
              }

              try {
                if (type === "image") {
                  const blob = await response.blob();
                  const url = URL.createObjectURL(blob);
                  const img = new Image(1, 1);
                  await new Promise((res, rej) => {
                    img.onerror = rej;
                    img.onload = res;
                    img.src = url;
                  });
                } else if (type === "js") {
                  const text = await response.text();
                  if (!text.includes("This script is intentionally empty")) {
                    throw "";
                  }
                } else {
                  return resolve("UNKNOWN TYPE");
                }
              } catch(_) {
                return resolve("TYPE MISMATCH");
              }

              resolve(response.redirected ? "REDIRECTED" : "LOADED");
            }).catch(_ => {
              resolve("BLOCKED");
            });
        """,
            url,
            type,
        )