Stop open driver on Module unload

itfranck · itfranck · commit 12e8e68dc208 · 2020-10-11T14:35:53.000-04:00
diff --git a/Internal/init.ps1 b/Internal/init.ps1
@@ -1,5 +1,14 @@
 ﻿using namespace System.Collections.Generic
 
+
+if (!$PSCommandPath.EndsWith('init.ps1')) {
+    $MyInvocation.MyCommand.ScriptBlock.Module.OnRemove = { 
+         Get-SeDriver | Stop-SeDriver
+    }
+
+}
+
+
 Function New-Condition {
     Param([Parameter(Mandatory = $true)]$Text, [Type]$ValueType, $Tooltip, [Switch]$OptionalValue, $ElementRequired = $true )
     return [PSCustomObject]@{
diff --git a/Output/Selenium/Selenium.psm1 b/Output/Selenium/Selenium.psm1
diff --git a/Selenium.psm1 b/Selenium.psm1
@@ -13,5 +13,10 @@ ForEach ($folder in $functionFolders) {
         }
     }    
 }
+
+$MyInvocation.MyCommand.ScriptBlock.Module.OnRemove = { 
+    Get-SeDriver | Stop-SeDriver
+}
+
 $publicFunctions = (Get-ChildItem -Path "$PSScriptRoot\Public" -Filter '*.ps1').BaseName
 Export-ModuleMember -Function $publicFunctions

Original file line number	Diff line number	Diff line change
`@@ -13,5 +13,10 @@ ForEach ($folder in $functionFolders) {`
`13`	`13`	`}`
`14`	`14`	`}`
`15`	`15`	`}`
	`16`	`+`
	`17`	`+$MyInvocation.MyCommand.ScriptBlock.Module.OnRemove = {`
	`18`	`+ Get-SeDriver \| Stop-SeDriver`
	`19`	`+}`
	`20`	`+`
`16`	`21`	`$publicFunctions = (Get-ChildItem -Path "$PSScriptRoot\Public" -Filter '*.ps1').BaseName`
`17`	`22`	`Export-ModuleMember -Function $publicFunctions`