add qawolf.stopVideos() to templates

qawolf · Mar 11, 2020 · 28e9911 · 28e9911
1 parent 517c8b7
commit 28e9911
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 4 deletions.
diff --git a/src/build-code/buildTemplate.ts b/src/build-code/buildTemplate.ts
@@ -82,6 +82,7 @@ if (require.main === module) {
     ${buildNewContext(device)}
     await qawolf.register(context);
     await ${validName}(context);
+    await qawolf.stopVideos();
     await browser.close();
   })();
 }`;
@@ -107,7 +108,10 @@ beforeAll(async () => {
   page = await context.newPage();
 });
 
-afterAll(() => browser.close());
+afterAll(async () => {
+  await qawolf.stopVideos();
+  await browser.close();
+});
 
 test('${name}', async () => {
   await page.goto("${url}");${buildSetState(statePath)}

diff --git a/test/build-code/__snapshots__/buildTemplate.test.ts.snap b/test/build-code/__snapshots__/buildTemplate.test.ts.snap
@@ -18,6 +18,7 @@ if (require.main === module) {
     const context = await browser.newContext();
     await qawolf.register(context);
     await myScript(context);
+    await qawolf.stopVideos();
     await browser.close();
   })();
 }"
@@ -37,7 +38,10 @@ beforeAll(async () => {
   page = await context.newPage();
 });
 
-afterAll(() => browser.close());
+afterAll(async () => {
+  await qawolf.stopVideos();
+  await browser.close();
+});
 
 test('myTest', async () => {
   await page.goto(\\"www.qawolf.com\\");
@@ -63,6 +67,7 @@ if (require.main === module) {
     const context = await browser.newContext();
     await qawolf.register(context);
     await myScript(context);
+    await qawolf.stopVideos();
     await browser.close();
   })();
 }"
@@ -91,6 +96,7 @@ if (require.main === module) {
     });
     await qawolf.register(context);
     await myScript(context);
+    await qawolf.stopVideos();
     await browser.close();
   })();
 }"
@@ -115,7 +121,10 @@ beforeAll(async () => {
   page = await context.newPage();
 });
 
-afterAll(() => browser.close());
+afterAll(async () => {
+  await qawolf.stopVideos();
+  await browser.close();
+});
 
 test('myTest', async () => {
   await page.goto(\\"www.qawolf.com\\");
@@ -142,6 +151,7 @@ if (require.main === module) {
     const context = await browser.newContext();
     await qawolf.register(context);
     await myScript(context);
+    await qawolf.stopVideos();
     await browser.close();
   })();
 }"
@@ -161,7 +171,10 @@ beforeAll(async () => {
   page = await context.newPage();
 });
 
-afterAll(() => browser.close());
+afterAll(async () => {
+  await qawolf.stopVideos();
+  await browser.close();
+});
 
 test('myTest', async () => {
   await page.goto(\\"www.qawolf.com\\");