1 | "use strict";
|
2 | var __assign = (this && this.__assign) || function () {
|
3 | __assign = Object.assign || function(t) {
|
4 | for (var s, i = 1, n = arguments.length; i < n; i++) {
|
5 | s = arguments[i];
|
6 | for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p))
|
7 | t[p] = s[p];
|
8 | }
|
9 | return t;
|
10 | };
|
11 | return __assign.apply(this, arguments);
|
12 | };
|
13 | var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
|
14 | function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
|
15 | return new (P || (P = Promise))(function (resolve, reject) {
|
16 | function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
|
17 | function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
|
18 | function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
|
19 | step((generator = generator.apply(thisArg, _arguments || [])).next());
|
20 | });
|
21 | };
|
22 | var __generator = (this && this.__generator) || function (thisArg, body) {
|
23 | var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g;
|
24 | return g = { next: verb(0), "throw": verb(1), "return": verb(2) }, typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
|
25 | function verb(n) { return function (v) { return step([n, v]); }; }
|
26 | function step(op) {
|
27 | if (f) throw new TypeError("Generator is already executing.");
|
28 | while (_) try {
|
29 | if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
|
30 | if (y = 0, t) op = [op[0] & 2, t.value];
|
31 | switch (op[0]) {
|
32 | case 0: case 1: t = op; break;
|
33 | case 4: _.label++; return { value: op[1], done: false };
|
34 | case 5: _.label++; y = op[1]; op = [0]; continue;
|
35 | case 7: op = _.ops.pop(); _.trys.pop(); continue;
|
36 | default:
|
37 | if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
|
38 | if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
|
39 | if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
|
40 | if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
|
41 | if (t[2]) _.ops.pop();
|
42 | _.trys.pop(); continue;
|
43 | }
|
44 | op = body.call(thisArg, _);
|
45 | } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
|
46 | if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
|
47 | }
|
48 | };
|
49 | Object.defineProperty(exports, "__esModule", { value: true });
|
50 | exports.puppeteerTest = void 0;
|
51 | var node_logger_1 = require("@storybook/node-logger");
|
52 | var url_1 = require("./url");
|
53 | var config_1 = require("./config");
|
54 | exports.puppeteerTest = function (customConfig) {
|
55 | if (customConfig === void 0) { customConfig = {}; }
|
56 | var _a = __assign(__assign({}, config_1.defaultPuppeteerTestConfig), customConfig), storybookUrl = _a.storybookUrl, chromeExecutablePath = _a.chromeExecutablePath, getGotoOptions = _a.getGotoOptions, customizePage = _a.customizePage, getCustomBrowser = _a.getCustomBrowser, testBody = _a.testBody, setupTimeout = _a.setupTimeout, testTimeout = _a.testTimeout;
|
57 | var browser;
|
58 | var page;
|
59 | var testFn = function (_a) {
|
60 | var context = _a.context;
|
61 | return __awaiter(void 0, void 0, void 0, function () {
|
62 | var kind, framework, name, id, url, options, e_1;
|
63 | return __generator(this, function (_b) {
|
64 | switch (_b.label) {
|
65 | case 0:
|
66 | kind = context.kind, framework = context.framework, name = context.name, id = context.id;
|
67 | if (framework === 'react-native') {
|
68 |
|
69 | node_logger_1.logger.error("It seems you are running puppeteer test on RN app and it's not supported. Skipping test.");
|
70 | return [2 ];
|
71 | }
|
72 | url = url_1.constructUrl(storybookUrl, id);
|
73 | options = { context: context, url: url };
|
74 | if (testBody.filter != null && !testBody.filter(options)) {
|
75 | return [2 ];
|
76 | }
|
77 | if (!browser || !page) {
|
78 | node_logger_1.logger.error("Error when running puppeteer test for " + kind + " - " + name + " : It seems the headless browser is not running.");
|
79 | throw new Error('no-headless-browser-running');
|
80 | }
|
81 | _b.label = 1;
|
82 | case 1:
|
83 | _b.trys.push([1, 4, , 5]);
|
84 | return [4 , customizePage(page)];
|
85 | case 2:
|
86 | _b.sent();
|
87 | return [4 , page.goto(url, getGotoOptions(options))];
|
88 | case 3:
|
89 | _b.sent();
|
90 | return [3 , 5];
|
91 | case 4:
|
92 | e_1 = _b.sent();
|
93 | node_logger_1.logger.error("Error when connecting to " + url + ", did you start or build the storybook first? A storybook instance should be running or a static version should be built when using puppeteer test feature.");
|
94 | throw e_1;
|
95 | case 5: return [4 , testBody(page, options)];
|
96 | case 6:
|
97 | _b.sent();
|
98 | return [2 ];
|
99 | }
|
100 | });
|
101 | });
|
102 | };
|
103 | testFn.timeout = testTimeout;
|
104 | var cleanup = function () { return __awaiter(void 0, void 0, void 0, function () {
|
105 | return __generator(this, function (_a) {
|
106 | switch (_a.label) {
|
107 | case 0:
|
108 | if (!(getCustomBrowser && page)) return [3 , 2];
|
109 | return [4 , page.close()];
|
110 | case 1:
|
111 | _a.sent();
|
112 | return [3 , 4];
|
113 | case 2:
|
114 | if (!browser) return [3 , 4];
|
115 | return [4 , browser.close()];
|
116 | case 3:
|
117 | _a.sent();
|
118 | _a.label = 4;
|
119 | case 4: return [2 ];
|
120 | }
|
121 | });
|
122 | }); };
|
123 | process.on('SIGINT', function () { return __awaiter(void 0, void 0, void 0, function () {
|
124 | return __generator(this, function (_a) {
|
125 | switch (_a.label) {
|
126 | case 0: return [4 , cleanup()];
|
127 | case 1:
|
128 | _a.sent();
|
129 | process.exit();
|
130 | return [2 ];
|
131 | }
|
132 | });
|
133 | }); });
|
134 | testFn.afterAll = cleanup;
|
135 | var beforeAll = function () { return __awaiter(void 0, void 0, void 0, function () {
|
136 | var puppeteer;
|
137 | return __generator(this, function (_a) {
|
138 | switch (_a.label) {
|
139 | case 0:
|
140 | if (!getCustomBrowser) return [3 , 2];
|
141 | return [4 , getCustomBrowser()];
|
142 | case 1:
|
143 | browser = _a.sent();
|
144 | return [3 , 4];
|
145 | case 2:
|
146 | puppeteer = require('puppeteer');
|
147 | return [4 , puppeteer.launch({
|
148 | args: ['--no-sandbox ', '--disable-setuid-sandbox', '--disable-dev-shm-usage'],
|
149 | executablePath: chromeExecutablePath,
|
150 | })];
|
151 | case 3:
|
152 |
|
153 | browser = _a.sent();
|
154 | _a.label = 4;
|
155 | case 4: return [4 , browser.newPage()];
|
156 | case 5:
|
157 | page = _a.sent();
|
158 | return [2 ];
|
159 | }
|
160 | });
|
161 | }); };
|
162 | beforeAll.timeout = setupTimeout;
|
163 | testFn.beforeAll = beforeAll;
|
164 | return testFn;
|
165 | };
|