summaryrefslogtreecommitdiffstats
path: root/testing/web-platform/tests/encrypted-media/polyfill/make-polyfill-tests.py
blob: 97c6fc74e93ba0f89f34c75cab3d4fb6ad183626 (plain)
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
#!/usr/bin/python

import os, re, os.path, glob

head = re.compile( r"^(\s*</head>)", re.MULTILINE )
runtest = re.compile( r"runTest\(\s*(\S.*?)\s*\)", re.DOTALL )

scripts = '''
    <!-- Polyfill files (NOTE: These are added by auto-generation script) -->
    <script src=/encrypted-media/polyfill/chrome-polyfill.js></script>
    <script src=/encrypted-media/polyfill/firefox-polyfill.js></script>
    <script src=/encrypted-media/polyfill/edge-persistent-usage-record.js></script>
    <script src=/encrypted-media/polyfill/edge-keystatuses.js></script>
    <script src=/encrypted-media/polyfill/clearkey-polyfill.js></script>'''

def process_file( infile, outfile ) :
    with open( outfile, "w" ) as output :
        with open( infile, "r" ) as input :
            output.write( runtest.sub( r"runTest( \1, 'polyfill: ' )", head.sub( scripts + r"\1", input.read() ) ) )

if __name__ == '__main__' :
    if (not os.getcwd().endswith('polyfill')) :
        print("Please run from polyfill directory")
        exit( 1 )

    for infile in glob.glob( "../*.html" ) :
        process_file( infile, os.path.basename( infile ) )

    for infile in glob.glob( "../resources/*.html" ) :
        process_file( infile, os.path.join( "resources", os.path.basename( infile ) ) )