Improved Set-SeCookie Function.

Added support for multiple cookie parameters: Name, Value, Path,Domain, ExpiryDate

Author: the-mentor

Selenium.psm1

@@ -263,10 +263,55 @@ function Remove-SeCookie {
 }
 
 function Set-SeCookie {
-    param($Driver, $name, $value)
+    param(
+        $Driver, 
+        [string]$Name, 
+        [string]$Value,
+        [string]$Path,
+        [string]$Domain,
+        [datetime]$ExpiryDate
+        )
+
+    <# Selenium Cookie Information
+    Cookie(String, String)
+    Initializes a new instance of the Cookie class with a specific name and value.
+    Cookie(String, String, String)
+    Initializes a new instance of the Cookie class with a specific name, value, and path.
+    Cookie(String, String, String, Nullable<DateTime>)
+    Initializes a new instance of the Cookie class with a specific name, value, path and expiration date.
+    Cookie(String, String, String, String, Nullable<DateTime>)
+    Initializes a new instance of the Cookie class with a specific name, value, domain, path and expiration date. 
+    #>
+
+    if($Name -and $Value -and (!$Path -and !$Domain -and !$ExpiryDate)){
+        $cookie = New-Object -TypeName OpenQA.Selenium.Cookie -ArgumentList $Name,$Value
+    }
+    Elseif($Name -and $Value -and $Path -and (!$Domain -and !$ExpiryDate)){
+        $cookie = New-Object -TypeName OpenQA.Selenium.Cookie -ArgumentList $Name,$Value,$Path
+    }
+    Elseif($Name -and $Value -and $Path -and $ExpiryDate -and !$Domain){
+        $cookie = New-Object -TypeName OpenQA.Selenium.Cookie -ArgumentList $Name,$Value,$Path,$ExpiryDate
+    }
+    Elseif($Name -and $Value -and $Path -and $ExpiryDate -and $Domain){
+        if($Driver.Url -match $Domain){
+            $cookie = New-Object -TypeName OpenQA.Selenium.Cookie -ArgumentList $Name,$Value,$Domain,$Path,$ExpiryDate
+        }
+        else{
+            Throw 'In order to set the cookie the browser needs to be on the cookie domain URL'
+        }
+    }
+    else{
+        Throw "Incorrect Cookie Layout:
+        Cookie(String, String)
+        Initializes a new instance of the Cookie class with a specific name and value.
+        Cookie(String, String, String)
+        Initializes a new instance of the Cookie class with a specific name, value, and path.
+        Cookie(String, String, String, Nullable<DateTime>)
+        Initializes a new instance of the Cookie class with a specific name, value, path and expiration date.
+        Cookie(String, String, String, String, Nullable<DateTime>)
+        Initializes a new instance of the Cookie class with a specific name, value, domain, path and expiration date."
+    }
 
-    $cookie = New-Object -TypeName OpenQA.Selenium.Cookie -ArgumentList $Name, $value
-    
     $Driver.Manage().Cookies.AddCookie($cookie)
 }