1// Copyright (C) 2018 The Android Open Source Project
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7//      http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15const fs = require('fs');
16const path = require('path');
17const NodeEnvironment = require('jest-environment-node');
18const puppeteer = require('puppeteer');
19
20module.exports = class HeadlessEnvironment extends NodeEnvironment {
21  constructor(config) {
22    super(config);
23  }
24
25  async setup() {
26    await super.setup();
27    this.global.__BROWSER__ = await puppeteer.launch();
28  }
29
30  async teardown() {
31    await this.global.__BROWSER__.close();
32    await super.teardown();
33  }
34
35  runScript(script) {
36    return super.runScript(script);
37  }
38}
39