'use strict';

function EnsureExt(extName, shouldHave=true) {
    EnsureExtFor('webgl', extName, shouldHave);
    EnsureExtFor('webgl2', extName, shouldHave);
}

function EnsureExtFor(contextType, extName, shouldHave=true) {
    var c = document.createElement('canvas');
    var gl = c.getContext(contextType);

    if (!gl) {
        todo(false, 'Failed to create context: ' + contextType);
        return;
    }

    var ext = gl.getExtension(extName);
    var haveText = ' have ' + contextType + ' extension ' + extName + '.';
    if (shouldHave) {
        ok(ext, 'Should' + haveText);
    } else {
        ok(!ext, 'Should not' + haveText);
    }
}

function Lastly_WithDraftExtsEnabled(func) {
    SimpleTest.waitForExplicitFinish();

    var fnEnsure = function() {
        func();
        SimpleTest.finish();
    };

    if ('SpecialPowers' in window) {
        var prefStateList = [
            ['webgl.enable-draft-extensions', true],
        ];
        var prefEnv = {'set': prefStateList};
        SpecialPowers.pushPrefEnv(prefEnv, fnEnsure);
    } else {
        console.log('Couldn\'t use SpecialPowers to enable draft extensions.');
        fnEnsure();
    }
}